Flow Matching with Uncertainty Quantification and Guidance
Uncertainty-aware flow matching (UA-Flow) is proposed, a lightweight extension of flow matching that predicts the velocity field together with heteroscedastic uncertainty that tracks sample fidelity more consistently than element-wise baselines and that these signals also carry information about data-manifold support.