arXiv:2606.19162cs.LGcs.CV2026-06

用判别器引导强化学习,让生成模型更真实、结构更合理。

The Reward Was in Your Data All Along: Correcting Flow Matching with Discriminator-Guided RL

论文配图:The Reward Was in Your Data All Along: Correcting Flow Matching with Discriminator-Guided RL
图 1 · 摘自论文原文
  • 用预训练表示空间的判别器生成奖励信号,替代人工偏好。
  • 在SiT模型上,无指导FID从9.38降至2.62,语义距离降低至19.3。
  • 无需人类标注,即可提升图像质量与偏好对齐,适合生成模型优化场景。

Score-和flow-matching模型常依赖基于偏好的强化学习,以对齐主观偏好,并意外恢复视觉真实性和对象结构等本应由数据自身学习的属性。我们指出这反映结构不匹配:匹配损失在训练时边际下衡量速度或得分场的ℓ₂回归误差,与推理时决定样本质量的视觉和语义特性关联较弱。若拥有与这些特性对齐的奖励,强化学习可直接评估模型自生成样本并沿奖励梯度优化。关键挑战是获取此类奖励而不依赖昂贵的人类偏好。本文提出判别器引导强化学习(DRL):在预训练表示空间中训练判别器区分数据与基模型生成样本,以其logit作为KL正则化强化学习中的奖励。该预训练空间限制判别器仅关注感知有意义的方向,logit估计数据与模型间的对数似然比,即最优奖励以逼近数据分布。在SiT、JiT、REPA和RAE上,DRL显著降低无指导FID(如SiT从9.38→2.62)与语义空间FD(如DINOv3上SiT从88.2→19.3),所有骨干网络均表现一致提升,且无需训练于人类偏好仍改善其效果。后续偏好后训练中,也获得更优帕累托前沿,增强对齐同时减少过饱和、过亮等低级伪影。

原文摘要 · Abstract (English)

Score- and flow-matching models often rely on preference-based reinforcement learning for two purposes: aligning with subjective preferences and, surprisingly, recovering properties such as visual realism and coherent object structure that matching-based training is intended to learn from the data itself. We argue that this reflects a structural mismatch. Matching losses measure $\ell_2$ regression error on the velocity or score field under training-time marginals, a proxy poorly aligned with the visual and semantic properties that determine sample quality at inference. Given a reward aligned with these properties, RL sidesteps the mismatch by evaluating the model on its own samples and following the reward landscape directly. The challenge is to obtain such a reward without relying on human preferences, which are expensive and conflate data realism with annotator inclinations. We propose Discriminator-Guided RL (DRL). DRL trains a discriminator to separate data from base-model samples in a pretrained representation space and uses its logit as the reward in KL-regularized RL. The pretrained space restricts the discriminator to perceptually meaningful directions, and the logit estimates the log-likelihood ratio between data and model, which is the optimal reward for targeting the data distribution. Across SiT, JiT, REPA, and RAE, DRL reduces guidance-free FID (e.g., $9.38 \to 2.62$ on SiT) and semantic-space FD (e.g., $88.2 \to 19.3$ on DINOv3 for SiT), with consistent gains across all backbones, and improves human-preference rewards without training on them. It also yields a better Pareto frontier between preference reward and image fidelity under subsequent preference-based post-training, increasing alignment while reducing low-level artifacts such as oversaturation and excessive brightness.

生成模型强化学习判别器图像质量

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