提出高效算法解决一般策略参数化下的鲁棒MDP问题,兼顾理论保证与实际计算效率。
Provably Efficient Algorithms for S- and Non-Rectangular Robust MDPs with General Parameterization
- 将平均奖励鲁棒MDP转化为熵正则化折扣鲁棒MDP,实现可解的均衡计算
- 设计新型多级蒙特卡洛梯度估计器,样本复杂度降至$\tilde{\mathcal{O}}(ε^{-2})$
- 首次给出非矩形不确定性下平均奖励鲁棒MDP的样本复杂度保证
我们研究在一般策略参数化下具有s-矩形和非矩形不确定性集的鲁棒马尔可夫决策过程(RMDPs)。以往工作大多局限于表格型策略,因此或缺乏样本复杂度保证,或计算成本过高。本文方法将平均奖励RMDPs转化为熵正则化的折扣鲁棒MDP,恢复强对偶性,实现可处理的均衡计算。我们证明了针对一般策略参数化的新型Lipschitz与Lipschitz光滑性性质,适用于无限状态空间。为解决无限时域梯度估计问题,引入一种多级蒙特卡洛梯度估计器,样本复杂度为$\tilde{\mathcal{O}}(ε^{-2})$,相较之前工作提升$\mathcal{O}(ε^{-2})$。基于此,我们设计了用于s-矩形不确定性的投影梯度下降算法($\mathcal{O}(ε^{-5})$)和用于非矩形不确定性的Frank--Wolfe算法(折扣情形$\mathcal{O}(ε^{-4})$,平均奖励情形$\mathcal{O}(ε^{-10.5})$),显著优于先前结果。本工作是首个在超越$(s,a)$-矩形性条件下,为一般策略参数化鲁棒MDP提供样本复杂度保证的研究;也是首个在平均奖励设置下给出此类保证的工作,并改进了现有折扣鲁棒MDP的界。
原文摘要 · Abstract (English)
We study robust Markov decision processes (RMDPs) with general policy parameterization under s-rectangular and non-rectangular uncertainty sets. Prior work is largely limited to tabular policies, and hence either lacks sample complexity guarantees or incurs high computational cost. Our method reduces the average reward RMDPs to entropy-regularized discounted robust MDPs, restoring strong duality and enabling tractable equilibrium computation. We prove novel Lipschitz and Lipschitz-smoothness properties for general policy parameterizations that extends to infinite state spaces. To address infinite-horizon gradient estimation, we introduce a multilevel Monte Carlo gradient estimator with $\tilde{\mathcal{O}}(ε^{-2})$ sample complexity, a factor of $\mathcal{O}(ε^{-2})$ improvement over prior work. Building on this, we design a projected gradient descent algorithm for s-rectangular uncertainty ($\mathcal{O}(ε^{-5})$) and a Frank--Wolfe algorithm for non-rectangular uncertainty ($\mathcal{O}(ε^{-4})$ discounted, $\mathcal{O}(ε^{-10.5})$ average reward), significantly improving prior results in both the discounted setting and average reward setting. Our work is the first one to provide sample complexity guarantees for RMDPs with general policy parameterization beyond $(s, a)$-rectangularity. It also provides the first such guarantees in the average reward setting and improves existing bounds for discounted robust MDPs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。