Near Minimax Rate for Quantile Distributional Reinforcement Learning

Abstract

In this paper, we study quantile distributional policy evaluation, where the return distribution under a given policy is approximated by the quantile Bellman fixed point $\bmη_m$. We investigate the statistical accuracy of estimating $\bmη_m$ across quantile resolutions under Wasserstein distances. Assuming access to a generative model, we construct a quantile fixed point estimator $\bmη_m^{(n)}$ from an empirical Markov decision process and establish non-asymptotic error bounds under Wasserstein distances. When reward densities are uniformly bounded away from zero, we obtain a root-$n$ high-probability bound simultaneously over all resolutions $m$ and all Wasserstein orders $1\le p\le\infty$. When reward densities vanish at the support endpoints, we show that continuity and endpoint monotonicity suffice to preserve the same resolution-independent rate for $1\le p\le2$. For $p>2$, we establish a root-$n$ bound with coefficient $m^{1/2-1/p}$, together with a slower, resolution-independent $n^{-1/p}$ bound for every finite $p>2$. We further obtain resolution-independent $W_\infty$ consistency governed by the continuity modulus of reward quantile functions. To investigate the sharpness of these bounds, we characterize the exact local asymptotic minimax risk at each fixed resolution as the expected Wasserstein-coordinate norm of the efficient Gaussian limit and prove that our estimator attains it. We then examine the resolution dependence of this optimal local risk. A regular one-state instance has resolution-independent local risk constants, whereas a single endpoint-vanishing two-state instance exhibits growth of order $m^{1/2-1/p}$, up to logarithmic factors along a resolution subsequence, for every $p>2$.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

More in Machine Learning (Statistics)

• arXiv preprint · 21 October 2026

  A Jump-Modified Wasserstein Gradient Flow for Maximum Mean Discrepancy with Energy Kernels

  Ansh Desai, Jason Liu, Srujan Roplekar and 1 other
• arXiv preprint · 22 October 2026

  Lifetime-valid decisions by continually evolving AI analysts

  Xiaojing Sun, Yuhan Zeng and Xiao Wang
• arXiv preprint · 21 October 2026

  Internal Guidance as a Sampling Method for Tilted Diffusion Models

  Curtis McDonald and Peter Bartlett
• arXiv preprint · 22 October 2026

  Residual Calibration via Local Feature-Space Refinement

  Xihao Wang, Chengyao Yu and Ruixing Ming
• arXiv preprint · 21 October 2026

  Curvature-Scaled Sharpness-Aware Minimization: Convergence Beyond Smoothness

  Dimitris Oikonomou and Nicolas Loizou

All new papers in Machine Learning (Statistics)