arXiv · 2411.04712
SEE-DPO: Self Entropy Enhanced Direct Preference Optimization
Abstract
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.
Explore related subjects
Keep this discovery
Shivanshu Shekhar, Shreyas Singh, Tong Zhang. 2024-11-06. SEE-DPO: Self Entropy Enhanced Direct Preference Optimization. https://arxiv.org/abs/2411.04712
Cite the original work for its findings. Save a collection to share your selection of sources.