Revisiting Bellman Errors for Offline Model Selection

Joshua P Zitovsky, Rishabh Agarwal, Daniel de Marchi, Michael R Kosorok

Keywords: reinforcement learning, offline reinforcement learning, deep reinforcement learning, off-policy evaluation, model selection, Atari

TL;DR: We demonstrate that Bellman errors can be very useful for hyperparameter tuning in deep offline reinforcement learning, contrary to popular belief and previous literature.

Abstract: Applying offline reinforcement learning in real-world settings necessitates the ability to tune hyperparameters offline, a task known as $\textit{offline model selection}$. It is well-known that the empirical Bellman errors are poor predictors of value function estimation accuracy and policy performance. This has led researchers to abandon model selection procedures based on Bellman errors and instead focus on evaluating the expected return under policies of interest. The problem with this approach is that it can be very difficult to use an offline dataset generated by one policy to estimate the expected returns of a different policy. In contrast, we argue that Bellman errors can be useful for offline model selection, and that the discouraging results in past literature has been due to estimating and utilizing them incorrectly. We propose a new algorithm, $\textit{Supervised Bellman Validation}$, that estimates the expected squared Bellman error better than the empirical Bellman errors. We demonstrate the relative merits of our method over competing methods through both theoretical results and empirical results on datasets from the Atari benchmark. We hope that our results will challenge current attitudes and spur future research into Bellman errors and their utility in offline model selection.

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/arxiv:2302.00141/code)