Reinforcement Learning with Non-Cumulative Objective
Wei Cui, Wei Yu
TL;DR
The paper addresses the limitation that many systems, especially in networking, have non-cumulative objectives that are not naturally expressed as reward sums. It introduces generalized Bellman updates, replacing the summation with a function $g(\cdot,\cdot)$ to optimize objectives like bottleneck, maximum, and harmonic mean, and proves convergence under clear conditions. Through theoretical results and multiple experiments (CartPole, Breakout, single-path maximum-flow routing, and wireless ad hoc routing), it demonstrates that generalized TD-style learning can achieve competitive or superior performance while enabling new problem formulations. The approach enables efficient learning with non-cumulative objectives in both single-agent and multi-agent settings, with practical impact on routing, spectrum access, and other control tasks where bottleneck or min-max criteria are central.
Abstract
In reinforcement learning, the objective is almost always defined as a \emph{cumulative} function over the rewards along the process. However, there are many optimal control and reinforcement learning problems in various application fields, especially in communications and networking, where the objectives are not naturally expressed as summations of the rewards. In this paper, we recognize the prevalence of non-cumulative objectives in various problems, and propose a modification to existing algorithms for optimizing such objectives. Specifically, we dive into the fundamental building block for many optimal control and reinforcement learning algorithms: the Bellman optimality equation. To optimize a non-cumulative objective, we replace the original summation operation in the Bellman update rule with a generalized operation corresponding to the objective. Furthermore, we provide sufficient conditions on the form of the generalized operation as well as assumptions on the Markov decision process under which the globally optimal convergence of the generalized Bellman updates can be guaranteed. We demonstrate the idea experimentally with the bottleneck objective, i.e., the objectives determined by the minimum reward along the process, on classical optimal control and reinforcement learning tasks, as well as on two network routing problems on maximizing the flow rates.
