LGNov 23, 2022

On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximation

Princeton
arXiv:2211.13208v224 citationsh-index: 58
Originality Highly original
AI Analysis

This work addresses the sample efficiency challenge in offline RL for researchers and practitioners, offering improved theoretical guarantees over prior minimax bounds, though it is incremental in advancing existing frameworks.

The paper tackles the problem of deriving instance-dependent bounds for offline reinforcement learning with linear function approximation, achieving a fast rate of Õ(1/K) under partial data coverage and a positive gap in optimal Q-values, and absolute zero sub-optimality error under specific conditions.

Sample-efficient offline reinforcement learning (RL) with linear function approximation has recently been studied extensively. Much of prior work has yielded the minimax-optimal bound of $\tilde{\mathcal{O}}(\frac{1}{\sqrt{K}})$, with $K$ being the number of episodes in the offline data. In this work, we seek to understand instance-dependent bounds for offline RL with function approximation. We present an algorithm called Bootstrapped and Constrained Pessimistic Value Iteration (BCP-VI), which leverages data bootstrapping and constrained optimization on top of pessimism. We show that under a partial data coverage assumption, that of \emph{concentrability} with respect to an optimal policy, the proposed algorithm yields a fast rate of $\tilde{\mathcal{O}}(\frac{1}{K})$ for offline RL when there is a positive gap in the optimal Q-value functions, even when the offline data were adaptively collected. Moreover, when the linear features of the optimal actions in the states reachable by an optimal policy span those reachable by the behavior policy and the optimal actions are unique, offline RL achieves absolute zero sub-optimality error when $K$ exceeds a (finite) instance-dependent threshold. To the best of our knowledge, these are the first $\tilde{\mathcal{O}}(\frac{1}{K})$ bound and absolute zero sub-optimality bound respectively for offline RL with linear function approximation from adaptive data with partial coverage. We also provide instance-agnostic and instance-dependent information-theoretical lower bounds to complement our upper bounds.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes