解决流模型强化微调中的视觉幻觉问题,提升生成一致性。
ConsistentRFT: Reducing Visual Hallucinations in Flow-based Reinforcement Fine-Tuning
- 设计动态粒度采样机制,平衡全局语义与局部细节探索。
- 提出一致策略梯度优化,保持模型跨步骤生成一致性,降低幻觉率49%~38%。
- 适用于需高保真生成的场景,如图像编辑、可控生成任务。
在流模型上进行强化微调(RFT)对偏好对齐至关重要,但常引发过度优化细节和语义错位等视觉幻觉。本文从统一视角分析现有RFT方法,揭示其源于探索与利用两个层面的问题:(1)随机微分方程(SDE)推演中探索受限,导致过度关注局部细节而牺牲全局语义;(2)策略梯度方法固有的轨迹模仿过程,扭曲模型基础向量场及其跨步一致性。为此,我们提出ConsistentRFT框架,设计动态粒度采样(DGR)机制,通过动态调度不同噪声源,平衡全局与局部探索;引入一致策略梯度优化(CPGO),通过与更稳定的先验对齐,保持策略一致性。大量实验表明,ConsistentRFT显著减少视觉幻觉,低层与高层感知幻觉平均分别降低49%和38%。同时,在域外指标上优于其他RFT方法,相较基线提升5.1%(基准下降0.4%),在FLUX1.dev上表现突出。
原文摘要 · Abstract (English)
Reinforcement Fine-Tuning (RFT) on flow-based models is crucial for preference alignment. However, they often introduce visual hallucinations like over-optimized details and semantic misalignment. This work preliminarily explores why visual hallucinations arise and how to reduce them. We first investigate RFT methods from a unified perspective, and reveal the core problems stemming from two aspects, exploration and exploitation: (1) limited exploration during stochastic differential equation (SDE) rollouts, leading to an over-emphasis on local details at the expense of global semantics, and (2) trajectory imitation process inherent in policy gradient methods, distorting the model's foundational vector field and its cross-step consistency. Building on this, we propose ConsistentRFT, a general framework to mitigate these hallucinations. Specifically, we design a Dynamic Granularity Rollout (DGR) mechanism to balance exploration between global semantics and local details by dynamically scheduling different noise sources. We then introduce a Consistent Policy Gradient Optimization (CPGO) that preserves the model's consistency by aligning the current policy with a more stable prior. Extensive experiments demonstrate that ConsistentRFT significantly mitigates visual hallucinations, achieving average reductions of 49\% for low-level and 38\% for high-level perceptual hallucinations. Furthermore, ConsistentRFT outperforms other RFT methods on out-of-domain metrics, showing an improvement of 5.1\% (v.s. the baseline's decrease of -0.4\%) over FLUX1.dev. This is \href{https://xiaofeng-tan.github.io/projects/ConsistentRFT}{Project Page}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。