arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2411.04712cs.CVcs.LG

SEE-DPO:自熵增强的直接偏好优化

SEE-DPO: Self Entropy Enhanced Direct Preference Optimization

Siebel School of Computing and Data Science(计算与数据科学学院);University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Fractal AI Research(Fractal AI研究院)

Shivanshu Shekhar, Shreyas Singh, Tong Zhang

AI总结:

针对基于DPO的文本到图像扩散模型易过拟合、存在奖励黑客问题的缺陷,提出融入自熵正则化的SEE-DPO方法,提升训练稳定性与图像质量,在关键生成指标上达SOTA。

摘要:

直接偏好优化(DPO)已成功用于根据人类偏好对齐大语言模型(LLM),近期也被应用于提升文本到图像扩散模型的质量。然而,基于DPO的方法如SPO、Diffusion-DPO和D3PO极易出现过拟合和奖励黑客问题,尤其是在生成模型经过长时间训练、被优化以拟合分布外数据时。为克服这些挑战并稳定扩散模型的训练,我们在基于人类反馈的强化学习中引入了一种自熵正则化机制。该增强机制通过鼓励更广泛的探索和更强的鲁棒性来改进DPO训练。我们的正则化技术有效缓解了奖励黑客问题,提升了训练稳定性,并改善了整个潜空间内的图像质量。大量实验表明,将人类反馈与自熵正则化相结合,可显著提升图像的多样性和特异性,在关键图像生成指标上取得了当前最优(SOTA)结果。

英文摘要:

Direct Preference Optimization (DPO) has been successfully used to align large language models (LLMs) according to human preferences, and more recently it has also been applied to improving the quality of text-to-image diffusion models. However, DPO-based methods such as SPO, Diffusion-DPO, and D3PO are highly susceptible to overfitting and reward hacking, especially when the generative model is optimized to fit out-of-distribution during prolonged training. To overcome these challenges and stabilize the training of diffusion models, we introduce a self-entropy regularization mechanism in reinforcement learning from human feedback. This enhancement improves DPO training by encouraging broader exploration and greater robustness. Our regularization technique effectively mitigates reward hacking, leading to improved stability and enhanced image quality across the latent space. Extensive experiments demonstrate that integrating human feedback with self-entropy regularization can significantly boost image diversity and specificity, achieving state-of-the-art results on key image generation metrics.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑