无需修改模型,一步计算生成样本不确定性
Uncertainty in a Single Pass: A Closed-Form Identity for One-Step Flow Matching

- 基于速度场散度推导出后验协方差的闭式解
- 单步前向传播即可获得精确不确定性估计
- 适合需要高可靠性生成的医学影像任务
流匹配为生成建模提供了高效框架,但其生成样本的不确定性估计仍是根本性挑战。现有方法依赖辅助方差头、模型集成或迭代协方差传播,必然在噪声到图像轨迹中引入累积误差和结构干扰。本文通过将Tweedie公式适配至线性流匹配插值器,推导出后验协方差的精确闭式表达式。该表达式仅依赖于学习到的速度场的散度,可对预训练模型进行事后评估,无需架构修改。此解析形式在单步生成框架中尤为强大:通过一次前向传播计算端到端后验协方差,彻底规避了序列积分带来的误差累积。该机制显著提升鲁棒性,提供高精度且稳定的不确定性估计,避免多步近似中的数值退化。我们在CIFAR-10及真实脑部MRI扫描等基准数据集上验证了该框架。基于散度的不确定性图高度可解释,且与经验重建误差紧密相关。尤其在医学影像中,对结构模糊性的精确定位为肿瘤边界勾画与神经外科规划等高风险临床任务提供了关键决策支持。
原文摘要 · Abstract (English)
Flow matching provides a highly effective framework for generative modeling, yet estimating the uncertainty of its generated samples remains a fundamental challenge. Existing methods rely on auxiliary variance heads, model ensembles, or iterative covariance propagation, which invariably introduce accumulated errors and structural interference along the noise-to-image trajectory. We resolve these limitations by adapting Tweedie's formula to the linear flow matching interpolant, yielding an exact, closed-form identity for the posterior covariance. Because this identity relies exclusively on the divergence of the learned velocity field, it can be evaluated post hoc on pretrained models without any architectural modifications. This analytical formulation proves exceptionally powerful for single-step generative frameworks. By computing the end-to-end posterior covariance in one forward pass, our approach completely bypasses the compounding errors inherent to sequential integration. This mechanism fundamentally enhances robustness, providing highly accurate and stable uncertainty estimates while preventing the numerical degradation typical of multi-step approximations. We validate our framework across benchmark datasets, including CIFAR-10 and real brain MRI scans. The resulting divergence-based uncertainty maps are highly interpretable and tightly correlated with empirical reconstruction errors. Crucially, in medical imaging, this precise localization of structural ambiguity provides essential decision support for high-stakes clinical tasks such as tumor boundary delineation and neurosurgical planning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。