Linear Bellman Completeness Suffices for Efficient Online Reinforcement Learning with Few Actions (2406.11640v2)

Abstract: One of the most natural approaches to reinforcement learning (RL) with function approximation is value iteration, which inductively generates approximations to the optimal value function by solving a sequence of regression problems. To ensure the success of value iteration, it is typically assumed that BeLLMan completeness holds, which ensures that these regression problems are well-specified. We study the problem of learning an optimal policy under BeLLMan completeness in the online model of RL with linear function approximation. In the linear setting, while statistically efficient algorithms are known under BeLLMan completeness (e.g., Jiang et al. (2017); Zanette et al. (2020)), these algorithms all rely on the principle of global optimism which requires solving a nonconvex optimization problem. In particular, it has remained open as to whether computationally efficient algorithms exist. In this paper we give the first polynomial-time algorithm for RL under linear BeLLMan completeness when the number of actions is any constant.