Dual-Difficulty Curriculum Learning for Direct Preference Optimization
Summary
2D-Curri-DPO is a novel framework enhancing Direct Preference Optimization (DPO) for Large Language Models (LLMs) by introducing a two-dimensional curriculum. This approach addresses limitations of traditional DPO and prior one-dimensional curriculum methods, which only consider pairwise distinguishability (PD), by jointly modeling Prompt Complexity (PC) and PD. The framework quantifies prompt semantic complexity and response preference clarity using dual difficulty metrics, defines a curriculum strategy space for task adaptation, and incorporates a KL-divergence-based adaptive mechanism for dynamic reference model updates to enhance training stability. Comprehensive experiments show that 2D-Curri-DPO significantly outperforms standard DPO and existing curriculum methods across benchmarks like MT-Bench, Vicuna Bench, and WizardLM. It achieves leading performance on UltraFeedback, with the S+PD strategy reaching an 89.5% win rate. Ablation studies confirm the benefits of the 2D structure and adaptive mechanisms.
Key takeaway
For Machine Learning Engineers optimizing LLM alignment, consider implementing 2D-Curri-DPO to significantly improve model performance and robustness. By structuring your DPO training with a two-dimensional curriculum that accounts for both prompt complexity and pairwise distinguishability, you can achieve superior results on challenging benchmarks. Prioritize sum-based strategies like S+PD for general alignment or S+PC for enhanced reasoning. This approach offers a more stable and effective path than traditional DPO or 1D curriculum methods, particularly for difficult alignment cases.
Key insights
Effective LLM alignment requires curriculum learning that considers both prompt complexity and preference distinguishability.
Principles
- LLM alignment benefits from structured curriculum learning.
- Dual difficulty dimensions improve DPO performance.
- Adaptive reference model updates enhance training stability.
Method
2D-Curri-DPO constructs a 2D (PC, PD) difficulty grid, quantifies PC via perplexity fluctuation, and uses KL-adaptive dynamic reference model updates for iterative DPO training across curriculum stages.
In practice
- Use quantile-based binning for PC and PD.
- Employ sum-based strategies (S+PD, S+PC) for best results.
- Set KL threshold δ=0.05 for adaptive updates.
Topics
- Large Language Models
- Direct Preference Optimization
- Curriculum Learning
- Prompt Complexity
- Pairwise Distinguishability
- LLM Alignment
Code references
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.