arXiv:2606.03216cs.CV2026-06

用奖励模型集成提升图像修复的偏好对齐,效果超越现有方法。

Follow-Your-Preference++: Rethinking Preference Alignment for Image Inpainting

论文配图:Follow-Your-Preference++: Rethinking Preference Alignment for Image Inpainting
图 1 · 摘自论文原文
  • 通过集成多个奖励模型构建更可靠的偏好数据
  • 在多个基准上显著优于现有SOTA,人类评估也更优
  • 适用于修复与物体移除任务,且无需改动模型结构

本文研究图像修复中的偏好对齐问题。不提出新方法,而是从基础出发重新审视核心挑战。采用主流的直接偏好优化框架,利用公开可用的奖励模型构建偏好训练数据。实证研究涵盖九个奖励模型、两个基准和两种不同架构与生成机制的基线修复模型。主要发现:(1) 多数奖励模型能提供有效信号,但部分不可靠;(2) 在候选与采样规模变化下,偏好数据呈现一致趋势;(3) 奖励模型存在明显偏差,尤其在亮度、构图与色彩上,易导致奖励劫持;(4) 简单集成可缓解偏差,实现鲁棒且泛化性强的性能。此外,偏好对齐可迁移至物体移除任务,目标从创造性生成转为连贯背景补全;进一步分析表明,校准后的集成方法可进一步抑制劫持并提升鲁棒性。无需修改模型架构或引入额外数据集,本方法在标准指标、大视觉语言模型评估及人工评价中均显著超越先前最优模型。代码已开源:https://github.com/shenytzzz/Follow-Your-Preference。

原文摘要 · Abstract (English)

We study preference alignment for image inpainting. Rather than proposing yet another method, we revisit the problem from first principles and reassess its core challenges. We adopt the widely used direct preference optimization framework and construct preference training data with publicly available reward models. Our empirical study spans nine reward models, two benchmarks, and two baseline inpainting models that differ in architecture and generative mechanism. Our main findings are: (1) Most reward models provide valid signals for preference data construction, although some are unreliable as evaluators. (2) Across models and benchmarks, preference data exhibits consistent trends under both candidate and sample scaling. (3) Reward models display pronounced biases--particularly in brightness, composition, and color scheme--that make them prone to inducing reward hacking. (4) A simple ensemble of reward models mitigates such biases and yields robust, generalizable performance. {\color{rebuttal_blue}(5) Preference alignment is transferable to the object removal task, where the goal shifts from open-ended creative generation to coherent background completion. (6) Further analysis reveals that a calibrated ensemble method further mitigates hacking and improves robustness.} Without modifying model architectures or introducing additional datasets, our models substantially outperform prior state-of-the-art models on standard metrics, large vision-language model evaluations, and human assessments. Our code is available at: https://github.com/shenytzzz/Follow-Your-Preference.

图像修复偏好对齐奖励模型集成学习

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