Rectifying Regression in Reinforcement Learning
On the map: Deep reinforcement learning methods → RL theory and sample complexity
Abstract (arXiv)
This paper investigates the impact of the loss function in value-based methods for reinforcement learning through an analysis of underlying prediction objectives. We theoretically show that mean absolute error is a better prediction objective than the traditional mean squared error for controlling the learned policy's suboptimality gap. Furthermore, we present results that different loss functions are better aligned with these different regression objectives: binary and categorical cross-entropy losses with the mean absolute error and squared loss with the mean squared error. We then provide empirical evidence that algorithms minimizing these cross-entropy losses can outperform those based on the squared loss in linear reinforcement learning.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in RL theory and sample complexity, one of 15 directions in Deep reinforcement learning methods. Builds on 15 of its 31 references that are in the map. First posted on arXiv 1 Oct 2025.
Builds on
15 of 31 references.
Closest ancestors · 1
Same direction, newest first
Other references · 14
- 2017A Distributional Perspective on Reinforcement Learningcited by 765
- 2019Provably Efficient Reinforcement Learning with Linear Function Approximationcited by 520
- 2017Distributional Reinforcement Learning with Quantile Regressioncited by 474
- 2019Information-Theoretic Considerations in Batch Reinforcement Learningcited by 287
- 2020Model-Based Reinforcement Learning with Value-Targeted Regressioncited by 234
- 2021Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient Algorithmscited by 223
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RLcited by 144
- 2018Improving Regression Performance with Distributional Lossescited by 67
- 2021Efficient First-Order Contextual Bandits: Prediction, Allocation, and Triangular Discriminationcited by 53
- 2019A Comparative Analysis of Expected and Distributional Reinforcement Learningcited by 47
- 2017Generalized Self-Concordant Functions: A Recipe for Newton-Type Methodscited by 33
- 2016Fast Rates for General Unbounded Loss Functions: from ERM to Generalized Bayescited by 27
2 more
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Eluder dimension: localise it!cited by 3
- 2026Eluder dimension: localise it!14 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2024The Central Role of the Loss Function in Reinforcement Learning
- 2024A Note on Loss Functions and Error Compounding in Model-based Reinforcement Learning
- 2025Calibrated Value-Aware Model Learning with Probabilistic Environment Models
- 2024A Multi-step Loss Function for Robust Learning of the Dynamics in Model-based Reinforcement Learning
- 2024On the Limited Representational Power of Value Functions and its Links to Statistical (In)Efficiency
- 2024The Perils of Optimizing Learned Reward Functions: Low Training Error Does Not Guarantee Low Regret
- 2022Beyond the Policy Gradient Theorem for Efficient Policy Updates in Actor-Critic Algorithms
- 2025Balancing Interpretability and Performance in Reinforcement Learning: An Adaptive Spectral Based Linear Approach