arXiv:2510.25229cs.CV2025-10NeurIPS被引 5

通过引入真实图像提升生成效率,让流模型更准更快。

Balanced conic rectified flow

  • 用真实与生成图像混合训练,减少对大量生成数据的依赖。
  • 在CIFAR-10上仅用1/8的生成对,FID显著降低。
  • 路径更直线、不饱和,适合追求高效高质生成的研究者。

矩形流是一种基于常微分方程(ODE)学习分布间平滑传输映射的生成模型。与需昂贵数值积分的扩散模型不同,矩形流通过称为re-flow的迭代过程学习平滑且笔直的ODE路径,实现高效高质量图像生成。然而,该方法仍面临挑战:1)re-flow需大量生成样本以保留目标分布,计算成本高;2)因仅用生成图像对训练,性能严重依赖1-矩形流模型,易偏向生成数据。本文实验揭示原方法局限,并提出新方案,将真实图像纳入训练。通过保持真实图像的ODE路径,有效降低对海量生成数据的依赖。在CIFAR-10上,仅使用原方法1/8的生成对,即可在单步生成和全步模拟中获得更优的FID得分。同时,路径更直线,re-flow期间不饱和于生成图像,提升了ODE学习鲁棒性并更好保留真实分布。

原文摘要 · Abstract (English)

Rectified flow is a generative model that learns smooth transport mappings between two distributions through an ordinary differential equation (ODE). Unlike diffusion-based generative models, which require costly numerical integration of a generative ODE to sample images with state-of-the-art quality, rectified flow uses an iterative process called reflow to learn smooth and straight ODE paths. This allows for relatively simple and efficient generation of high-quality images. However, rectified flow still faces several challenges. 1) The reflow process requires a large number of generative pairs to preserve the target distribution, leading to significant computational costs. 2) Since the model is typically trained using only generated image pairs, its performance heavily depends on the 1-rectified flow model, causing it to become biased towards the generated data. In this work, we experimentally expose the limitations of the original rectified flow and propose a novel approach that incorporates real images into the training process. By preserving the ODE paths for real images, our method effectively reduces reliance on large amounts of generated data. Instead, we demonstrate that the reflow process can be conducted efficiently using a much smaller set of generated and real images. In CIFAR-10, we achieved significantly better FID scores, not only in one-step generation but also in full-step simulations, while using only of the generative pairs compared to the original method. Furthermore, our approach induces straighter paths and avoids saturation on generated images during reflow, leading to more robust ODE learning while preserving the distribution of real images.

生成模型流网络高效生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。