Adaptive KL Control for Direct Preference Optimization in Instruction-Following LLMs

Read full text
Authors: Association for Artificial Intelligence 2026, Yi Chai
Cited by: 0

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