Not All Preferences Deserve Gradients: Understanding Gradient Utility in Offline Reasoning Alignment
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Offline preference optimization aligns reasoning models from fixed chosen--rejected pairs, yet standard methods apply gradient updates from every pair regardless of its training value under the current policy. We argue that this uniform treatment is wasteful and potentially harmful. From the perspective of gradient utility, we show that a pair's contribution depends jointly on informativeness and stability. Pair utility drifts as the policy evolves, high-gradient samples can coincide with high-curvature regions, leading to noisy and destabilizing updates, and the most effective supervision comes from stable confident errors where the model is reliably wrong yet curvature remains low. These findings motivate SAGE (Stability-Aware Gradient Efficiency), which maintains difficulty-stratified candidate pools refreshed during training and selects pairs within each pool by a forward-pass signal-to-curvature score. Only pairs with high current utility receive gradient computation; the rest are excluded from backpropagation. On mathematical reasoning benchmarks across multiple model scales, SAGE outperforms full-data and size-matched baselines while producing substantially smoother optimization trajectories.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 15 of its 21 references that are in the map. First posted on arXiv 1 Feb 2026.
Builds on
15 of 21 references.
Closest ancestors · 4
Same direction, newest first
- 2025Training Language Models to Reason Efficiently6 Feb
- 2024KTO: Model Alignment as Prospect Theoretic Optimization2 Feb
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 11
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2024Contrastive Preference Optimization: Pushing the Boundaries of LLM Performance in Machine Translationcited by 390
- 2024MathScale: Scaling Instruction Tuning for Mathematical Reasoningcited by 144
- 2024MATES: Model-Aware Data Selection for Efficient Pretraining with Data Influence Modelscited by 96
- 2024Noise Contrastive Alignment of Language Models with Explicit Rewardscited by 85
- 2024MARIO: MAth Reasoning with code Interpreter Output -- A Reproducible Pipelinecited by 81
- 2024When Search Engine Services meet Large Language Models: Visions and Challengescited by 59
- 2024JiuZhang3.0: Efficiently Improving Mathematical Reasoning by Training Small Data Synthesis Modelscited by 44
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025From Noisy Traces to Stable Gradients: Bias-Variance Optimized Preference Optimization for Aligning Large Reasoning Models
- 2026GRACE: Gradient-aligned Reasoning Data Curation for Efficient Post-training
- 2026Invariant Gradient Alignment for Robust Reasoning Distillation
- 2025DaGRPO: Rectifying Gradient Conflict in Reasoning via Distinctiveness-Aware Group Relative Policy Optimization
- 2026Gradients Know What Outcomes Don't: Unlocking Reinforcement Learning for LLM Reasoning with Gradient-Aligned Rewards
- 2026Mining or Synthesis? Rethinking Exploration Efficiency in Iterative Alignment of Mathematical Reasoning
- 2025Turning Drift into Constraint: Robust Reasoning Alignment in Non-Stationary Multi-Stream Environments
- 2026In LLM Reasoning, there is Irrationality on top of Value Misalignment