Adaptive KL Control for Direct Preference Optimization in Instruction-Following LLMs
Abstract
Direct Preference Optimization (DPO) typically relies on a fixed inverse-temperature β that controls divergence from a reference model. Fixed β is brittle: too small causes underregularization (verbosity, safety drift); too large causes overregularization (underfitting). I propose an adaptive per-token KL controller using EMA smoothing, deadband filtering, and clipping to dynamically adjust β throughout training. Initial results on a 7B model show 72% win rate vs. base SFT and 60% vs. fixed-β DPO. The goal is a practical recipe for stable, compute-efficient DPO with reduced manual tuning.
Fields of Study & Subjects
Reinforcement Learning in Robotics
Topic Modeling
Natural Language Processing Techniques
Publication details
- Journal / source
- Underline Science Inc.
- Publication date
- 7 January 2026
- Language
- EN