Skip to main navigation Skip to search Skip to main content

Model-Free Robust Reinforcement Learning with Sample Complexity Analysis

  • Yudan Wang
  • , Shaofeng Zou
  • , Yue Wang
  • SUNY Buffalo
  • University of Central Florida

Research output: Contribution to journalConference articlepeer-review

2 Scopus citations

Abstract

Distributionally Robust Reinforcement Learning (DR-RL) aims to derive a policy optimizing the worst-case performance within a predefined uncertainty set. Despite extensive research, previous DR-RL algorithms have predominantly favored model-based approaches, with limited availability of model-free methods offering convergence guarantees or sample complexities. This paper proposes a model-free DR-RL algorithm leveraging the Multi-level Monte Carlo (MLMC) technique to close such a gap. Our innovative approach integrates a threshold mechanism that ensures finite sample requirements for algorithmic implementation, a significant improvement than previous model-free algorithms. We develop algorithms for uncertainty sets defined by total variation, Chi-square divergence, and KL divergence, and provide finite sample analyses under all three cases. Remarkably, our algorithms represent the first model-free DR-RL approach featuring finite sample complexity for total variation and Chi-square divergence uncertainty sets, while also offering an improved sample complexity and broader applicability compared to existing model-free DR-RL algorithms for the KL divergence model. The complexities of our method establish the tightest results for all three uncertainty models in model-free DR-RL, underscoring the effectiveness and efficiency of our algorithm, and highlighting its potential for practical applications.

Original languageEnglish
Pages (from-to)3470-3513
Number of pages44
JournalProceedings of Machine Learning Research
Volume244
StatePublished - 2024
Event40th Conference on Uncertainty in Artificial Intelligence, UAI 2024 - Barcelona, Spain
Duration: Jul 15 2024Jul 19 2024

Fingerprint

Dive into the research topics of 'Model-Free Robust Reinforcement Learning with Sample Complexity Analysis'. Together they form a unique fingerprint.

Cite this