分位数分布式强化学习达到极小化极大速率的理论研究
Near Minimax Rate for Quantile Distributional Reinforcement Learning
这篇论文把分位数分布式 RL 的估计误差界做到了极小化极大最优,统计方向的同学可以细看证明思路。
论文研究分位数分布式策略评估,通过经验 MDP 构造分位数不动点估计量 η_m^(n),并在 Wasserstein 距离下建立非渐近误差界。当奖励密度一致有下界时,估计量在所有分辨率 m 和所有 1≤p≤∞ 的 Wasserstein 阶上同时获得 root-n 高概率界。论文进一步刻画每个固定分辨率下的局部渐近极小化极大风险,证明该估计量达到该风险,并给出正则实例与端点消失实例的风险增长阶对比。
Near Minimax Rate for Quantile Distributional Reinforcement Learning
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$.