Breaking the Bias Barrier in Concave Multi-Objective Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
While standard reinforcement learning optimizes a single reward signal, many applications require optimizing a nonlinear utility f(J1π,dots,JMπ) over multiple objectives, where each Jmπ denotes the expected discounted return of a distinct reward function. A common approach is concave scalarization, which captures important trade-offs such as fairness and risk sensitivity. However, nonlinear scalarization introduces a fundamental challenge for policy gradient methods: the gradient depends on partial f(Jπ), while in practice only empirical return estimates hat J are available. Because f is nonlinear, the plug-in estimator is biased (mathbbE[partial f(hat J)] neq partial f(mathbbE[hat J])), leading to persistent gradient bias that degrades sample complexity. In this work we identify and overcome this bias barrier in concave-scalarized multi-objective reinforcement learning. We show that existing policy-gradient methods suffer an intrinsic widetildeO(ε-4) sample complexity due to this bias. To address this issue, we develop a Natural Policy Gradient (NPG) algorithm equipped with a multi-level Monte Carlo (MLMC) estimator that controls the bias of the scalarization gradient while maintaining low sampling cost. We prove that this approach achieves the optimal widetildeO(ε-2) sample complexity for computing an ε-optimal policy. Furthermore, we show that when the scalarization function is second-order smooth, the first-order bias cancels automatically, allowing vanilla NPG to achieve the same widetildeO(ε-2) rate without MLMC. Our results provide the first optimal sample complexity guarantees for concave multi-objective reinforcement learning under policy-gradient methods.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 21 of its 30 references that are in the map. First posted on arXiv 9 Mar 2026.
Builds on
21 of 30 references.
References · 21
- 2018Provably Efficient Maximum Entropy Explorationcited by 240
- 2019On the Theory of Policy Gradient Methods: Optimality, Approximation, and Distribution Shiftcited by 174
- 2020Variational Policy Gradient Method for Reinforcement Learning with General Utilitiescited by 108
- 2021Reward is enough for convex MDPscited by 82
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 67
- 2022An Improved Analysis of (Variance-Reduced) Policy Gradient and Natural Policy Gradient Methodscited by 63
- 2021Concave Utility Reinforcement Learning: the Mean-Field Game Viewpointcited by 54
- 2023Stochastic Policy Gradient Methods: Improved Sample Complexity for Fisher-non-degenerate Policiescited by 53
- 2024Fair Resource Allocation in Multi-Task Learningcited by 34
- 2023Improved Sample Complexity Analysis of Natural Policy Gradient Algorithm with General Parameterization for Infinite Horizon Discounted Reward Markov Decision Processescited by 24
- 2021Concave Utility Reinforcement Learning with Zero-Constraint Violationscited by 18
- 2023Beyond Exponentially Fast Mixing in Average-Reward Reinforcement Learning via Multi-Level Monte Carlo Actor-Criticcited by 18
9 more
- 2019Reinforcement Learning for Joint Optimization of Multiple Rewardscited by 15
- 2022Anchor-Changing Regularized Natural Policy Gradient for Multi-Objective Reinforcement Learningcited by 15
- 2021Joint Optimization of Multi-Objective Reinforcement Learning with Policy Gradient Based Algorithmcited by 12
- 2024Order-Optimal Regret with Novel Policy Gradient Approaches in Infinite-Horizon Average Reward MDPscited by 11
- 2020L2B: Learning to Balance the Safety-Efficiency Trade-off in Interactive Crowd-aware Robot Navigationcited by 7
- 2023Multi-objective Reinforcement Learning with Nonlinear Preferences: Provable Approximation for Maximizing Expected Scalarized Returncited by 7
- 2024On the Global Optimality of Policy Gradient Methods in General Utility Reinforcement Learningcited by 6
- 2025Regret Analysis of Average-Reward Unichain MDPs via an Actor-Critic Approachcited by 6
- 2019Optimal Server Selection for Straggler Mitigationcited by 2
Built on by
Citing papers per year: 2026: 2.
2 papers.