用少量数据高效对齐扩散模型,效果超越主流方法。
CRAFT: Aligning Diffusion Models with Fine-Tuning Is Easier Than You Think
- 用复合奖励筛选构建高质量小样本训练集
- 仅需100样本即超越千样本优化方法
- 训练速度比基线快11-220倍,适合资源有限者
对齐扩散模型在生成高质量、符合人类偏好的图像方面取得显著进展。现有方法如监督微调(SFT)依赖昂贵的高质量图像,而基于DPO的偏好优化则需要大规模且质量不一的偏好数据集,且计算效率低。为此,我们提出轻量级的复合奖励辅助微调(CRAFT):首先通过复合奖励过滤(CRF)构建高质量一致的小样本数据集,再进行增强版SFT。理论上证明CRAFT优化了群体强化学习的下界,建立了选样SFT与强化学习的原理联系。实验表明,CRAFT仅用100样本即可超越使用数千样本的最新偏好优化方法,且收敛速度比基线快11–220倍,展现出极高效率。
原文摘要 · Abstract (English)
Aligning Diffusion models has achieved remarkable breakthroughs in generating high-quality, human preference-aligned images. Existing techniques, such as supervised fine-tuning (SFT) and DPO-style preference optimization, have become principled tools for fine-tuning diffusion models. However, SFT relies on high-quality images that are costly to obtain, while DPO-style methods depend on large-scale preference datasets, which are often inconsistent in quality. Beyond data dependency, these methods are further constrained by computational inefficiency. To address these two challenges, we propose Composite Reward Assisted Fine-Tuning (CRAFT), a lightweight yet powerful fine-tuning paradigm that requires significantly reduced training data while maintaining computational efficiency. It first leverages a Composite Reward Filtering (CRF) technique to construct a high-quality and consistent training dataset and then perform an enhanced variant of SFT. We also theoretically prove that CRAFT actually optimizes the lower bound of group-based reinforcement learning, establishing a principled connection between SFT with selected data and reinforcement learning. Our extensive empirical results demonstrate that CRAFT with only 100 samples can easily outperform recent SOTA preference optimization methods with thousands of preference-paired samples. Moreover, CRAFT can even achieve 11-220$\times$ faster convergences than the baseline preference optimization methods, highlighting its extremely high efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。