arXiv:2410.21966cs.CV2024-10NeurIPS被引 25

用强化学习让修复图像更符合人类审美,效果显著提升。

PrefPaint: Aligning Image Inpainting Diffusion Model with Human Preference

  • 构建5.1万张带人类偏好标注的数据集,训练奖励模型
  • 通过强化学习优化扩散模型,使修复结果更贴近人眼偏好
  • 适用于需要视觉美感的生成任务,如图像扩展与3D重建

本文首次提出通过强化学习框架,将图像修复扩散模型与人类审美标准对齐,显著提升修复图像的质量与视觉吸引力。我们构建了一个包含近51,000张带人类偏好标注的图像数据集,用于训练奖励模型;随后采用强化学习过程,沿更高奖励方向微调预训练扩散模型的分布。此外,我们从理论上推导了奖励模型误差的上界,说明其在强化对齐过程中具有潜在可信度,有助于精确正则化。在图像修复对比及图像扩展、3D重建等下游任务上的大量实验表明,本方法相比现有最优方法在人眼偏好对齐方面有显著提升。该研究不仅推动了图像修复领域发展,还为基于奖励建模精度的生成模型迭代优化提供了通用框架,对视觉驱动型AI应用设计具有广泛意义。代码与数据集已公开于https://prefpaint.github.io。

原文摘要 · Abstract (English)

In this paper, we make the first attempt to align diffusion models for image inpainting with human aesthetic standards via a reinforcement learning framework, significantly improving the quality and visual appeal of inpainted images. Specifically, instead of directly measuring the divergence with paired images, we train a reward model with the dataset we construct, consisting of nearly 51,000 images annotated with human preferences. Then, we adopt a reinforcement learning process to fine-tune the distribution of a pre-trained diffusion model for image inpainting in the direction of higher reward. Moreover, we theoretically deduce the upper bound on the error of the reward model, which illustrates the potential confidence of reward estimation throughout the reinforcement alignment process, thereby facilitating accurate regularization. Extensive experiments on inpainting comparison and downstream tasks, such as image extension and 3D reconstruction, demonstrate the effectiveness of our approach, showing significant improvements in the alignment of inpainted images with human preference compared with state-of-the-art methods. This research not only advances the field of image inpainting but also provides a framework for incorporating human preference into the iterative refinement of generative models based on modeling reward accuracy, with broad implications for the design of visually driven AI applications. Our code and dataset are publicly available at https://prefpaint.github.io.

图像修复扩散模型强化学习人类偏好

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