No history yet

策略优化算法

策略优化算法

在深度强化学习中,直接优化参数化策略是一种主流方法。与基于价值的算法(如DQN)不同,策略优化方法直接学习一个将状态映射到动作的策略函数 πθ(as)\pi_{\theta}(a|s),并通过梯度上升来更新策略参数 θ\theta 以最大化期望回报。然而,简单的策略梯度方法存在一个核心问题:更新步长(学习率)过大可能导致策略性能急剧下降,且难以恢复。为了解决这一问题,研究人员提出了更稳健的优化方法,其中信任域策略优化(TRPO)和近端策略优化(PPO)是最具代表性的两种算法。

信任域策略优化 (TRPO)

TRPO的核心思想是,在更新策略时,既要最大化“替代”目标函数(Surrogate Objective Function),又要确保新的策略与旧的策略不会偏离太远。这种“偏离”的程度由一个以KL散度(Kullback-Leibler Divergence)为度量的“信任域”进行约束。通过这种方式,TRPO保证了策略的单调改进,避免了因过大的更新步长而导致的性能崩溃。

maximizeθE^t[πθ(atst)πθold(atst)A^t]subject toE^t[DKL(πθold(st)πθ(st))]δ\begin{aligned} \underset{\theta}{\text{maximize}} & \quad \hat{\mathbb{E}}_t \left[ \frac{\pi_{\theta}(a_t|s_t)}{\pi_{\theta_{old}}(a_t|s_t)} \hat{A}_t \right] \\ \text{subject to} & \quad \hat{\mathbb{E}}_t \left[ D_{KL}(\pi_{\theta_{old}}(\cdot|s_t) || \pi_{\theta}(\cdot|s_t)) \right] \leq \delta \end{aligned}

为了求解上述约束问题,TRPO首先对目标函数进行线性化处理,并对KL散度约束进行二阶泰勒展开,近似为 12(θθold)TF(θθold)δ\frac{1}{2}(\theta - \theta_{old})^T F (\theta - \theta_{old}) \leq \delta,其中 FF 是费雪信息矩阵(Fisher Information Matrix, FIM)。FIM可以看作是策略参数空间中的一个度量,用于衡量策略随参数变化的敏感度。

然后,TRPO使用共轭梯度(Conjugate Gradient)算法来高效求解更新方向,而无需显式计算和求逆复杂的FIM。最后,通过线性搜索(Line Search)确保更新后的策略既满足KL散度约束,又能实际提升性能。这种复杂的二阶优化方法虽然理论上保证了单调改进,但也带来了巨大的计算开销和实现难度。

近端策略优化 (PPO)

PPO旨在简化TRPO的复杂性,同时保持其稳健的性能。它放弃了复杂的二阶优化和严格的KL散度约束,转而采用一种更简单的一阶方法来实现类似的目标。PPO最常见的实现方式是裁剪替代目标函数 (Clipped Surrogate Objective)

LCLIP(θ)=E^t[min(rt(θ)A^t,clip(rt(θ),1ϵ,1+ϵ)A^t)]L^{CLIP}(\theta) = \hat{\mathbb{E}}_t \left[ \min(r_t(\theta)\hat{A}_t, \text{clip}(r_t(\theta), 1-\epsilon, 1+\epsilon)\hat{A}_t) \right]

通过这种裁剪机制,PPO隐式地将新策略限制在旧策略的一个小邻域内,从而达到了类似于TRPO信任域的效果。由于其目标函数的设计,PPO可以直接使用标准的随机梯度下降(SGD)及其变体(如Adam)进行优化,大大简化了算法实现,并提升了计算效率。这种简单性和高效性使其成为深度强化学习研究和应用中的默认选择。

由于其易用性和良好的性能,PPO已成为OpenAI默认的强化学习算法。

比较与应用

TRPO和PPO都旨在解决策略梯度方法中的破坏性大更新问题,但它们采取了截然不同的路径。

特性信任域策略优化 (TRPO)近端策略优化 (PPO)
核心思想使用KL散度作为硬约束使用裁剪目标函数作为软约束
优化方法二阶优化(共轭梯度)一阶优化(SGD,Adam)
实现复杂度高,涉及FIM和线性搜索低,易于实现和调试
样本效率通常较高,但每次迭代计算成本大样本效率略低于TRPO,但迭代速度快得多
性能稳定且强大通常与TRPO相当,有时略逊一筹,但总体性价比高
适用性适用于对稳定性要求极高的任务适用范围广,是大多数应用的首选

在实际应用中,PPO凭借其简单性、高效性和稳健的性能,在各种复杂环境中都取得了巨大成功。从机器人控制、自动驾驶到大规模游戏(如Dota 2),PPO都展现了其强大的能力。它在样本复杂性、实现简易性和计算性能之间取得了出色的平衡。

TRPO虽然在某些基准测试中可能表现出略微更优的样本效率,但其高昂的计算成本和实现难度限制了其在需要快速迭代和大规模部署的场景中的应用。因此,在大多数实际项目中,PPO通常是更实用和高效的选择。

Quiz Questions 1/6

信任域策略优化(TRPO)和近端策略优化(PPO)算法主要旨在解决传统策略梯度方法中的哪个核心问题?

Quiz Questions 2/6

TRPO算法通过什么机制来确保新策略与旧策略不会偏离太远,从而保证性能的单调改进?