A Probability-Quality Trade-off in Aligned Language Models and its Relation to Sampling Adaptors
Naaman Tan, Josef Valvoda, Tianyu Liu, Anej Svete, Yanxia Qin, Min-Yen Kan, Ryan Cotterell
Abstract
The relationship between the quality of a string, as judged by a human reader, and its probability, p(y) under a language model undergirds the development of better language models. For example, many popular algorithms for sampling from a language model have been conceived with the goal of manipulating p(y) to place higher probability on strings that humans deem of high quality (Fan et al., 2018; Holtzman et al., 2020) . In this article, we examine the probability-quality relationship in language models explicitly aligned to human preferences, e.g., through reinforcement learning through human feedback. We show that, when sampling corpora from an aligned language model, there exists a trade-off between the strings' average reward and average log-likelihood under the prior language model, i.e., the same model before alignment with human preferences. We provide a formal treatment of this phenomenon and demonstrate how a choice of sampling adaptor allows for a selection of how much likelihood we exchange for the reward. https://github.com/tanyjnaaman/ probability-quality-paradox
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on16
- Direct Preference Optimization: Your Language Model is Secretly a Reward ModelRafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D. Manning et al.NeurIPS 2023 · 10,924 citations
- The Curious Case of Neural Text DegenerationAri Holtzman, Jan Buys, Li Du, Maxwell Forbes et al.ICLR 2020 · 4,112 citations
- Scaling Laws for Reward Model OveroptimizationLeo Gao, John Schulman, Jacob HiltonICML 2023 · 963 citations
- A Contrastive Framework for Neural Text GenerationYixuan Su, Tian Lan, Yan Wang, Dani Yogatama et al.NeurIPS 2022 · 349 citations
- Understanding Dataset Difficulty with V-Usable InformationKawin Ethayarajh, Yejin Choi, Swabha SwayamdiptaICML 2022 · 337 citations
Related papers
- Towards Efficient Exact Optimization of Language Model AlignmentHaozhe Ji, Cheng Lu, Yilin Niu, Pei Ke et al.ICML 2024 · 32 citations
- Design Considerations in Offline Preference-based RLAlekh Agarwal, Christoph Dann, Teodor Vanislavov MarinovICML 2025
- COMAL: A Convergent Meta-Algorithm for Aligning LLMs with General PreferencesYixin Liu, Argyris Oikonomou, Weiqiang Zheng, Yang Cai et al.ICLR 2026 · 5 citations
- Variational Best-of-N AlignmentAfra Amini, Tim Vieira, Elliott Ash, Ryan CotterellICLR 2025 · 1 citation
- Reducing the Probability of Undesirable Outputs in Language Models Using Probabilistic InferenceStephen Zhao, Aidan Li, Rob Brekelmans, Roger Baker GrosseNeurIPS 2025
