Dual-Difficulty Curriculum Learning for Direct Preference Optimization

· Source: cs.AI updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, extended

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

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

Topics

Code references

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.