Averaging n-step Returns Reduces Variance in Reinforcement Learning
Brett Daley, Martha White, Marlos C. Machado
Abstract
Multistep returns, such as -step returns and -returns, are commonly used to improve the sample efficiency of reinforcement learning (RL) methods. The variance of the multistep returns becomes the limiting factor in their length; looking too far into the future increases variance and reverses the benefits of multistep learning. In our work, we demonstrate the ability of compound returns -- weighted averages of -step returns -- to reduce variance. We prove for the first time that any compound return with the same contraction modulus as a given -step return has strictly lower variance. We additionally prove that this variance-reduction property improves the finite-sample complexity of temporal-difference learning under linear function approximation. Because general compound returns can be expensive to implement, we introduce two-bootstrap returns which reduce variance while remaining efficient, even when using minibatched experience replay. We conduct experiments showing that compound returns often increase the sample efficiency of -step deep RL agents like DQN and PPO.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Chunking the Critic: A Transformer-based Soft Actor-Critic with N-Step ReturnsDong Tian, Onur Celik, Gerhard NeumannICLR 2026 · 18 citations
- Value-Guided Decision Transformer: A Unified Reinforcement Learning Framework for Online and Offline SettingsHongling Zheng, Li Shen, Yong Luo, Deheng Ye et al.NeurIPS 2025 · 6 citations
Builds on3
- Beyond Variance Reduction: Understanding the True Impact of Baselines on Policy OptimizationWesley Chung, Valentin Thomas, Marlos C. Machado, Nicolas Le RouxICML 2021 · 35 citations
- Revisiting Peng's Q(λ) for Modern Reinforcement LearningTadashi Kozuno, Yunhao Tang, Mark Rowland, Rémi Munos et al.ICML 2021 · 19 citations
- Trajectory-Aware Eligibility Traces for Off-Policy Reinforcement LearningBrett Daley, Martha White, Christopher Amato, Marlos C. MachadoICML 2023 · 4 citations
Related papers
- Learning Expected Emphatic Traces for Deep RLRay Jiang, Shangtong Zhang, Veronica Chelu, Adam White et al.AAAI 2022 · 14 citations
- Uncorrected Least-Squares Temporal Difference with Lambda-ReturnTakayuki OsogamiAAAI 2020 · 1 citation
- Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement LearningHaoxin Lin, Yu-Yan Xu, Yihao Sun, Zhilong Zhang et al.ICLR 2025
- A Generalized Bootstrap Target for Value-Learning, Efficiently Combining Value and Feature PredictionsAnthony GX-Chen, Veronica Chelu, Blake A. Richards, Joelle PineauAAAI 2022 · 1 citation
- Partially Observable Reinforcement Learning with Memory TracesOnno Eberhard, Michael Muehlebach, Claire VernadeICML 2025
