How Sampling Shapes LLM Alignment: From One-Shot Optima to Iterative Dynamics

Feb 7, 2026·
Yurong Chen
,
Yu He
,
Michael I. Jordan
Fan Yao
Fan Yao
· 1 min read
Abstract
Standard methods for aligning large language models with human preferences learn from pairwise comparisons among sampled candidate responses and regularize toward a reference policy. Despite their effectiveness, the effects of sampling and reference choices are poorly understood theoretically. We investigate these effects through Identity Preference Optimization, a widely used preference alignment framework, and show that proper instance-dependent sampling can yield stronger ranking guarantees, while skewed on-policy sampling can induce excessive concentration under structured preferences. We then analyze iterative alignment dynamics in which the learned policy feeds back into future sampling and reference policies, reflecting a common practice of model-generated preference data. We prove that these dynamics can exhibit persistent oscillations or entropy collapse for certain parameter choices, and characterize regimes that guarantee stability. Our theoretical insights extend to Direct Preference Optimization, indicating the phenomena we captured are common to a broader class of preference-alignment methods. Experiments on real-world preference data validate our findings.
Type
publications

This a working paper under review…

Fan Yao
Authors
Assistant Professor

I am an assistant professor in the Department of Statistics and Operations Research (STOR) at UNC Chapel Hill. I received my PhD in Computer Science from the University of Virginia, advised by Haifeng Xu and Hongning Wang. Prior to obtaining my PhD, I spent two years at the University of Chicago and received both my BS and MS in Computational Mathematics from Peking University. My CV is available here.

My research interests include social and ethical aspects of AI, human-centered machine learning, strategic and multi-agent systems, recommender systems and digital platform economy. If you share similar interests, feel free to contact me via email. Prospective students are encouraged to review this before contacting me.

👈🏻 Find my email and working address by clicking the icons on the left.

Current status: ☕️ In Chapel Hill (☕️regular ✈️traveling 🏖vacation ⏳deadline mode)