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

Jun 23, 2026

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

Many reinforcement learning (RL) problems in the infinite-horizon average-reward setting require optimizing multiple conflicting objectives while satisfying multiple safety constraints. A common approach is concave scalarization, where the agent maximizes a utility f(Jr1π,…,JrMπ)f(J^π_{r_1}, \ldots, J^π_{r_M}) subject to a scalarized constraint g(Jc1π,…,JcNπ)≥0g(J^π_{c_1}, \ldots, J^π_{c_N}) \ge 0, where JrmπJ^π_{r_m} and JcnπJ^π_{c_n} denote the average-reward and cost under policy ππ. However, the nonlinearity of ff and gg introduces bias in policy-gradient and actor-critic methods, since gradients must be evaluated using noisy estimates of Jπ,J^π, and E[∂f(Jπ)]≠∂f(E[Jπ]), \mathbb{E}[\partial f(J^π)] \neq \partial f(\mathbb{E}[J^π]), and this bias propagates through both primal and dual updates. We propose an MLMC-based primal-dual Natural Actor-Critic algorithm for average-reward MDPs that controls bias in scalarized objectives, constraint evaluation, and actor-critic estimation without requiring mixing-time knowledge. We show that the algorithm achieves optimal global convergence and constraint-violation rates of O~(1/T)\tilde{O}(1/\sqrt{T}). To our knowledge, this is the first result establishing optimal convergence for concave scalarized multi-objective RL in the average-reward setting, both with and without constraints, and the first to do so without mixing-time information even in the absence of scalarization.
Oct 6, 2026

Convex-Concave Reinforcement Learning

Policy learning drives many of the most consequential and heavily-invested applications of reinforcement learning today. Yet the core optimization problem it rests on (maximizing expected return) is notoriously non-convex, even under a direct policy parameterization, and the field has largely responded by avoiding it: optimizing convex surrogate approximations of the return under trust-region constraints (NPG, TRPO, PPO, AWR). We show that this seemingly unstructured problem is not actually structureless. In log-density-ratio coordinates y:=log⁡[π/πn]y := \log[π/π_n], the exact per-iteration objective, computable via per-decision importance sampling (PDIS), is a difference-of-convex-constrained difference-of-convex (DC-constrained DC) program. This structure lets us move beyond surrogate approximations: it recovers CPI, NPG, TRPO, and AWR as special cases along interpretable axes, and it opens a multi-step axis kk that couples consecutive decisions. We solve the per-iteration program with sequential convex programming (SCP), the standard solver for difference-of-convex problems, and give convergence guarantees under mild conditions, bridging the difference-of-convex optimization and RL literatures. Empirically, multi-step Convex-Concave RL (CCRL) wins on diagnostic MDPs where credit must propagate across a horizon (its advantage growing with the dependency length), is competitive with a tuned PPO on classic control, and on a realistic, stochastic, mid-horizon healthcare domain converges markedly faster than tuned PPO to the same near-optimal survival, with an 11.3% higher area under the training curve.
Jul 30, 2026

Hypergradient-based Bilevel Reinforcement Learning with Improved Sample Complexity

Bilevel reinforcement learning (RL) is an important framework within the literature of RL that can be used to formalize various categories of problems, such as meta-learning, hierarchical task decomposition, and reinforcement learning from human feedback (RL-HF). Most of the bilevel RL algorithms are either not scalable because of using hypergradient with Hessian, or they suffer from high sample complexity because of using penalty-based approximation methods. In this work, we propose a hypergradient-based bilevel RL algorithm using the optimality of the Boltzmann policy for the entropy regularized discounted RL objective function. Our proposed algorithm is Hessian-free and obtains an iteration complexity of O(ε−1)O(ε^{-1}) and state-of-the-art sample complexity of O~(ε−2)\tilde{O}(ε^{-2}) under mild regularity conditions. Further, in our convergence analysis, we are able to remove the assumption of the Polyak-Lojasiewicz (PL) condition on the outer-level objective function present in the prior state-of-the-art sample complexity work.