The Optimal Reward Baseline for Gradient-Based Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
There exist a number of reinforcement learning algorithms which learnby climbing the gradient of expected reward. Their long-runconvergence has been proved, even in partially observableenvironments with non-deterministic actions, and without the need fora system model. However, the variance of the gradient estimator hasbeen found to be a significant practical problem. Recent approacheshave discounted future rewards, introducing a bias-variance trade-offinto the gradient estimate. We incorporate a reward baseline into thelearning system, and show that it affects variance without introducingfurther bias. In particular, as we approach the zero-bias,high-variance parameterization, the optimal (or variance minimizing)constant reward baseline is equal to the long-term average expectedreward. Modified policy-gradient algorithms are presented, and anumber of experiments demonstrate their improvement over previous work.
Key facts. Cited by 84 papers in the map (data through 7 Oct 2026). First posted on arXiv 10 Jan 2013.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 14; 2022: 18; 2023: 15; 2024: 9; 2025: 12; 2026: 15.
84 papers.
Sort:
By area:
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 184
- 2022GFlowNets and variational inferencecited by 103
- 2022On Reinforcement Learning and Distribution Matching for Fine-Tuning Language Models with no Catastrophic Forgettingcited by 80
- 2021Settling the Variance of Multi-Agent Policy Gradientscited by 45
- 2021Learning Algebraic Recombination for Compositional Generalizationcited by 31
- 2023Bridging Discrete and Backpropagation: Straight-Through and Beyondcited by 31
- 2025On-Policy RL with Optimal Reward Baselinecited by 30
- 2024Improving Reinforcement Learning from Human Feedback Using Contrastive Rewardscited by 23
- 2022Gradient Estimation with Discrete Stein Operatorscited by 22
- 2020Average-reward model-free reinforcement learning: a systematic review and literature mappingcited by 19
- 2021Sequence-Level Training for Non-Autoregressive Neural Machine Translationcited by 18
- 2022Asynchronous Actor-Critic for Multi-Agent Reinforcement Learningcited by 16
38 more
- 2022Risk-Sensitive Reinforcement Learning with Exponential Criteriacited by 15
- 2024QuantFactor REINFORCE: Mining Steady Formulaic Alpha Factors with Variance-bounded REINFORCEcited by 13
- 2026DORA: A Scalable Asynchronous Reinforcement Learning System for Language Model Trainingcited by 13
- 2021A unified view of likelihood ratio and reparameterization gradientscited by 12
- 2025Formalizing Learning from Language Feedback with Provable Guaranteescited by 10
- 2023System-status-aware Adaptive Network for Online Streaming Video Understandingcited by 9
- 2023IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARLcited by 9
- 2023Simultaneous Machine Translation with Tailored Referencecited by 9
- 2024Policy Gradient with Active Importance Samplingcited by 9
- 2026When Right Meets Wrong: Bilateral Context Conditioning with Reward-Confidence Correction for GRPOcited by 9
- 2021Factored Policy Gradients: Leveraging Structure for Efficient Learning in MOMDPscited by 8
- 2021Double Control Variates for Gradient Estimation in Discrete Latent Variable Modelscited by 8
- 2025Aligning Constraint Generation with Design Intent in Parametric CADcited by 8
- 2025ReDit: Reward Dithering for Improved LLM Policy Optimizationcited by 8
- 2022MAD for Robust Reinforcement Learning in Machine Translationcited by 7
- 2023Dynamic Feature-based Deep Reinforcement Learning for Flow Control of Circular Cylinder with Sparse Surface Pressure Sensingcited by 7
- 2026Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMscited by 7
- 2021Policy Gradients Incorporating the Futurecited by 6
- 2022DeepCreativity: Measuring Creativity with Deep Learning Techniquescited by 6
- 2024On Divergence Measures for Training GFlowNetscited by 6
- 2021Off-Policy Actor-Critic with Emphatic Weightingscited by 5
- 2023GFN-SR: Symbolic Regression with Generative Flow Networkscited by 5
- 2023An Invitation to Deep Reinforcement Learningcited by 5
- 2024Fisher-Rao Gradient Flows of Linear Programs and State-Action Natural Policy Gradientscited by 5
- 2026The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RLcited by 5
- 2021End-to-End Learning of Deep Kernel Acquisition Functions for Bayesian Optimizationcited by 4
- 2022Variance Reduction for Policy-Gradient Methods via Empirical Variance Minimizationcited by 4
- 2021Diversity-Driven Combination for Grammatical Error Correctioncited by 3
- 2021Towards Learning to Speak and Hear Through Multi-Agent Communication over a Continuous Acoustic Channelcited by 3
- 2022An Unpooling Layer for Graph Generationcited by 3
- 2022Cooperation and Competition: Flocking with Evolutionary Multi-Agent Reinforcement Learningcited by 3
- 2022Rephrasing the Reference for Non-Autoregressive Machine Translationcited by 3
- 2023Policy Gradient with Tree Expansioncited by 3
- 2023Sample Dropout: A Simple yet Effective Variance Reduction Technique in Deep Policy Optimizationcited by 3
- 2024Differentiable Discrete Event Simulation for Queuing Network Controlcited by 3
- 2022Interpretable and Effective Reinforcement Learning for Attacking against Graph-based Rumor Detectioncited by 2
- 2022Policy Learning and Evaluation with Randomized Quasi-Monte Carlocited by 2
- 2022Data-driven End-to-end Learning of Pole Placement Control for Nonlinear Dynamics via Koopman Invariant Subspacescited by 2
50 most cited of 84 All 84 (JSON)
- 2026SAGE: Mitigating Long-Horizon Reasoning Biases via Topological Guidance24 Sep 2026
- 2026Shared Actors Need Not Share Critics: Effects of Value Mismatch in Parallel Reinforcement Learning27 Aug 2026
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Models15 Jul 2026
- 2026On the Policy Gradient Foundations of Group Relative Policy Optimization: Credit Assignment, Gradient Sparsity, and Rank Collapse28 Jun 2026
- 2026Space-sampled Value Decay: Forgetting Mechanisms for Non-stationary Reinforcement Learning10 Jun 2026
- 2026On Advantage Estimates for Max@K Policy Gradients4 Jun 2026
- 2026Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal States8 May 2026
- 2026Bayesian policy gradient and actor-critic algorithms30 Apr 2026
- 2026DORA: A Scalable Asynchronous Reinforcement Learning System for Language Model Training29 Apr 2026
- 2026When Right Meets Wrong: Bilateral Context Conditioning with Reward-Confidence Correction for GRPO13 Mar 2026
- 2026Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMs19 Feb 2026
- 2026The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL6 Feb 2026
38 more
- 2026Explicit Credit Assignment through Local Rewards and Dependence Graphs in Multi-Agent Reinforcement Learning29 Jan 2026
- 2026Draw it like Euclid: Teaching transformer models to generate CAD profiles using ruler and compass construction steps14 Jan 2026
- 2026Reinforcement Learning of Large Language Models for Interpretable Credit Card Fraud Detection9 Jan 2026
- 2025GRADE: Replacing Policy Gradients with Backpropagation for LLM Alignment30 Dec 2025
- 2025Yahtzee: Reinforcement Learning Techniques for Stochastic Combinatorial Games18 Dec 2025
- 2025Fluent Alignment with Disfluent Judges: Post-training for Lower-resource Languages9 Dec 2025
- 2025EGG-SR: Embedding Symbolic Equivalence into Symbolic Regression via Equality Graph8 Nov 2025
- 2025Malliavin Calculus as Stochastic Backpropogation2 Nov 2025
- 2025Multi-level Advantage Credit Assignment for Cooperative Multi-Agent Reinforcement Learning9 Aug 2025
- 2025ReDit: Reward Dithering for Improved LLM Policy Optimization23 Jun 2025
- 2025Formalizing Learning from Language Feedback with Provable Guarantees12 Jun 2025
- 2025On-Policy RL with Optimal Reward Baseline29 May 2025
- 2025Aligning Constraint Generation with Design Intent in Parametric CAD17 Apr 2025
- 2025A Multi-Fidelity Control Variate Approach for Policy Gradient Estimation7 Mar 2025
- 2025Reducing Variance Caused by Communication in Decentralized Multi-agent Deep Reinforcement Learning10 Feb 2025
- 2024On Divergence Measures for Training GFlowNets12 Oct 2024
- 2024QuantFactor REINFORCE: Mining Steady Formulaic Alpha Factors with Variance-bounded REINFORCE8 Sep 2024
- 2024Differentiable Discrete Event Simulation for Queuing Network Control5 Sep 2024
- 2024KEHRL: Learning Knowledge-Enhanced Language Representations with Hierarchical Reinforcement Learning24 Jun 2024
- 2024Policy Gradient with Active Importance Sampling9 May 2024
- 2024R4: Reinforced Retriever-Reorder-Responder for Retrieval-Augmented Large Language Models4 May 2024
- 2024Centralized vs. Decentralized Multi-Agent Reinforcement Learning for Enhanced Control of Electric Vehicle Charging Networks18 Apr 2024
- 2024Fisher-Rao Gradient Flows of Linear Programs and State-Action Natural Policy Gradients28 Mar 2024
- 2024Improving Reinforcement Learning from Human Feedback Using Contrastive Rewards12 Mar 2024
- 2023An Invitation to Deep Reinforcement Learning13 Dec 2023
- 2023GFN-SR: Symbolic Regression with Generative Flow Networks1 Dec 2023
- 2023Simultaneous Machine Translation with Tailored Reference20 Oct 2023
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Models16 Oct 2023
- 2023Dynamic Feature-based Deep Reinforcement Learning for Flow Control of Circular Cylinder with Sparse Surface Pressure Sensing5 Jul 2023
- 2023IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL20 Jun 2023
- 2023DoMo-AC: Doubly Multi-step Off-policy Actor-Critic Algorithm29 May 2023
- 2023GiPH: Generalizable Placement Learning for Adaptive Heterogeneous Computing23 May 2023
- 2023Bridging Discrete and Backpropagation: Straight-Through and Beyond17 Apr 2023
- 2023System-status-aware Adaptive Network for Online Streaming Video Understanding28 Mar 2023
- 2023AR3n: A Reinforcement Learning-based Assist-As-Needed Controller for Robotic Rehabilitation28 Feb 2023
- 2023Revisiting the Gumbel-Softmax in MADDPG23 Feb 2023
- 2023Sample Dropout: A Simple yet Effective Variance Reduction Technique in Deep Policy Optimization5 Feb 2023
- 2023Distillation Policy Optimization1 Feb 2023
50 newest of 84 All 84 (JSON)
- 2025On-Policy RL with Optimal Reward Baselinecited by 30
- 2026When Right Meets Wrong: Bilateral Context Conditioning with Reward-Confidence Correction for GRPOcited by 9
- 2025ReDit: Reward Dithering for Improved LLM Policy Optimizationcited by 8
- 2026Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal Statescited by 2
- 2026On Advantage Estimates for Max@K Policy Gradientscited by 2
- 2026On the Policy Gradient Foundations of Group Relative Policy Optimization: Credit Assignment, Gradient Sparsity, and Rank Collapsecited by 2
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Modelscited by 1
- 2026SAGE: Mitigating Long-Horizon Reasoning Biases via Topological Guidancecited by 0
- 2026SAGE: Mitigating Long-Horizon Reasoning Biases via Topological Guidance24 Sep 2026
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Models15 Jul 2026
- 2026On the Policy Gradient Foundations of Group Relative Policy Optimization: Credit Assignment, Gradient Sparsity, and Rank Collapse28 Jun 2026
- 2026On Advantage Estimates for Max@K Policy Gradients4 Jun 2026
- 2026Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal States8 May 2026
- 2026When Right Meets Wrong: Bilateral Context Conditioning with Reward-Confidence Correction for GRPO13 Mar 2026
- 2025ReDit: Reward Dithering for Improved LLM Policy Optimization23 Jun 2025
- 2025On-Policy RL with Optimal Reward Baseline29 May 2025
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 184
- 2024Improving Reinforcement Learning from Human Feedback Using Contrastive Rewardscited by 23
- 2020Average-reward model-free reinforcement learning: a systematic review and literature mappingcited by 19
- 2025Fluent Alignment with Disfluent Judges: Post-training for Lower-resource Languagescited by 0
- 2025GRADE: Replacing Policy Gradients with Backpropagation for LLM Alignmentcited by 0
- 2025GRADE: Replacing Policy Gradients with Backpropagation for LLM Alignment30 Dec 2025
- 2025Fluent Alignment with Disfluent Judges: Post-training for Lower-resource Languages9 Dec 2025
- 2024Improving Reinforcement Learning from Human Feedback Using Contrastive Rewards12 Mar 2024
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Models16 Oct 2023
- 2020Average-reward model-free reinforcement learning: a systematic review and literature mapping18 Oct 2020
- 2022GFlowNets and variational inferencecited by 103
- 2022On Reinforcement Learning and Distribution Matching for Fine-Tuning Language Models with no Catastrophic Forgettingcited by 80
- 2021Settling the Variance of Multi-Agent Policy Gradientscited by 45
- 2021Learning Algebraic Recombination for Compositional Generalizationcited by 31
- 2023Bridging Discrete and Backpropagation: Straight-Through and Beyondcited by 31
- 2022Gradient Estimation with Discrete Stein Operatorscited by 22
- 2021Sequence-Level Training for Non-Autoregressive Neural Machine Translationcited by 18
- 2022Asynchronous Actor-Critic for Multi-Agent Reinforcement Learningcited by 16
- 2022Risk-Sensitive Reinforcement Learning with Exponential Criteriacited by 15
- 2024QuantFactor REINFORCE: Mining Steady Formulaic Alpha Factors with Variance-bounded REINFORCEcited by 13
- 2026DORA: A Scalable Asynchronous Reinforcement Learning System for Language Model Trainingcited by 13
- 2021A unified view of likelihood ratio and reparameterization gradientscited by 12
12 most cited of 70 All 70 (JSON)
- 2026Shared Actors Need Not Share Critics: Effects of Value Mismatch in Parallel Reinforcement Learning27 Aug 2026
- 2026Space-sampled Value Decay: Forgetting Mechanisms for Non-stationary Reinforcement Learning10 Jun 2026
- 2026Bayesian policy gradient and actor-critic algorithms30 Apr 2026
- 2026DORA: A Scalable Asynchronous Reinforcement Learning System for Language Model Training29 Apr 2026
- 2026Stable Asynchrony: Variance-Controlled Off-Policy RL for LLMs19 Feb 2026
- 2026The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL6 Feb 2026
- 2026Explicit Credit Assignment through Local Rewards and Dependence Graphs in Multi-Agent Reinforcement Learning29 Jan 2026
- 2026Draw it like Euclid: Teaching transformer models to generate CAD profiles using ruler and compass construction steps14 Jan 2026
- 2026Reinforcement Learning of Large Language Models for Interpretable Credit Card Fraud Detection9 Jan 2026
- 2025Yahtzee: Reinforcement Learning Techniques for Stochastic Combinatorial Games18 Dec 2025
- 2025EGG-SR: Embedding Symbolic Equivalence into Symbolic Regression via Equality Graph8 Nov 2025
- 2025Malliavin Calculus as Stochastic Backpropogation2 Nov 2025
12 newest of 70 All 70 (JSON)