English

Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes

Machine Learning 2024-06-28 v2 Statistics Theory Statistics Theory

Abstract

In offline reinforcement learning (RL), the absence of active exploration calls for attention on the model robustness to tackle the sim-to-real gap, where the discrepancy between the simulated and deployed environments can significantly undermine the performance of the learned policy. To endow the learned policy with robustness in a sample-efficient manner in the presence of high-dimensional state-action space, this paper considers the sample complexity of distributionally robust linear Markov decision processes (MDPs) with an uncertainty set characterized by the total variation distance using offline data. We develop a pessimistic model-based algorithm and establish its sample complexity bound under minimal data coverage assumptions, which outperforms prior art by at least O~(d)\widetilde{O}(d), where dd is the feature dimension. We further improve the performance guarantee of the proposed algorithm by incorporating a carefully-designed variance estimator.

Keywords

Cite

@article{arxiv.2403.12946,
  title  = {Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes},
  author = {He Wang and Laixi Shi and Yuejie Chi},
  journal= {arXiv preprint arXiv:2403.12946},
  year   = {2024}
}

Comments

accepted by Reinforcement Learning Conference (RLC)

R2 v1 2026-06-28T15:26:05.684Z