Flow Matching with Uncertainty Quantification and Guidance
Uncertainty-aware flow matching (UA-Flow) is an extension of flow matching that predicts both the mean and variance of the velocity field, modeling heteroscedastic uncertainty. It propagates this velocity uncertainty through the flow dynamics to estimate per-sample reliability signals. These signals are then used for guidance techniques like uncertainty-aware classifier guidance and classifier-free guidance to steer generation toward higher fidelity samples.