Learning Latent Reasoning Traces for Scalar Reward Models End-to-End
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Reward models (RMs) are central to aligning large language models with human preferences via reinforcement learning. Although traditional scalar RMs enable efficient and probabilistic reward modeling, they rely on superficial cues that fail to generalize to complex or out-of-distribution (OOD) tasks. Conversely, generative RMs leverage extensive reasoning to improve robustness on challenging tasks, but their natural language-based scores lack the numerical flexibility and probabilistic interpretability that scalar RMs offer. While recent approaches combine both paradigms through off-policy multi-task learning, such parallel optimization does not guarantee that generated reasoning traces actively align with or benefit downstream scalar reward prediction. To address this mismatch, we propose LatentRM, a reward modeling framework that learns intermediate reasoning traces as discrete latent variables to explicitly maximize the likelihood of downstream scalar rewards. Through on-policy optimization of the latent reasoning space end-to-end, LatentRM tightly couples deep reasoning-based evaluation with precise scoring. Extensive validations on in-distribution and OOD datasets and RLHF show that LatentRM outperforms scalar, generative, and hybrid RMs on preference modeling and policy alignment across tasks ranging from open-ended conversation to complex reasoning.
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 10 of its 43 references that are in the map. First posted on arXiv 31 Jul 2026.
Builds on
10 of 43 references.
Closest ancestors · 3
Same direction, newest first
- 2025Inference-Time Scaling for Generalist Reward Modeling3 Apr
- 2024Skywork-Reward: Bag of Tricks for Reward Modeling in LLMs24 Oct
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 7
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,634
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
- 2025AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning datasetcited by 165
- 2024Critique-out-Loud Reward Modelscited by 105
- 2025PaTaRM: Bridging Pairwise and Pointwise Signals via Preference-Aware Task-Adaptive Reward Modelingcited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025RM-R1: Reward Modeling as Reasoning
- 2025Think-RM: Enabling Long-Horizon Reasoning in Generative Reward Models
- 2025GRAM-R2: Self-Training Generative Foundation Reward Models for Reward Reasoning
- 2026UniRRM: Unified Reasoning Reward Models Across Languages and Evaluation Paradigms
- 2025Reward Reasoning Model
- 2025ReasonGRM: Enhancing Generative Reward Models through Large Reasoning Models
- 2026RewardExplainer: Learning Reward Model Explanations from Counterfactual Preference Feedback
- 2025Enhancing Large Language Model Reasoning with Reward Models: An Analytical Survey