提升图像生成质量与多样性平衡,让模型更符合人类偏好。
ImageReFL: Balancing Quality and Diversity in Human-Aligned Diffusion Models
- 后期才引入奖励调优,前期保留基础模型以维持多样性
- 在标准评测中同时优于传统方法的质量与多样性表现
- 适合需要高质量且多样的图像生成场景
扩散模型在图像生成方面取得显著进展,但如何对齐人类偏好仍具挑战。基于人类反馈的奖励微调虽能提升对齐度,却常牺牲多样性,导致输出趋同。本文提出两种改进:一是提出“联合生成”采样策略,仅在生成后期使用奖励调优模型,早期保留基础模型以避免过拟合并保持全局结构;二是提出ImageReFL微调方法,通过真实图像训练并引入扩散损失与ReFL损失等多种正则化项,在质量损失极小的前提下显著提升多样性。实验显示该方法在质量和多样性指标上均优于传统奖励调优,用户研究进一步验证其在人类偏好与视觉多样性间实现更好平衡。代码已开源。
原文摘要 · Abstract (English)
Recent advances in diffusion models have led to impressive image generation capabilities, but aligning these models with human preferences remains challenging. Reward-based fine-tuning using models trained on human feedback improves alignment but often harms diversity, producing less varied outputs. In this work, we address this trade-off with two contributions. First, we introduce \textit{combined generation}, a novel sampling strategy that applies a reward-tuned diffusion model only in the later stages of the generation process, while preserving the base model for earlier steps. This approach mitigates early-stage overfitting and helps retain global structure and diversity. Second, we propose \textit{ImageReFL}, a fine-tuning method that improves image diversity with minimal loss in quality by training on real images and incorporating multiple regularizers, including diffusion and ReFL losses. Our approach outperforms conventional reward tuning methods on standard quality and diversity metrics. A user study further confirms that our method better balances human preference alignment and visual diversity. The source code can be found at https://github.com/ControlGenAI/ImageReFL .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。