通过数据与模型双重课程优化,提升文生图偏好学习效果
Curriculum-DPO++: Direct Preference Optimization via Data and Model Curricula for Text-to-Image Generation
- 分阶段增加模型容量,先弱后强逐步训练
- 在9个基准上优于现有方法,文本对齐与美学更优
- 适合追求高质量文生图生成的开发者和研究者
直接偏好优化(DPO)作为人类反馈强化学习(RLHF)的高效替代方案,但未考虑不同偏好学习难度差异,导致优化效率不足。针对文生图任务,我们提出 Curriculum-DPO++,结合数据级课程与新型模型级课程。具体而言,动态提升去噪网络的学习能力:首先仅初始化部分可训练层,随训练进程逐步解冻;其次,在基于低秩适配(LoRA)的微调中,初始时降低低秩矩阵维度,逐步增大直至达到基线值。此外,设计新的排序策略。在九个基准上对比显示,Curriculum-DPO++ 在文本对齐、美学评分及人工偏好方面均超越现有方法。代码已开源。
原文摘要 · Abstract (English)
Direct Preference Optimization (DPO) has been proposed as an effective and efficient alternative to reinforcement learning from human feedback (RLHF). However, neither RLHF nor DPO take into account the fact that learning certain preferences is more difficult than learning other preferences, rendering the optimization process suboptimal. To address this gap in text-to-image generation, we recently proposed Curriculum-DPO, a method that organizes image pairs by difficulty. In this paper, we introduce Curriculum-DPO++, an enhanced method that combines the original data-level curriculum with a novel model-level curriculum. More precisely, we propose to dynamically increase the learning capacity of the denoising network as training advances. We implement this capacity increase via two mechanisms. First, we initialize the model with only a subset of the trainable layers used in the original Curriculum-DPO. As training progresses, we sequentially unfreeze layers until the configuration matches the full baseline architecture. Second, as the fine-tuning is based on Low-Rank Adaptation (LoRA), we implement a progressive schedule for the dimension of the low-rank matrices. Instead of maintaining a fixed capacity, we initialize the low-rank matrices with a dimension significantly smaller than that of the baseline. As training proceeds, we incrementally increase their rank, allowing the capacity to grow until it converges to the same rank value as in Curriculum-DPO. Furthermore, we propose an alternative ranking strategy to the one employed by Curriculum-DPO. Finally, we compare Curriculum-DPO++ against Curriculum-DPO and other state-of-the-art preference optimization approaches on nine benchmarks, outperforming the competing methods in terms of text alignment, aesthetics and human preference. Our code is available at https://github.com/CroitoruAlin/Curriculum-DPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。