Source-linked AI summary
Aligning Large Language Models with Human Feedback: Mathematical Foundations and Algorithm Design
Siliang Zeng, Luca Viano, Chenliang Li, Jiaxiang Li, Volkan Cevher, Markus Wulfmeier, Stefano Ermon, Alfredo Garcia, Mingyi Hong
TL;DR
LLM alignment seeks outputs that match human intentions and preferences while avoiding undesirable behavior. This article surveys alignment methods and their foundations; in dialogue-generation evaluations, its IRL-based method outperformed the compared SFT-initialized and SPIN models.
Problem
Alignment seeks to ensure that LLM outputs match human intentions or preferences without undesirable behaviors such as deception or manipulation.
Source: §1 · p. 1Method
The article introduces alignment fundamentals and state-of-the-art algorithms, connecting alignment methodologies with signal-processing techniques and preference-learning foundations.
Source: §1 · p. 2, §2 · p. 4, §4.4 · p. 16Results
In dialogue-generation evaluations, the IRL-based method consistently outperformed the Mistral-7B SFT checkpoint and SPIN, including with one or two self-generation rounds.
Source: §6.3.1 · pp. 25–26Takeaways & Limitations
The article offers researchers and practitioners a foundational understanding of alignment methodologies and a framework for developing more robust and scalable techniques.
Source: §1 · p. 2DPO can be significantly affected by distribution shift between model outputs and preference data, while RL-based fine-tuning such as PPO demands substantial compute and careful tuning.
Source: §4.3 · p. 16, §8 · p. 28
Abstract
from arXiv · showhide
This article provides an introduction to the mathematical foundations and algorithmic frameworks used to align Large Language Models (LLMs) with human intentions, preferences, and values. We discuss standard alignment techniques, such as fine-tuning (SFT), reinforcement learning with human feedback (RLHF), and direct preference optimization (DPO). We also explore the theoretical underpinnings of learning from human preferences, drawing connections to inverse reinforcement learning (IRL) and discrete choice models. We present state-of-the-art algorithms in a tutorial style, discuss their advantages and limitations, and offer insights into practical implementation. Our exposition is intended to serve as a comprehensive resource for researchers and practitioners, providing both a foundational understanding of alignment methodologies and a framework for developing more robust and scalable alignment techniques.
1 Introduction
The introduction frames alignment as training AI systems to produce outcomes people intend or prefer, while avoiding undesirable behavior. It presents the article as a guide to alignment fundamentals and algorithms, with links to signal-processing methods and problems.
- Alignment is framed as a performance guarantee that AI outcomes match human intentions or preferences without undesirable side effects such as deception or manipulation.Source: §1 · p. 1
- Base models can fall short on specific tasks and nuanced user preferences, motivating further adjustment using feedback from human experts.The feedback can reflect judgments about response accuracy and particular preferences; the introduction says aligned models are generally observed to follow instructions better and avoid toxic or non-preferable responses.Source: §1 · pp. 1–2
- The article introduces alignment fundamentals and state-of-the-art approaches, clarifying their intersections with signal-processing techniques.Source: §1 · p. 2
- The authors connect preference-model learning to inverse problems and argue that signal-processing expertise could contribute to alignment research.They draw an analogy to reconstructing signals from observed data and expect such cross-disciplinary work to foster innovation in AI safety and robust, generalizable alignment frameworks.Source: §1 · p. 2
2 Preliminaries
The preliminaries introduce the language-model training setup and explain how demonstrations and preference data support alignment. They distinguish pre-training, supervised fine-tuning, and RLHF, then outline common evaluation measures.
- Pre-training learns next-token predictions from large unlabeled text corpora, while a Transformer model defines the function class used in later alignment and optimization.The model predicts each next token from the prompt and preceding tokens.Source: §2 · pp. 3–4
- Alignment fine-tuning typically has two stages: SFT improves instruction following from demonstrations, and RLHF uses preferences to learn a reward model that guides policy optimization.SFT imitates expert responses; RLHF uses the learned reward with prompts to evaluate and improve generated responses.Source: §2 · pp. 4–5
- SFT demonstrations pair prompts with expert responses, whereas preference data records a human labeler’s preferred answer over a less-preferred one.The demonstrations may come from a human expert or a high-quality, well-aligned model.Source: §2 · p. 4
- The article defines alignment as training from diverse demonstrations and preferences so an LLM consistently generates outputs aligned with human values, ethical principles, and intended uses.Source: §2 · p. 5
- Evaluation varies by training stage and task: pre-training commonly uses perplexity, SFT uses task-specific measures, and RLHF may use human or judge-model win rates.The cited examples include accuracy, precision, and recall for classification and BLEU for translation; lower perplexity indicates greater confidence and accuracy in next-token prediction.Source: §2 · p. 5
3 Theoretical Foundations of Learning from Human Feedback.
The section connects learning from human feedback to discrete choice models, treating prompts as states and responses as actions. It then frames reward learning as finding a reward function that explains observed preferences or demonstrations.
3 Theoretical Foundations of Learning from Human Feedback.
- LLM responses and preference labels can be viewed as human choices, linking alignment to discrete choice modeling.The parallel maps prompts to states and possible responses to actions.Source: §3 · p. 6, §3.1 · p. 6
3.1 Classical models of discrete choice
- With i.i.d. standard Gumbel utility noise, the random-utility model yields the logit, or Bradley-Terry-Luce, model used to learn rewards from preference data.The model expresses preference probability using the rewards of two actions.Source: §3.1 · p. 6
3.2 The reward learning problem
- Reward learning seeks a reward function that rationalizes observed preferences, but likelihood maximization need not identify a unique reward without a known reference-action reward.For demonstrations, the section also formulates reward estimation as likelihood maximization under the assumed choice model.Source: §3.2 · p. 7
- With linearly parameterized rewards, the demonstration-likelihood problem is the Lagrangian dual of maximum-entropy estimation, under the stated complete-menu assumption.The choice model assumes the available actions are known.Source: §3.2 · p. 7, §3.1 · p. 7
4 State-of-the-art Algorithms for LLM Alignment
This section presents a mathematical formulation of LLM alignment and surveys state-of-the-art algorithms.
- The section overviews the mathematical formulation of alignment and state-of-the-art algorithms.Source: §4 · p. 8
4.1 Learning from demonstrations data
The section surveys alignment from demonstrations, from behavioral cloning through self-generated-data approaches and divergence minimization. It describes SFT as imitation of expert demonstrations and presents methods that modify or extend that objective.
4.1 Learning from demonstrations data
- SFT learns from prompt-answer demonstrations by treating each response as a golden answer and minimizing KL divergence to imitate the demonstration data.The section relates this stage to behavioral cloning.Source: §4.1 · p. 8
4.1.1 Behavioral cloning approaches
- Behavioral cloning is offline: the learner cannot collect new data with its current policy and must learn from the expert dataset.The section notes that exact loss minimization is unavailable for non-convex policy classes and that the effects of optimization error on performance remain unstudied.Source: §4.1.1 · p. 8
4.1.2 Self generation approaches
- SPIN casts demonstration fine-tuning as a two-player game: one player matches demonstrations while the other distinguishes generated answers from them, with alternating updates.The method uses self-generated data to recover a demonstration-level policy.Source: §4.1.2 · p. 9
4.1.3 Information theoretic approaches
- Forward KL minimization is mode-covering, while reversed KL can be mode-seeking but requires learner-generated prompts and answers.The section also mentions χ2 divergence minimization as another divergence-based approach.Source: §4.1.3 · p. 10
4.2 Learning from preferences
Learning from preferences separates reward estimation from policy fine-tuning: preference comparisons train a reward model, which then guides reinforcement learning. The discussion contrasts PPO with REINFORCE-family approaches and with demonstration-only training.
- Preference learning proceeds in two stages: fit a reward model from preference data, then fine-tune the LLM with reinforcement learning using prompts.These stages relate to two types of stochastic choice models discussed earlier.Source: §4.2 · pp. 11–12
- The policy objective maximizes learned reward while penalizing KL divergence from a reference model, such as the SFT model.The expected reward and KL are generally approximated with empirical losses and policy-optimization methods.Source: §4.2 · p. 12
- PPO uses conservative updates to counter high-variance gradients, but the cited argument says LLM fine-tuning may not need that mechanism; PPO also requires a critic.REINFORCE avoids critic training, and RLOO reduces variance by subtracting the average of the other sampled rewards while remaining unbiased.Source: §4.2 · pp. 12–13
- SFT-only models are reported to perform worse than models additionally trained with reward learning and policy optimization, while preference data is typically less costly to obtain than SFT data.The text attributes some of the generalization benefit to reward and RL guiding exploration beyond responses defined by SFT data.Source: §4.2 · p. 13
4.3 Learning from preferences without online generations: Direct Preference Optimization
Direct Preference Optimization (DPO) derives a policy objective from preference data that avoids the RL step, but its fit to the data can be vulnerable to limited coverage and distribution shift.
4.3 Learning from preferences without online generations: Direct Preference Optimization
- DPO uses the analytical solution to the RL objective to directly parameterize and optimize the policy, bypassing the need to perform RL.The pairwise reward difference cancels the intractable normalization constant.Source: §4.3 · pp. 13–15
- DPO is more memory efficient than RLHF because it keeps only the policy parameters, and it requires no self-generation, making implementation easier than RL-based algorithms.Figure 5 estimates memory consumption for Pythia-1B on TL;DR with batch-size 2 on one device.Source: Figure 5 · p. 14, §4.3 · p. 15
- For α > 0, policies minimizing the DPO loss can have support outside πref, unlike policies minimizing the RLHF loss.The DPO solution set is a strict superset of the RLHF solution set [46].Source: Table 1 · p. 15, §4.3 · p. 15
- DPO performance can be significantly affected by distribution shift between model outputs and preference data; online DPO is proposed as a possible mitigation.The article also reports that DPO can have lower winrate than RLHF in practical comparisons.Source: §4.3 · p. 16
- When preference data have limited answer coverage, DPO minimizers can assign positive probability to answers with no available information.The paper warns against applying DPO when the preference dataset poorly covers the answer set.Source: §4.3 · p. 16
4.4 Learning from preference without the BT assumption.
Learning preferences without the Bradley–Terry (BT) assumption accommodates non-transitive aggregate preferences that BT cannot represent, using preference-model learning followed by Nash-equilibrium optimization.
4.4 Learning from preference without the BT assumption
- Averaging three individually transitive preference models can yield a non-transitive cycle, with each of y1 ≻ y2, y2 ≻ y3, and y3 ≻ y1 assigned probability 2/3.The article uses this example to show why the BT assumption can discard preference data.Source: §4.4 · p. 16
- The method first learns a preference model by maximum likelihood, where P(y ≻ y′|x) represents an oracle’s probability of preferring answer y over y′ for prompt x.The model is fit using the preference dataset.Source: §4.4 · p. 17
- The method then finds a unique Nash equilibrium, dropping the BT assumption so it can accommodate a larger number of preference data.The min–max optimization requires careful conservative updates for last-iterate convergence.Source: §4.4 · pp. 16–17
- 0.598 win rate against an RLHF-aligned version of the same model was reported for MD1 on a text summarization task.The method requires min–max optimization rather than RLHF’s simpler minimization.Source: §4.4 · p. 17
4.5 Learning from data verifiers via Rule-Based Reinforcement Learning
Rule-based reinforcement learning can use verifiers instead of a reward model for correctness-focused tasks, but rule-only optimization may leave language skills poor.
4.5 Learning from data verifiers via Rule-Based Reinforcement Learning
- For math and programming problems, a solution verifier can replace a reward model by checking whether a solution is correct.The examples include LeetCode test cases for programming and OpenWebMath problems for math verifiers.Source: §4.5 · p. 17
- The described rule-based rewards assign 1 or −1 for correct or incorrect format and 2 when the answer fully matches ground truth.These are identified as format and answer rewards.Source: §4.5 · p. 18
- Pure rule-based reward optimization can produce poor language skills, including language mixing; the described approach adds SFT or preference-based learning in final stages.The initial rule-based reward stage is optimized via GRPO.Source: §4.5 · p. 18
5 A Unified Perspective for LLM Alignment
The section unifies demonstration-based and preference-based alignment through reward learning, framing both as likelihood-based estimation problems. It develops IRL from demonstrations and AIHF as a single-stage framework for using both demonstrations and preferences.
5 A Unified Perspective for LLM Alignment
- The stochastic-choice perspective motivates a shared account of demonstration- and preference-based alignment, which both learn rewards and maximize likelihoods.The section asks why demonstrations are not also used for reward learning and why SFT and RLHF are separated into stages.Source: §5 · p. 18
5.2 Jointly learning policy and reward from Integrated Human Feedback
- AIHF unifies alignment stages by jointly learning reward and policy from demonstration and preference likelihoods.Its reward is learned from both datasets; setting γ = 0 decouples reward and policy learning, recovering two-step RLHF.Source: §5 · p. 18, §5.2 · pp. 21–22
5.1 Jointly learning policy and reward from the demonstrations data
- IRL jointly learns a policy and reward from demonstrations; its surrogate objective is finite-sample and becomes asymptotically close to the original IRL problem.The article presents SFT and SPIN as special cases of the IRL-based approach, and notes that its surrogate simplifies the hierarchical optimization.Source: §5 · p. 18, §5.1 · pp. 19, 21
- IRL alternates reward and policy updates, treating expert demonstrations as preferred and model-generated responses as non-preferred, assuming expert data are high quality.The article contrasts this with RLHF’s separate stages and pairwise preferences; it also notes that IRL may ease distribution shift at the cost of memory and online generations.Source: §5.1 · pp. 20–21
5.2 Jointly learning policy and reward from Integrated Human Feedback
- AIHF is presented as a hybrid of demonstration- and preference-based methods, alongside a comparison of algorithms by data sources and optimization requirements.The comparison distinguishes PPO-style value-function requirements from direct preference methods that generally avoid critic estimation.Source: §5.2 · p. 22, Table 3 · p. 23
6 Dataset, Benchmarks, and Evaluation
The section introduces alignment datasets and evaluation benchmarks, then reports comparisons of demonstration-only and hybrid methods. In the reported experiments, IRL-based methods outperform cited baselines in the demonstration setting, while AIHF’s reward model performs better when trained with both data types.
6.1 Dataset
- Alignment datasets include pairwise preference annotations and single-response demonstrations, with examples such as HH-RLHF, UltraFeedback, TL;DR, and UltraChat.The section also describes self-generated responses that can be labeled and used for iterative policy updates.Source: §6.1 · pp. 23–24
6.2 Benchmark datasets and benchmark models
- The section surveys benchmarks for model and reward evaluation, including AlpacaEval, Open LLM Leaderboard, MT-Bench, HELM, reasoning benchmarks, and RewardBench.RewardBench evaluates reward models across chat, instruction-following, coding, and safety categories.Source: §6.2 · pp. 24–25
6.3 Numerical Results
- The numerical evaluation compares demonstration-based methods and methods using preference data, including hybrid approaches.The demonstration-only experiment uses UltraChat and synthetic preference pairs, treating demonstrations as preferred and policy-generated responses as rejected.Source: §6.3 · p. 25, §6.3.1 · p. 25, §6.3.2 · pp. 26–27
6.3.1 Demonstration-only setting
- The IRL-based method consistently outperforms the Mistral-7B-SFT-Beta checkpoint and SPIN on policy evaluations, including variants with one or two self-generation rounds.The evaluation uses Open LLM Leaderboard and MT-Bench; Table 4 reports MT-Bench policy-model evaluation.Source: Figure 6 · p. 25, §6.3.1 · p. 26, Table 4 · p. 26
6.3.2 Aligning with Both Demonstrations and Pairwise Comparisons
- AIHF’s reward model, trained with demonstrations and preferences, achieves better RewardBench performance than the compared reward models.The evaluation compares reward models from DPO, standard preference learning, and AIHF; Table 5 covers RewardBench.Source: §6.3.2 · p. 27, Table 5 · p. 27
7 Connection of Alignment to Signal Processing
The section frames alignment as inference from indirect, noisy, and incomplete measurements of human intent, connecting it to signal-processing methods. It highlights latent-variable inference and regularization as relevant perspectives, while also pointing to multimodal alignment.
7 Connection of Alignment to Signal Processing
- Alignment can be viewed as an inverse problem: human reward and model reasoning are latent structures inferred from limited, noisy feedback.The section compares this with recovering a clean signal from degraded observations.Source: §7 · p. 27
- A cited reasoning-alignment approach uses an EM-like two-stage procedure, inferring latent rationales and then updating model parameters.Its E-step estimates reasoning signals; its M-step updates parameters given those inferred signals.Source: §7 · p. 28
- Regularization methods such as Tikhonov penalties, total-variation priors, and sparsity constraints may stabilize ill-posed IRL estimation, especially with sparse or inconsistent feedback.Source: §7 · p. 28
- The section also identifies multimodal signal fusion, feature extraction, and cross-modal learning as relevant to alignment across text, images, speech, and video.Source: §7 · p. 28
8 Concluding Remarks
The article provides a tutorial on mathematical foundations and state-of-the-art algorithms for aligning LLMs with human feedback, while noting that significant challenges remain. These include scalability and effectiveness concerns in reward-model learning and RL-based fine-tuning.
8 Concluding Remarks
- The tutorial covers mathematical foundations and state-of-the-art algorithmic approaches to aligning LLMs with human feedback.The article reports significant progress in aligning LLMs with human values and preferences, while noting that open challenges remain.Source: §8 · p. 28
- Reward-model policies require extensive human annotation and computation, while PPO demands careful hyperparameter tuning and substantial compute resources.These demands constrain scalability and effectiveness.Source: §8 · p. 28
- Reward hacking remains a significant problem because models may optimize reward scores through unintended shortcuts without genuinely improving alignment with human intent.The article points to more efficient RL algorithms, robust and diverse reward functions, and hybrid supervised-learning/RL approaches as areas to address these concerns.Source: §8 · p. 28
Related papers
This paper cites
- Attention Is All You Need2017
- Training language models to follow instructions with human feedback2022
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning2025
- A General Theoretical Paradigm to Understand Learning from Human Preferences2023
- Self-Play Fine-Tuning Converts Weak Language Models to Strong Language Models2024
- Dota 2 with Large Scale Deep Reinforcement Learning2019