acceptodds
Published as a conference paper at NeurIPS 2026

Margin Dynamics for Large Language Model Alignment

Xingzi Xu, Saygin Seyfioglu, Karim Bouyarmane

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.