本文提出AREX,一种无需训练即可用于预训练流匹配模型的采样器。其核心思路是利用目标分布的均值与协方差,捕捉采样动态中可解析处理的部分:作者证明,矩匹配高斯目标的速度场是边缘速度场的L2最优仿射近似,据此将学习到的动态分解为贯穿整个采样路径的仿射分量(由目标前两阶矩确定)与神经残差项。AREX保留仿射分量并用显式矩阵值传播子进行积分,仅需对残差项做数值积分,这区别于只能解析处理各向同性线性动态的标量指数积分器。在图像生成与文本到图像生成任务中,AREX在少步采样条件下持续提升样本保真度,且无需重新训练底层模型。
| Flow Matching | 一种生成建模框架,通过匹配概率流的速度场来学习从噪声到数据的映射。 |
| Affine-Residual Decomposition | 将学习到的速度场分解为基于目标矩的仿射分量和神经残差项的方法。 |
| Exponential Integrator | 一种数值积分方法,利用矩阵指数精确处理线性部分,从而更稳定高效地求解微分方程。 |
| Few-Step Sampling | 在生成模型中仅使用少量采样步骤(如1-5步)快速生成高质量样本的技术。 |
| Moment Matching | 使近似分布的均值和协方差与目标分布一致的方法,常用于高斯近似。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅