Online Episodic Convex Reinforcement Learning
Bianca Marin Moreno, Khaled Eldowa, Pierre Gaillard, Margaux Brégère, Nadia Oudjane
TL;DR
The paper tackles online CURL in episodic finite-horizon MDPs where the loss at each episode is a convex function of the policy-induced state-action occupancy. It introduces Bonus O-MD-CURL, an online mirror-descent algorithm that incorporates carefully designed exploration bonuses to cope with unknown dynamics, achieving sublinear regret in the full-information setting and extending to bandit feedback settings via gradient estimation. For full-information CURL with unknown p, the authors prove a near-optimal regret of $ ilde{O}(L N^3 |𝒳|^{3/2} \,√{|𝒜| T})$, using a varying-constraint MD framework and a closed-form update. They also develop two bandit-CURL approaches: (i) an entropic-regularization method providing a $ ilde{O}$(√{L(L+1)/ε} |𝒳|^{5/4} |𝒜|^{5/4} N^3 T^{3/4}) + … bound under a state-distribution assumption, and (ii) a self-concordant, barrier-based method for known MDPs achieving $ ilde{O}(√{L} N^{7/4} (|𝒳||𝒜| T)^{3/4})$. The work also includes RL-specific bandit results and empirical validation showing exploration bonuses materially improve learning in multi-objective and constrained tasks. Overall, the paper provides a theoretically grounded, practical framework for online CURL under unknown dynamics and bandit feedback, bridging convex optimization techniques with reinforcement learning in a principled way.
Abstract
We study online learning in episodic finite-horizon Markov decision processes (MDPs) with convex objective functions, known as the concave utility reinforcement learning (CURL) problem. This setting generalizes RL from linear to convex losses on the state-action distribution induced by the agent's policy. The non-linearity of CURL invalidates classical Bellman equations and requires new algorithmic approaches. We introduce the first algorithm achieving near-optimal regret bounds for online CURL without any prior knowledge on the transition function. To achieve this, we use an online mirror descent algorithm with varying constraint sets and a carefully designed exploration bonus. We then address for the first time a bandit version of CURL, where the only feedback is the value of the objective function on the state-action distribution induced by the agent's policy. We achieve a sub-linear regret bound for this more challenging problem by adapting techniques from bandit convex optimization to the MDP setting.
