arXiv:2511.18834cs.CVcs.AI2025-11被引 2

让流匹配生成更快更准,通过引导学生模仿教师的真实生成路径。

FlowSteer: Guiding Few-Step Image Synthesis with Authentic Trajectories

  • 用在线轨迹对齐修复训练中的分布错位问题。
  • 在少步采样下,SD3上生成质量显著提升。
  • 适合追求高效高质图像生成的研究者与开发者。

流匹配在视觉生成中取得成功,但采样效率仍是实际应用的关键瓶颈。在加速方法中,ReFlow虽具理论一致性却因性能不佳被忽视,主要因其在实际场景中表现逊于一致性蒸馏和得分蒸馏。本文研究该问题,提出FlowSteer,通过引导学生模型沿教师的原始生成轨迹进行蒸馏,释放ReFlow潜力。首先,发现分段式ReFlow受训练阶段分布不匹配影响,提出在线轨迹对齐(OTA)解决此问题;其次,引入直接作用于微分方程轨迹的对抗性蒸馏目标,增强学生对教师轨迹的遵循;此外,发现并修复广泛使用的FlowMatchEulerDiscreteScheduler中一个此前未察觉的缺陷,显著改善少步推理质量。实验结果在SD3上验证了方法有效性。

原文摘要 · Abstract (English)

With the success of flow matching in visual generation, sampling efficiency remains a critical bottleneck for its practical application. Among flow models' accelerating methods, ReFlow has been somehow overlooked although it has theoretical consistency with flow matching. This is primarily due to its suboptimal performance in practical scenarios compared to consistency distillation and score distillation. In this work, we investigate this issue within the ReFlow framework and propose FlowSteer, a method unlocks the potential of ReFlow-based distillation by guiding the student along teacher's authentic generation trajectories. We first identify that Piecewised ReFlow's performance is hampered by a critical distribution mismatch during the training and propose Online Trajectory Alignment(OTA) to resolve it. Then, we introduce a adversarial distillation objective applied directly on the ODE trajectory, improving the student's adherence to the teacher's generation trajectory. Furthermore, we find and fix a previously undiscovered flaw in the widely-used FlowMatchEulerDiscreteScheduler that largely degrades few-step inference quality. Our experiment result on SD3 demonstrates our method's efficacy.

图像生成流匹配少步采样蒸馏

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