arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.01734cs.LGmath.OC
收藏

随机策略梯度方法:Fisher 非退化策略的改进样本复杂度

Stochastic Policy Gradient Methods: Improved Sample Complexity for Fisher-non-degenerate Policies

ETH Zurich(苏黎世联邦理工学院)

Ilyas Fatkhullin, Anas Barakat, Anastasia Kireeva, Niao He

AI总结:

针对Fisher非退化参数化策略全局收敛样本复杂度高的问题,提出N-PG-IGT将复杂度降至ε^-2.5,并用Hessian校正的(N)-HARPG进一步达到ε^-2,算法单循环、省内存且适用于连续状态动作空间。

摘要:

近来,策略梯度(PG)方法在经验上的显著成功推动了其理论基础的发展。尽管人们为设计高效的随机 PG 类算法付出了巨大努力,但对其收敛到全局最优策略的理解仍然有限。在本工作中,我们针对一类广泛的 Fisher 非退化参数化策略建立了改进的全局收敛保证,从而能够处理连续状态-动作空间的情形。首先,我们提出一种带隐式梯度传输的归一化策略梯度方法(N-PG-IGT),并推导该方法用于寻找全局 ε-最优策略时的 $\tilde{\mathcal{O}}(\varepsilon^{-2.5})$ 样本复杂度。该算法改进了此前已知的 $\tilde{\mathcal{O}}(\varepsilon^{-3})$ 复杂度,不需要使用重要性采样或二阶信息,并且每次迭代仅采样一条轨迹。其次,通过考虑一种借助 Hessian-向量积校正而增强的 Hessian 辅助递归策略梯度((N)-HARPG)算法,我们进一步将该复杂度改进至 $\tilde{\mathcal{O}}(\varepsilon^{-2})$。有趣的是,两种算法都:(i) 简单且易于实现:单循环,不需要大批量轨迹,并且每次迭代至多采样两条轨迹;(ii) 计算和内存高效:它们在每次迭代中不需要昂贵的子程序,并且可以使用与参数维度呈线性关系的内存来实现。

英文摘要:

Recently, the impressive empirical success of policy gradient (PG) methods has catalyzed the development of their theoretical foundations. Despite the huge efforts directed at the design of efficient stochastic PG-type algorithms, the understanding of their convergence to a globally optimal policy is still limited. In this work, we develop improved global convergence guarantees for a general class of Fisher-non-degenerate parameterized policies which allows to address the case of continuous state action spaces. First, we propose a Normalized Policy Gradient method with Implicit Gradient Transport (N-PG-IGT) and derive a $\tilde{\mathcal{O}}(\varepsilon^{-2.5})$ sample complexity of this method for finding a global $\varepsilon$-optimal policy. Improving over the previously known $\tilde{\mathcal{O}}(\varepsilon^{-3})$ complexity, this algorithm does not require the use of importance sampling or second-order information and samples only one trajectory per iteration. Second, we further improve this complexity to $\tilde{ \mathcal{\mathcal{O}} }(\varepsilon^{-2})$ by considering a Hessian-Aided Recursive Policy Gradient ((N)-HARPG) algorithm enhanced with a correction based on a Hessian-vector product. Interestingly, both algorithms are $(i)$ simple and easy to implement: single-loop, do not require large batches of trajectories and sample at most two trajectories per iteration; $(ii)$ computationally and memory efficient: they do not require expensive subroutines at each iteration and can be implemented with memory linear in the dimension of parameters.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑