本文提出了一种名为“后验流匹配”(Posterior Flow Matching, PFM)的生成建模框架,旨在解决传统流匹配方法仅预测均值、无法直接反映中间状态端点不确定性的问题。PFM通过高斯后验模型同时学习条件速度均值与逐分量方差,使单次模型评估即可同时输出传输预测和端点不确定性,从而在生成全过程中提供后验诊断信息。基于学习到的矩,该方法支持均值驱动的常微分方程和后验驱动的随机微分方程两种采样方式,后者可根据预测不确定性自适应调整局部噪声尺度。在图像、点云、分子和动作生成任务中,PFM不仅提升了生成质量,还能在生成过程中持续提供端点后验读数,为生成模型的不确定性量化提供了新思路。
| 流匹配 | 一种生成建模方法,学习将噪声分布传输到数据分布的连续变换。 |
| 后验流匹配 | 本文提出的框架,通过高斯后验模型同时学习条件速度均值和方差,以提供端点不确定性。 |
| 常微分方程 | 一种确定性微分方程,用于描述状态随时间的连续演化,在生成模型中常用于均值驱动的传输。 |
| 随机微分方程 | 一种包含随机项的微分方程,用于描述带噪声的连续演化,在生成模型中用于后验驱动的采样。 |
| 端点不确定性 | 在生成过程的中间状态,对最终生成结果(端点)所剩余的不确定性的度量。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅