Convex-Concave Reinforcement Learning

arXiv:2610.09108 · cs.LG, cs.AI · Submitted 2026-10-06 · Read on arXiv

cs.LG, cs.AI

Submitted: 2026-10-06

Updated: 2026-10-06

Comments: 37 pages, 5 figures. Code: https://github.com/UMass-SCALAR-Lab/Convex-Concave-RL

Code: https://github.com/UMass-SCALAR-Lab/Convex-Concave-RL

Project page: https://rltheorybook.github.io

License: http://creativecommons.org/licenses/by/4.0/

The gist: Policy learning drives many of the most consequential and heavily-invested applications of reinforcement learning today.

Terminology

Abstract

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:= [π/π 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 k 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.

Sources

Related papers