arXiv:2410.05255cs.CVcs.LG2024-10被引 9

用自采样优化让扩散模型对齐更稳定且泛化更强。

Bridging SFT and DPO for Diffusion Model Alignment with Self-Sampling Preference Optimization

  • 通过历史检查点生成配对数据,避免依赖额外偏好数据。
  • 动态切换DPO与SFT,提升训练稳定性与样本质量评估精度。
  • 在图文和文生视频任务中均超越现有方法,适合高要求生成场景。

现有后训练技术主要分为监督微调(SFT)和强化学习(RL)两类:前者训练稳定但泛化能力有限,后者虽具更强泛化性,却需额外偏好数据或奖励模型,且存在奖励滥用风险。为兼顾SFT的稳定性与RL的泛化能力,本文提出自采样偏好优化(SSPO)方法。SSPO引入随机检查点重放(RCR)策略,利用历史检查点构建配对数据,有效缓解过拟合;同时采用自采样正则化(SSR)策略,动态评估生成样本质量,当样本更可能为优等样本时,自动从DPO切换至SFT,确保训练过程准确反映样本质量。实验表明,SSPO在文本到图像基准上优于所有现有方法,并在文本到视频任务中也展现出卓越性能。

原文摘要 · Abstract (English)

Existing post-training techniques are broadly categorized into supervised fine-tuning (SFT) and reinforcement learning (RL) methods; the former is stable during training but suffers from limited generalization, while the latter, despite its stronger generalization capability, relies on additional preference data or reward models and carries the risk of reward exploitation. In order to preserve the advantages of both SFT and RL -- namely, eliminating the need for paired data and reward models while retaining the training stability of SFT and the generalization ability of RL -- a new alignment method, Self-Sampling Preference Optimization (SSPO), is proposed in this paper. SSPO introduces a Random Checkpoint Replay (RCR) strategy that utilizes historical checkpoints to construct paired data, thereby effectively mitigating overfitting. Simultaneously, a Self-Sampling Regularization (SSR) strategy is employed to dynamically evaluate the quality of generated samples; when the generated samples are more likely to be winning samples, the approach automatically switches from DPO (Direct Preference Optimization) to SFT, ensuring that the training process accurately reflects the quality of the samples. Experimental results demonstrate that SSPO not only outperforms existing methods on text-to-image benchmarks, but its effectiveness has also been validated in text-to-video tasks. We validate SSPO across both text-to-image and text-to-video benchmarks. SSPO surpasses all previous approaches on the text-to-image benchmarks and demonstrates outstanding performance on the text-to-video benchmarks.

扩散模型对齐优化生成任务

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。