Skip to main navigation Skip to search Skip to main content

Near-Optimal Sample Complexity for Iterated CVaR Reinforcement Learning with a Generative Model

  • Zilong Deng
  • , Simon Khan
  • , Shaofeng Zou
  • Arizona State University
  • Air Force Research Laboratory

Research output: Contribution to journalConference articlepeer-review

1 Scopus citations

Abstract

In this work, we study the sample complexity problem of risk-sensitive Reinforcement Learning (RL) with a generative model, where we aim to maximize the Conditional Value at Risk (CVaR) with risk tolerance level τ at each step, named Iterated CVaR. We develop nearly matching upper and lower bounds on the sample complexity for this problem. Specifically, we first prove that a value iteration-based algorithm, ICVaR-VI, achieves an ϵ-optimal policy with at most Õ (SA/((1-γ )4τ2ϵ2) samples, where γ is the discount factor, and S,A are the sizes of the state and action spaces. Furthermore, if τ ≥ γ, then the sample complexity can be further improved to Õ (SA/(1- γ)3ϵ2). We further show a minimax lower bound of Õ ((1-γτ) SA/(1-γ)4τϵ2). For a constant risk level 0 < τ ≤ 1, our upper and lower bounds match with each other, demonstrating the tightness and optimality of our analyses. We also investigate a limiting case with a small risk level τ, called Worst-Path RL, where the objective is to maximize the minimum possible cumulative reward. We develop matching upper and lower bounds of Õ (SA/pmin), where pmin denotes the minimum non-zero reaching probability of the transition kernel.

Original languageEnglish
Pages (from-to)3907-3915
Number of pages9
JournalProceedings of Machine Learning Research
Volume258
StatePublished - 2025
Event28th International Conference on Artificial Intelligence and Statistics, AISTATS 2025 - Mai Khao, Thailand
Duration: May 3 2025May 5 2025

Fingerprint

Dive into the research topics of 'Near-Optimal Sample Complexity for Iterated CVaR Reinforcement Learning with a Generative Model'. Together they form a unique fingerprint.

Cite this