MMODELYST
Papers/P-Guide: Parameter-Efficient Prior Steering for Single-Pass CFG Inference
PAP

P-Guide: Parameter-Efficient Prior Steering for Single-Pass CFG Inference

May 7, 2026

arXiv
Abstract

Classifier-Free Guidance (CFG) is essential for high-fidelity conditional generation in flow matching, yet it imposes significant computational overhead by requiring dual forward passes at each sampling step. In this work, we address this bottleneck by introducing \textbf{P-Guide}, a framework that achieves high-quality guidance through a single inference pass by modulating only the initial latent state. We further show that, under a first-order approximation, P-Guide is equivalent to CFG in the sense that it steers generation from the prior space, without requiring explicit velocity field extrapolation during sampling. We consider both homoscedastic and \textbf{heteroscedastic} priors, and find that jointly modeling the mean and variance enables adaptive loss attenuation and improved robustness to data uncertainty. Extensive experiments demonstrate that P-Guide reduces inference latency by approximately 50\% while maintaining fidelity and prompt alignment competitive with standard dual-pass CFG baselines.

Select text to highlight · click a highlight to remove · saved in this browser only
Authors
Xin Peng, Ang Gao
Your notes (browser-local)
saved
arXiv:2605.06124