Papers

CorrFlow 提出自校正流匹配框架,用轻量修正器降低生成误差

CorrFlow: Self-Correcting Flow Matching

Why we picked it

一篇很巧的生成模型论文:给流匹配加个轻量修正器,每步多算一次就能纠错,图像和视频生成质量都有提升。

arXiv 论文 CorrFlow 针对流匹配模型不会用中间表征修正速度预测的问题,训练一个轻量 refiner 来预测残差修正。该修正会在每次采样更新前加到基础模型的速度上,每步只需一次 refiner 计算,无需额外的基础模型前向传播。论文证明了在相同训练算力下,refiner 训练的残差风险降低优于继续训练基础模型。实验覆盖类条件图像生成、文生视频和图生视频三类任务,在视觉质量和物理一致性上均超过对应基础模型。

Original · arXiv cs.AI

CorrFlow: Self-Correcting Flow Matching

Flow-matching models generate samples through iterative velocity predictions, but typically do not revisit these predictions using the intermediate representations that produced them. We introduce CorrFlow, a self-correcting flow-matching framework that reuses this information through a lightweight refiner. Under rectified-flow interpolation, the error of the one-step clean prediction equals the realized velocity residual scaled by the noise level. Motivated by this identity, the refiner learns to predict residual corrections from the base model's clean prediction, conditioning, noise level and hidden states from the same forward pass. The correction is added to the base velocity before each sampling update, so subsequent predictions are made from the corrected state. This requires no additional base-model evaluations and only one lightweight refiner evaluation per step. Our theoretical analyses establish conditions under which feature reuse outperforms raw-input affine predictors and refiner training achieves greater residual-risk reduction than continued base-model training under a matched training compute budget. Experiments on class-conditional image generation, text-to-video generation and image-to-video generation show consistent improvements over the corresponding base models, including gains in visual quality and physical consistency. Project Page: https://fogradio.github.io/corrflow-project