cs.LGMar 9, 2026

Breaking the Bias Barrier in Concave Multi-Objective Reinforcement Learning

Authors: Swetha Ganesh, Vaneet Aggarwal

Organizations: Purdue University, USA 47907

Abstract

While standard reinforcement learning optimizes a single reward signal, many applications require optimizing a nonlinear utility f(J1π,…,JMπ)f(J_1^π,\dots,J_M^π) over multiple objectives, where each JmπJ_m^π 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 ∂f(Jπ)\partial f(J^π), while in practice only empirical return estimates J^\hat J are available. Because ff is nonlinear, the plug-in estimator is biased (E[∂f(J^)]≠∂f(E[J^])\mathbb{E}[\partial f(\hat J)] \neq \partial f(\mathbb{E}[\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 O~(ε−4)\widetilde{\mathcal{O}}(ε^{-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 O~(ε−2)\widetilde{\mathcal{O}}(ε^{-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 O~(ε−2)\widetilde{\mathcal{O}}(ε^{-2}) rate without MLMC. Our results provide the first optimal sample complexity guarantees for concave multi-objective reinforcement learning under policy-gradient methods.

Figures & tables

Explore similar work

CardsList
  1. Bias-Controlled Primal-Dual Natural Actor-Critic: Optimal Rates for Constrained Multi-Objective Average-Reward RL

    Jun 23, 2026Ankur Naskar, Swetha Ganesh, Vaneet AggarwalMulti-Objective Reinforcement LearningReinforcement Learning

  2. Convex-Concave Reinforcement Learning

    Oct 6, 2026Shripad V. Deshmukh, Yaswanth Chittepu, Dhawal Gupta +2Reinforcement LearningPolicy Gradient Methods

  3. Hypergradient-based Bilevel Reinforcement Learning with Improved Sample Complexity

    Jul 30, 2026Naman Saxena, Mudit Gaur, Vaneet AggarwalReinforcement LearningHypergradient Optimization