强化学习PPO：Proximal Policy Optimization Algorithms解读

news2024/12/27 11:17:29

PPO算法是一类Policy Gradient强化学习方法，经典的Policy Gradient通过一个参数化决策模型 $\pi(a|s,\theta )$ 来根据状态 $s$ 确定动作，其参数更新是通过下式进行的：

$\theta_{t+1} = \theta_{t} + \alpha \partial_{\theta_{t}} J(\theta_t)$

$J(\theta_t)$ 用于衡量决策模型的优劣目标，决策模型的优化目标为寻找最优决策，使得该决策下整体价值最大。

$\text{max}_{\pi}\ J(\theta) = E_{s,a\sim \pi}[\pi(a|s,\theta)Q_{\pi}(s,a)]$

因为最优决策 $\pi$ 是未知的，一种简单思路是直接当前参数模型 $\pi_\theta$ 进行寻优。称为Vanilla Policy Gradient。

1. Vanilla Policy Gradient

Vanilla Policy Gradient定义优化目标为：

$\text{max}_{\theta }\ J(\theta) \\= E_{s,a\sim {\pi_{\theta }}}[\pi(a|s,\theta)A(s,a)]\\=\sum_\tau \sum_{t=0}^T \sum_a\pi(a|s_t,\theta)A(s_t,a)$

$\tau$ 表示一轮episode， $t$ 表示episode中的某个时刻， $T$ 表示一轮episode的终态。
$A(s_t,a_t)=Q(s_t,a_t)-v(s_t|w)=G(s_t,a_t)-v(s_t|w)$ 称为Advantages estimates，其在原来预估 $Q(s_t,a_t)$ 基础上减去了平均状态价值估计 $v(s_t|w)$ ，这么做好处是扭正不同状态价值间偏差，可以加快模型收敛，具体可以参考这篇。

梯度计算式子：

$\partial J(\theta) \\ =\sum_\tau \sum_{t=0}^T \sum_a\partial_{\theta}\pi(a|s_t,\theta)A(s_t,a) \\=\sum_\tau \sum_{t=0}^T \sum_a \pi(a|s_t,\theta) A(s_t,a) \frac{\partial_{\theta}\pi(a|s_t,\theta)}{\pi(a|s_t,\theta)} \\\doteq \sum_\tau \sum_{t=0}^T A(s_t,a_t) \frac{\partial_{\theta}\pi(a_t|s_t,\theta)}{\pi(a_t|s_t,\theta)} \\=\sum_\tau \sum_{t=0}^T \sum_a \pi(a|s_t,\theta) A(s_t,a) \frac{\partial_{\theta}\pi(a|s_t,\theta)}{\pi(a|s_t,\theta)} \\\doteq \sum_\tau \sum_{t=0}^T A(s_t,a_t) \partial_{\theta}\text{In}(\pi(a_t|s_t,\theta))$

参数更新式子：

$\theta_{t+1}=\theta + \alpha\partial_{\theta} J(\theta_t)$

此外平均状态价值估计 $v(s_t|w)$ 通过VE损失进行优化，其同Policy Gradient更新按照AC的框架进行两步迭代更新。

2. Trust Region Policy Optimization

前面我们所提到的Vanilla Policy Gradient算法是将训练分为多个epoch，每个epoch会根据当前决策模型采样多个episode组合成一个batch，并在该epoch中针对于该batch进行多轮训练。

由于该轮epoch的训练数据是根据上轮epoch的模型进行采样的，此时就可能出现更新的决策模型同采样决策模型不一致，这个不一致同off-policy所带来的问题是类似的，可能会导致模型训练有偏差。

虽然Vanilla Policy Gradient算法是on-policy的，但是由于对batch进行多轮训练，在实际上造成了target policy和behavior policy不一致的问题。因此Trust Region Policy Optimization（TRPO）根据off-policy在原来优化目标上添加了重要性采样。

$J(\theta) = E_{s,a\sim {\pi_{\theta^k }}}[\frac{\pi(a|s,\theta)}{\pi(a|s,\theta^k)}A^{\theta^k}(s,a)]$

$D_{KL}(\pi(a|s,\theta )||\pi(a|s,\theta^k )) \leq \delta$

直接根据上面目标以及约束很难进行优化，因此TRPO做了一定式子简化：

$\text{max}_{\theta}\ J(\theta) \approx E_{s,a\sim {\pi_{\theta^k }}}[\pi(a|s,\theta)A^{\theta^k}(s,a)][\theta - \theta^k]=g[\theta - \theta^k] \\ s.t. \ D_{KL} \approx \frac{1}{2}[\theta - \theta^k]^T H[\theta - \theta^k] \leq \delta$

同时其给出决策参数的更新式子，式子中的 $\alpha^j$ 用于控制KL散度约束，此处采用了Backtracking line search技术，其通过设置多个 $\alpha^j$ 因子，并在训练中不断从中选择一个因子使得约束满足同时梯度更新最大。

$\theta_{t+1}=\theta_{t} + \alpha^j \sqrt{\frac{2\delta }{g^T_tH^{-1}g_t}}H^{-1}g$

3. Proximal Policy Optimization

TRPO有一个很大问题，在于其KL约束是通过惩罚项加到优化目标上去的，因此惩罚项的权重参数是一个超参数，其设置影响整体训练效果，虽然TRPO采用了Backtracking line search技术，可以自适应的去选择超参数，但是对于整体训练效果仍是有损的。

因此PPO认为KL约束既然是为了约束 $\pi(a|s,\theta )$ 相比于 $\pi(a|s,\theta^k )$ 变动不大，而优化目标已经包含了这两项，为什么不把这个约束添加到优化目标中。通过截断来约束 $r(\theta |\theta_k)$ 变动在一定范围内，从而到达同KL散度约束类似的效果。

$r(\theta |\theta_k)=\frac{\pi(a|s,\theta)}{\pi(a|s,\theta_k)}$

$J(\theta) = E_{s,a\sim {\pi_{\theta^k }}}(min[r(\theta|\theta_k)A^{\theta^k}(s,a), clip(r(\theta|\theta_k),1-\epsilon, 1+\epsilon)A^{\theta^k}(s,a)])$