Margin Dynamics for Large Language Model Alignment
Abstract
Practitioners choose among LLM alignment methods largely by trial and error. We provide a theoretical foundation: each pairwise method defines a vector field on the probability simplex that governs how the model redistributes mass between preferred and dispreferred completions. Projecting onto a single preferred/dispreferred pair reduces this to a one-dimensional margin ODE with a closed-form solution. These solutions reveal that DPO margins grow as while IPO margins converge exponentially to ; that label noise induces a qualitative phase transition in DPO (unbounded bounded) but affects IPO only quantitatively; and that gradient concentration under asymmetric weighting governs trainability at scale. We confirm all three predictions across five model families spanning four benchmarks. The closed-form solutions also enable principled method design: we prove that adding any label smoothing to SimPO creates a finite margin attractor at , while (standard SimPO) has no attractor. We test this across three model scales (3.8B–9B) and confirm it: the theory-derived configuration (TPO-S, ) achieves the highest mean IFEval score on all three models.
Related papers
See more related submissions when the market opens.