Statistical Complexity of Soft Bellman Residual Minimization

원문 초록

Offline reinforcement learning and offline inverse reinforcement learning aim to recover near-optimal value functions or reward models from a fixed batch of logged trajectories, yet current practice still struggles to enforce Bellman consistency scalably. Bellman residual minimization (BRM) has emerged as an attractive remedy. While globally convergent gradient-based soft (i.e., entropy-regularized) BRM methods for neural networks have recently been established, their statistical complexity under stochastic gradient descent remains largely unknown. In this paper, we address this theoretical gap for soft BRM. Through a Lyapunov-based analysis under a uniform PL condition in the auxiliary variable, we establish an $\mathcal{O}(1 / n)$ primal argument stability bound up to optimization error and an $\mathcal{O}(1 / n)$ statistical complexity for the soft BRM objective under the stated sampling conditions.