Masking Frequent Tokens Sharpens Direct Preference Optimization
cs.CL, cs.LG
Submitted: 2026-09-26
Updated: 2026-09-26
License: http://creativecommons.org/licenses/by/4.0/
The gist: Direct Preference Optimization (DPO) aligns language models by optimizing over sequence-level sums of token-wise implicit reward differences.
Terminology
Abstract
Direct Preference Optimization (DPO) aligns language models by optimizing over sequence-level sums of token-wise implicit reward differences. However, we identify a pervasive pathology in this formulation: a disproportionately small subset of high-frequency token types dominates cumulative sequence scores while appearing symmetrically across both preferred and dispreferred responses. Specifically, under canonical Qwen tokenization on Anthropic HH-RLHF, merely 69 token types account for 55.1% of all response tokens and 85.9% of within-pair shared token mass, exhibiting substantially lower preference-side specificity than the remaining vocabulary. This symmetric ubiquity induces gradient entanglement and dilutes the discriminative preference signal propagated through the objective. To resolve this issue, we introduce Anisotropic DPO and its canonical realization, Frequency-Hard DPO. Using a fixed, label-agnostic vocabulary mask, our method zeroes the implicit reward contribution of high-frequency response tokens while assigning unit weight to informative positions, thereby suppressing gradient interference without modifying preference pairs, discarding context, or introducing learned parameters. Here, anisotropy designates non-uniform token-level objective weighting rather than representational geometry. Extensive empirical evaluations on AlpacaEval, MT-Bench, and Arena-Hard demonstrate that Frequency-Hard DPO consistently outperforms standard DPO across Qwen-2.5-7B-Instruct and Llama-3-8B-Instruct, establishing that selectively masking shared high-frequency tokens offers an effective, zero-overhead mechanism for robust preference alignment.
Sources
- Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback
- SparsePO: Controlling Preference Alignment of LLMs via Sparse Token Masks
- UltraFeedback: Boosting Language Models with Scaled AI Feedback
- Length-Controlled AlpacaEval: A Simple Way to Debias Automatic Evaluators
- Representation Degeneration Problem in Training Natural Language Generation Models
- AlphaPO: Reward Shape Matters for LLM Alignment
- Normalized Rewards for Preference Optimization
- Smaug: Fixing Failure Modes of Preference Optimisation with DPO-Positive
- ConfPO: Exploiting Policy Model Confidence for Critical Token Selection in Preference Optimization
- Token-level Direct Preference Optimization
- SLiC-HF: Sequence Likelihood Calibration with Human Feedback
- TGDPO: Harnessing Token-Level Reward Guidance for Enhancing Direct Preference Optimization
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering