PPO算法

PPO(Proximal Policy Optimization,近端策略优化)算法是一种在强化学习领域广泛应用的策略优化算法。它在2017年由John Schulman等人提出,是TRPO(Trust Region Policy Optimization,信任域策略优化)算法的改进版本,旨在解决TRPO计算复杂度高、实现困难的问题,同时保持良好的性能。PPO算法在许多实际应用中表现出色,尤其是在机器人控制、游戏AI等领域。

1. PPO算法的核心思想

PPO算法是一种基于策略梯度的方法,其目标是通过优化策略函数 π(a∣s)\pi(a|s)π(as) 来最大化累积奖励。策略函数表示在状态 sss 下选择动作 aaa 的概率分布。PPO的核心思想是通过更新策略函数,使得新的策略在一定程度上接近旧的策略,从而保证更新的稳定性。

PPO算法的关键在于引入了截断的概率比(clipped probability ratio)和信任域约束(trust region constraint)。具体来说,PPO通过限制策略更新的幅度来避免过度更新,从而保证更新过程的稳定性和收敛性。

解决了什么问题

    1. 策略更新不稳定

传统策略梯度方法(如REINFORCE)在优化过程中可能会导致策略更新过大,策略更新不稳定,从而引发训练过程中的剧烈波动,甚至导致策略崩溃。PPO通过引入剪切(Clipping)机制,限制新旧策略之间的差异,确保策略更新在一个可控范围内,从而提高训练的稳定性

    1. 样本效率低

传统方法中,每个数据样本通常只用于一次梯度更新,导致样本利用率较低。PPO通过重要性采样(Importance Sampling)技术,允许重复利用历史数据,从而提高样本效率。

2. PPO算法的理论的前置知识

好的!在介绍PPO算法之前,我们先来解释一下强化学习中的一些基本概念。这些概念是理解PPO算法的基础。

1. 强化学习(Reinforcement Learning)

强化学习是一种机器学习范式,智能体(Agent)通过与环境(Environment)的交互来学习如何做出最优决策。智能体的目标是最大化累积奖励(Cumulative Reward)。

2. 智能体(Agent)

智能体是强化学习中的主体,它根据当前的状态(State)选择一个动作(Action),并根据环境的反馈(奖励和新状态)来更新自己的策略(Policy)。

3. 环境(Environment)

环境是智能体所处的外部世界。环境根据智能体的动作给出反馈,包括新的状态(State)和奖励(Reward)。

4. 状态(State)

状态是环境在某一时刻的描述。智能体根据当前状态选择动作。状态可以是离散的(如网格世界中的位置)或连续的(如机器人的关节角度)。

5. 动作(Action)

动作是智能体在某一时刻根据状态做出的选择。动作可以是离散的(如“向上”“向下”)或连续的(如机器人的关节速度)。

6. 策略网络(Policy)

策略是智能体的行为规则,它定义了在给定状态下选择动作的概率分布。策略通常用 π(a∣s,θ)\pi(a|s,\theta)π(as,θ) 表示,即在状态 sss 下选择动作 aaa 的概率。我们可以使用深度学习的神经网络来表示策略网络。

7. 奖励(Reward)

奖励是环境对智能体动作的反馈,是一个标量值。奖励信号告诉智能体其动作的好坏。智能体的目标是最大化累积奖励。

8. 累积奖励(Cumulative Reward)

累积奖励是指智能体在一系列动作中获得的总奖励。通常用折扣累积奖励(Discounted Cumulative Reward)来表示,公式为:
Gt=Rt+1+γRt+2+γ2Rt+3+… G_t = R_{t+1} + \gamma R_{t+2} + \gamma^2 R_{t+3} + \dots Gt=Rt+1+γRt+2+γ2Rt+3+
其中,γ\gammaγ 是折扣因子(0≤γ≤10 \leq \gamma \leq 10γ1),用于平衡短期和长期奖励。

9. 马尔可夫决策过程(Markov Decision Process, MDP)

强化学习问题通常被建模为马尔可夫决策过程。MDP是一个五元组 (S,A,P,R,γ)(S, A, P, R, \gamma)(S,A,P,R,γ)

  • SSS 是状态空间。
  • AAA 是动作空间。
  • PPP 是状态转移概率,表示在状态 sss 下选择动作 aaa 后转移到状态 s′s's 的概率。
  • RRR 是奖励函数,表示在状态 sss 下选择动作 aaa 后获得的奖励。
  • γ\gammaγ 是折扣因子。

10. 价值函数(Value Function)

价值函数评估在某个策略下,某个状态或状态-动作对的“价值”。

  • 状态价值函数(State Value Function)Vπ(s)V_\pi(s)Vπ(s) 表示在策略 π\piπ 下,从状态 sss 开始的期望累积奖励。
    Vπ(s)=Eπ[Gt∣St=s] V_\pi(s) = \mathbb{E}_\pi[G_t | S_t = s] Vπ(s)=Eπ[GtSt=s]
  • 动作价值函数(Action Value Function)Qπ(s,a)Q_\pi(s, a)Qπ(s,a) 表示在策略 π\piπ 下,从状态 sss 开始并选择动作 aaa 的期望累积奖励。
    Qπ(s,a)=Eπ[Gt∣St=s,At=a] Q_\pi(s, a) = \mathbb{E}_\pi[G_t | S_t = s, A_t = a] Qπ(s,a)=Eπ[GtSt=s,At=a]

11. 优势函数(Advantage Function)

优势函数 A(s,a)A(s, a)A(s,a) 表示在状态 sss 下选择动作 aaa 的优势,即:
A(s,a)=Q(s,a)−V(s) A(s, a) = Q(s, a) - V(s) A(s,a)=Q(s,a)V(s)
它衡量了选择动作 aaa 相对于平均动作的价值。

12. 策略梯度(Policy Gradient)

策略梯度是强化学习中一种优化策略的方法。通过计算策略的梯度来更新策略参数,使得期望累积奖励最大化。策略梯度的基本公式为:
∇θJ(θ)=Eπ[∇θlog⁡π(a∣s)⋅Gt] \nabla_\theta J(\theta) = \mathbb{E}_\pi[\nabla_\theta \log \pi(a|s) \cdot G_t] θJ(θ)=Eπ[θlogπ(as)Gt]
其中,J(θ)J(\theta)J(θ) 是目标函数,θ\thetaθ 是策略参数。

13. 演员-评论家方法(Actor-Critic Method)

演员-评论家方法是一种强化学习算法框架,包含两个部分:

  • 演员(Actor):负责根据当前策略选择动作。
  • 评论家(Critic):负责评估当前策略的价值函数或优势函数,为演员提供更新信号。

14. 信任区域(Trust Region)

信任区域是一种优化策略,确保策略更新不会偏离当前策略太远。PPO算法通过剪辑目标函数或KL散度惩罚来实现信任区域的概念。

15. 剪辑目标函数(Clipped Objective Function)

PPO算法通过剪辑目标函数来限制策略更新的幅度。具体来说,它计算新策略与旧策略之间的概率比值,并通过一个剪辑函数将其限制在一定范围内,从而防止策略更新过大。

16. KL散度(KL Divergence)

KL散度是衡量两个概率分布之间差异的指标。PPO算法通过限制新策略与旧策略之间的KL散度,确保策略更新的稳定性。

17. 折扣因子(Discount Factor)

折扣因子 γ\gammaγ 用于平衡短期和长期奖励。当 γ=1\gamma = 1γ=1 时,智能体更关注长期奖励;当 γ<1\gamma < 1γ<1 时,智能体更关注短期奖励。

3. 策略梯度定理

我们详细推导梯度策略算法中损失函数的表达式。这个推导过程将从策略梯度的基本定理开始,逐步引入采样和平均,最终得到损失函数的表达式。

1. 策略梯度定理

策略梯度定理提供了一种计算策略参数梯度的方法,该梯度指向期望回报增加的方向。对于参数化策略 πθ(a∣s)\pi_\theta(a|s)πθ(as),其梯度可以表示为:

∇θJ(θ)=Eτ∼πθ[∑t=0T∇θlog⁡πθ(at∣st)⋅Rt] \nabla_\theta J(\theta) = \mathbb{E}_{\tau \sim \pi_\theta} \left[ \sum_{t=0}^T \nabla_\theta \log \pi_\theta(a_t | s_t) \cdot R_t \right] θJ(θ)=Eτπθ[t=0Tθlogπθ(atst)Rt]

其中:

  • J(θ)J(\theta)J(θ) 是策略的期望回报。
  • τ=(s0,a0,s1,a1,…,sT)\tau = (s_0, a_0, s_1, a_1, \ldots, s_T)τ=(s0,a0,s1,a1,,sT) 是一个完整的轨迹,包括状态、动作和时间步。
  • RtR_tRt 是在时间步 ttt 获得的奖励。
  • πθ(at∣st)\pi_\theta(a_t | s_t)πθ(atst) 是在状态 sts_tst 下采取动作 ata_tat 的概率。

2. 期望回报的最大化

为了最大化期望回报 J(θ)J(\theta)J(θ),我们需要沿着梯度的方向更新策略参数 θ\thetaθ。这可以通过梯度上升实现:

θ←θ+α∇θJ(θ) \theta \leftarrow \theta + \alpha \nabla_\theta J(\theta) θθ+αθJ(θ)

其中 α\alphaα 是学习率。

3. 采样和平均

在实际应用中,直接计算期望值通常是不可行的,因为我们无法遍历所有可能的轨迹。因此,我们通过采样 NNN 个独立的轨迹来估计期望值。每个轨迹的长度可能不同,设为 TnT_nTn

对于每个采样的轨迹 τn\tau^nτn,我们可以计算其总奖励 R(τn)R(\tau^n)R(τn),然后使用这些轨迹来估计梯度:

∇θJ(θ)≈1N∑n=1N∑t=0Tn∇θlog⁡πθ(atn∣stn)⋅R(τn) \nabla_\theta J(\theta) \approx \frac{1}{N} \sum_{n=1}^N \sum_{t=0}^{T_n} \nabla_\theta \log \pi_\theta(a_t^n | s_t^n) \cdot R(\tau^n) θJ(θ)N1n=1Nt=0Tnθlogπθ(atnstn)R(τn)

这里,atna_t^natnstns_t^nstn 分别是第 nnn 个轨迹在时间步 ttt 的动作和状态。

4. 构建损失函数

为了优化策略参数 θ\thetaθ,我们需要最小化损失函数。损失函数通常被定义为策略梯度的负期望值:

L(θ)=−Eτ∼πθ[∑t=0Tlog⁡πθ(at∣st)⋅Rt] L(\theta) = -\mathbb{E}_{\tau \sim \pi_\theta} \left[ \sum_{t=0}^T \log \pi_\theta(a_t | s_t) \cdot R_t \right] L(θ)=Eτπθ[t=0Tlogπθ(atst)Rt]

通过采样和平均,我们可以估计这个期望值:

L(θ)≈−1N∑n=1N∑t=0Tnlog⁡πθ(atn∣stn)⋅R(τn) L(\theta) \approx -\frac{1}{N} \sum_{n=1}^N \sum_{t=0}^{T_n} \log \pi_\theta(a_t^n | s_t^n) \cdot R(\tau^n) L(θ)N1n=1Nt=0Tnlogπθ(atnstn)R(τn)

5. 损失函数的最终形式

最终,我们得到损失函数的表达式:

Loss=−1N∑n=1N∑t=1TnR(τn)log⁡πθ(ant∣snt) \text{Loss} = -\frac{1}{N} \sum_{n=1}^N \sum_{t=1}^{T_n} R(\tau^n) \log \pi_\theta(a_n^t | s_n^t) Loss=N1n=1Nt=1TnR(τn)logπθ(antsnt)

这个损失函数用于梯度下降或上升算法中,以更新策略参数 θ\thetaθ,从而提高策略的性能。

5.策略梯度来训练策略网络

以马里奥游戏为例,我们首先将游戏的画面当作环境输入到策略网络中,策略网络会输出动作的概率,我们从中采样并且获得动作、执行动作得到新的环境。我们不断重复这个流程,我们就可以得到N个动作和奖励,可以作为一个batch来进行参数更新。我们采集数据和更新数据用的都是同一个模型,我们称之为On Policy。他有一个问题,我们的大部分时间都在采集数据,训练的非常慢,这也是PPO算法要改进的。

4、损失函数的优化

这个函数还有没有可以优化的地方呢,答案是有的,原式子中,我们的一个动作可以影响整个轨迹的回报的好坏,事实上我们的一个动作只能影响这个动作之后的轨迹的回报,而不会影响这个动作之前的奖励。我们将R(τn)R(\tau^n)R(τn)改为*RtnR_t^nRtn:从当前时间步ttt到结束的奖励

还有就是我们可以加入基线,让相对好的动作概率增加,让相对差的概率减少

1N∑n=1N∑t=1Tn(Rtn−B(snt))∇log⁡Pθ(ant∣snt) \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} (R_t^n - B(s_n^t)) \nabla \log P_\theta(a_n^t | s_n^t) N1n=1Nt=1Tn(RtnB(snt))logPθ(antsnt)

公式组成部分

  1. NNN:表示采样的轨迹数量。
  2. TnT_nTn:表示第 nnn 个轨迹的长度。
  3. RtnR_t^nRtn:表示从当前时间步 ttt 到结束的奖励。
  4. B(snt)B(s_n^t)B(snt):表示基线函数,通常是状态 snts_n^tsnt 的价值函数 V(s)V(s)V(s) 或优势函数 A(s,a)A(s, a)A(s,a) 的估计。基线用于减少梯度估计的方差。
  5. ∇log⁡Pθ(ant∣snt)\nabla \log P_\theta(a_n^t | s_n^t)logPθ(antsnt):表示在状态 snts_n^tsnt 下采取动作 anta_n^tant 的对数概率关于策略参数 θ\thetaθ 的梯度。这是策略梯度的一部分,用于指导策略参数的更新。

公式解释

  • 外层求和 ∑n=1N\sum_{n=1}^{N}n=1N:对所有采样的轨迹进行求和。
  • 内层求和 ∑t=1Tn\sum_{t=1}^{T_n}t=1Tn:对每个轨迹中的所有时间步进行求和。
  • (Rtn−B(snt))(R_t^n - B(s_n^t))(RtnB(snt)):计算奖励与基线的差值,这个差值表示了实际奖励与预期奖励之间的偏差。这个偏差用于调整策略,使其倾向于选择那些能够获得更高奖励的动作。
  • ∇log⁡Pθ(ant∣snt)\nabla \log P_\theta(a_n^t | s_n^t)logPθ(antsnt):计算对数概率的梯度,这个梯度指示了如何调整策略参数以增加在给定状态下选择特定动作的概率。

Rtn−B(snt)R_t^n - B(s_n^t)RtnB(snt)可以表示为优势函数,我们可以使用优势函数来替换基线的:

1N∑n=1N∑t=1Tn(Aθ(snt,ant))∇log⁡Pθ(ant∣snt) \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} (A_{\theta}(s_n^t,a_n^t)) \nabla \log P_\theta(a_n^t | s_n^t) N1n=1Nt=1Tn(Aθ(snt,ant))logPθ(antsnt)

更新策略参数

这个梯度表达式通常用于梯度上升算法中,以更新策略参数 θ\thetaθ

θ←θ+α⋅1N∑n=1N∑t=1Tn(Aθ(snt,ant))∇log⁡Pθ(ant∣snt) \theta \leftarrow \theta + \alpha \cdot \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} (A_{\theta}(s_n^t,a_n^t)) \nabla \log P_\theta(a_n^t | s_n^t) θθ+αN1n=1Nt=1Tn(Aθ(snt,ant))logPθ(antsnt)

其中,α\alphaα 是学习率,控制着参数更新的步长。

5. 广义优势估计

这个式子描述的是广义优势估计(Generalized Advantage Estimation,简称GAE)的计算公式。GAE是一种用于强化学习中策略梯度方法的算法,它通过结合多步估计来提高优势函数(Advantage Function)的估计效率和稳定性。

GAE公式解释

AθGAE(st,a)=(1−λ)(Aθ1+λ⋅Aθ2+λ2⋅Aθ3+⋯ ) A_{\theta}^{GAE}(s_t, a) = (1 - \lambda)(A_{\theta}^1 + \lambda \cdot A_{\theta}^2 + \lambda^2 \cdot A_{\theta}^3 + \cdots) AθGAE(st,a)=(1λ)(Aθ1+λAθ2+λ2Aθ3+)

  • AθGAE(st,a)A_{\theta}^{GAE}(s_t, a)AθGAE(st,a):表示在状态 sts_tst 下采取动作 aaa 的广义优势估计值。
  • λ\lambdaλ:是一个介于0和1之间的超参数,用于控制多步估计的权重。较小的 λ\lambdaλ 值意味着更多地关注单步估计,而较大的 λ\lambdaλ 值意味着更多地关注多步估计。
  • AθkA_{\theta}^kAθk:表示 kkk 步优势估计,即从当前状态 sts_tst 开始,考虑 kkk 步未来奖励与价值函数的差值。

优势函数

优势函数 Aθ(st,at)A_{\theta}(s_t, a_t)Aθ(st,at) 定义为采取动作 ata_tat 相对于平均策略的相对优势,它可以通过以下方式计算:

Aθ(st,at)=Qθ(st,at)−Vθ(st) A_{\theta}(s_t, a_t) = Q_{\theta}(s_t, a_t) - V_{\theta}(s_t) Aθ(st,at)=Qθ(st,at)Vθ(st)

其中:

  • Qθ(st,at)Q_{\theta}(s_t, a_t)Qθ(st,at) 是动作价值函数(Action-Value Function),表示在状态 sts_tst 下采取动作 ata_tat 并遵循策略 θ\thetaθ 后的期望回报。
  • Vθ(st)V_{\theta}(s_t)Vθ(st) 是状态价值函数(State-Value Function),表示在状态 sts_tst 下遵循策略 θ\thetaθ 后的期望回报。

多步优势估计

多步优势估计 AθkA_{\theta}^kAθk 考虑了从当前状态开始的 kkk 步未来奖励与价值函数的差值,可以表示为:

Aθk=∑l=1kδlTD⋅γl−1 A_{\theta}^k = \sum_{l=1}^{k} \delta_l^{TD} \cdot \gamma^{l-1} Aθk=l=1kδlTDγl1

其中:

  • δlTD\delta_l^{TD}δlTD 是时间差分误差(Temporal Difference Error),定义为 rt+l+γVθ(st+l)−Vθ(st)r_{t+l} + \gamma V_{\theta}(s_{t+l}) - V_{\theta}(s_t)rt+l+γVθ(st+l)Vθ(st),其中 rt+lr_{t+l}rt+l 是在时间步 t+lt+lt+l 获得的奖励,γ\gammaγ 是折扣因子。
  • γl−1\gamma^{l-1}γl1 是折扣因子的 l−1l-1l1 次幂,用于调整未来奖励的权重。

图中展示的是广义优势估计(GAE)的推导过程。GAE 是一种用于强化学习中策略梯度方法的算法,它通过结合多步估计来提高优势函数的估计效率和稳定性。下面是图中推导过程的详细解释:

推导过程:

  1. GAE 的定义
    AθGAE(st,a)=(1−λ)(Aθ1+λAθ2+λ2Aθ3+⋯ ) A_{\theta}^{GAE}(s_t, a) = (1 - \lambda)(A_{\theta}^1 + \lambda A_{\theta}^2 + \lambda^2 A_{\theta}^3 + \cdots) AθGAE(st,a)=(1λ)(Aθ1+λAθ2+λ2Aθ3+)
    这里,AθkA_{\theta}^kAθk 表示 kkk 步优势估计,λ\lambdaλ 是一个介于0和1之间的超参数,用于控制多步估计的权重。

  2. 展开 AθkA_{\theta}^kAθk
    Aθk=δtV+γδt+1V+γ2δt+2V+⋯+γk−1δt+k−1V A_{\theta}^k = \delta_t^V + \gamma \delta_{t+1}^V + \gamma^2 \delta_{t+2}^V + \cdots + \gamma^{k-1} \delta_{t+k-1}^V Aθk=δtV+γδt+1V+γ2δt+2V++γk1δt+k1V
    其中,δtV=rt+γV(st+1)−V(st)\delta_t^V = r_t + \gamma V(s_{t+1}) - V(s_t)δtV=rt+γV(st+1)V(st) 是时间差分误差,γ\gammaγ 是折扣因子。

  3. 代入 AθkA_{\theta}^kAθk
    AθGAE(st,a)=(1−λ)(δtV+λ(δtV+γδt+1V)+λ2(δtV+γδt+1V+γ2δt+2V)+⋯ ) A_{\theta}^{GAE}(s_t, a) = (1 - \lambda)(\delta_t^V + \lambda(\delta_t^V + \gamma \delta_{t+1}^V) + \lambda^2(\delta_t^V + \gamma \delta_{t+1}^V + \gamma^2 \delta_{t+2}^V) + \cdots) AθGAE(st,a)=(1λ)(δtV+λ(δtV+γδt+1V)+λ2(δtV+γδt+1V+γ2δt+2V)+)

  4. 整理
    AθGAE(st,a)=(1−λ)(δtV(1+λ+λ2+⋯ )+γδt+1V(λ+λ2+⋯ )+⋯ ) A_{\theta}^{GAE}(s_t, a) = (1 - \lambda)(\delta_t^V (1 + \lambda + \lambda^2 + \cdots) + \gamma \delta_{t+1}^V (\lambda + \lambda^2 + \cdots) + \cdots) AθGAE(st,a)=(1λ)(δtV(1+λ+λ2+)+γδt+1V(λ+λ2+)+)

  5. 利用等比数列求和公式
    AθGAE(st,a)=(1−λ)(δtV11−λ+γδt+1Vλ1−λ+⋯ ) A_{\theta}^{GAE}(s_t, a) = (1 - \lambda)(\delta_t^V \frac{1}{1 - \lambda} + \gamma \delta_{t+1}^V \frac{\lambda}{1 - \lambda} + \cdots) AθGAE(st,a)=(1λ)(δtV1λ1+γδt+1V1λλ+)

  6. 简化
    AθGAE(st,a)=∑b=0∞(γλ)bδt+bV A_{\theta}^{GAE}(s_t, a) = \sum_{b=0}^{\infty} (\gamma \lambda)^b \delta_{t+b}^V AθGAE(st,a)=b=0(γλ)bδt+bV

这个推导过程展示了如何从GAE的定义出发,通过展开多步优势估计并利用等比数列求和公式,最终得到一个简洁的表达式。这个表达式表明,GAE可以看作是所有未来时间步的时间差分误差的加权和,其中权重由 γ\gammaγλ\lambdaλ 决定。这种方法在强化学习中被广泛应用,因为它能够在不同时间尺度上平衡估计的偏差和方差。

6、最后推导出来的式子

图片中的内容是强化学习中计算广义优势估计(Generalized Advantage Estimation,简称GAE)的公式和损失函数的表达式。GAE是一种用于策略梯度方法中的优势函数估计技术,它通过结合多步估计来提高估计的效率和稳定性。下面是图片中各个部分的详细解释:

  1. 时间差分误差(Temporal Difference Error):
    δtV=rt+γ⋅Vθ(st+1)−Vθ(st) \delta_t^V = r_t + \gamma \cdot V_\theta(s_{t+1}) - V_\theta(s_t) δtV=rt+γVθ(st+1)Vθ(st)
    这里,rtr_trt 是在时间步 ttt 获得的奖励,γ\gammaγ 是折扣因子,Vθ(s)V_\theta(s)Vθ(s) 是状态 sss 的价值函数,参数为 θ\thetaθ

  2. 广义优势估计(GAE):
    AθGAE(st,a)=∑b=0∞(γλ)bδt+bV A_{\theta}^{GAE}(s_t, a) = \sum_{b=0}^{\infty} (\gamma \lambda)^b \delta_{t+b}^V AθGAE(st,a)=b=0(γλ)bδt+bV
    这个公式表示GAE是所有未来时间步的时间差分误差的加权和,其中权重由 γ\gammaγλ\lambdaλ 决定。λ\lambdaλ 是一个介于0和1之间的超参数,用于控制多步估计的权重。

  3. 损失函数:
    1N∑n=1N∑t=1TnAθGAE(snt,ant)∇log⁡Pθ(ant∣snt) \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta}^{GAE}(s_n^t, a_n^t) \nabla \log P_\theta(a_n^t | s_n^t) N1n=1Nt=1TnAθGAE(snt,ant)logPθ(antsnt)
    这个公式是用于策略梯度方法中的损失函数,其中:

    • NNN 是采样的轨迹数量。
    • TnT_nTn 是第 nnn 个轨迹的长度。
    • AθGAE(snt,ant)A_{\theta}^{GAE}(s_n^t, a_n^t)AθGAE(snt,ant) 是在状态 snts_n^tsnt 下采取动作 anta_n^tant 的广义优势估计值。
    • ∇log⁡Pθ(ant∣snt)\nabla \log P_\theta(a_n^t | s_n^t)logPθ(antsnt) 是在状态 snts_n^tsnt 下采取动作 anta_n^tant 的对数概率关于策略参数 θ\thetaθ 的梯度。

这个损失函数用于梯度下降或上升算法中,以更新策略参数 θ\thetaθ,从而提高策略的性能。通过最小化这个损失函数,我们可以有效地优化策略参数,使智能体在环境中的表现更好。

3. PPO算法的实现步骤(off policy)

off policy,它可以使用与当前正在学习的策略不同的策略生成的数据来学习。这种算法的优势在于它能够利用旧的数据进行学习,从而提高数据的利用效率。例如,Q-learning和DQN(Deep Q-Networks)是典型的off-policy算法。Off-policy算法的一个关键特点是它们允许从历史数据中学习,这些数据可能是由不同的探索策略产生的,与当前正在优化的策略并不一定相同。这种方法可以在不与环境进行实时互动的情况下,通过经验回放(Experience Replay)技术来训练智能体
PPO算法的实现通常包括以下几个步骤:

(1)重要性采样(Importance Sampling)

图片内容解释了重要性采样(Importance Sampling)的基本概念和数学推导。重要性采样是一种统计方法,用于估计一个分布下的期望值,当直接从该分布中采样困难或不可行时,可以从一个近似分布中采样,然后通过重要性权重进行调整。

推导过程:

  1. 期望的定义
    E(f(x))x∼p(x)=∑xf(x)⋅p(x) \mathbb{E}(f(x))_{x \sim p(x)} = \sum_x f(x) \cdot p(x) E(f(x))xp(x)=xf(x)p(x)
    这里,E(f(x))x∼p(x)\mathbb{E}(f(x))_{x \sim p(x)}E(f(x))xp(x) 表示函数 f(x)f(x)f(x) 在分布 p(x)p(x)p(x) 下的期望值。

  2. 引入另一个分布 q(x)q(x)q(x)
    =∑xf(x)⋅p(x)q(x)q(x) = \sum_x f(x) \cdot p(x) \frac{q(x)}{q(x)} =xf(x)p(x)q(x)q(x)
    这里,我们在 p(x)p(x)p(x) 前乘以 q(x)/q(x)q(x)/q(x)q(x)/q(x),这样做的目的是为了能够将 p(x)p(x)p(x) 转换为 q(x)q(x)q(x)

  3. 重写求和表达式
    =∑xf(x)p(x)q(x)⋅q(x) = \sum_x f(x) \frac{p(x)}{q(x)} \cdot q(x) =xf(x)q(x)p(x)q(x)
    这一步将 p(x)p(x)p(x)q(x)q(x)q(x) 分开,为后续的期望转换做准备。

  4. 转换为 q(x)q(x)q(x) 下的期望
    =E(f(x)p(x)q(x))x∼q(x) = \mathbb{E}\left(f(x) \frac{p(x)}{q(x)}\right)_{x \sim q(x)} =E(f(x)q(x)p(x))xq(x)
    这一步表明,我们现在可以在分布 q(x)q(x)q(x) 下计算 f(x)p(x)q(x)f(x) \frac{p(x)}{q(x)}f(x)q(x)p(x) 的期望值。

  5. 近似估计
    ≈1N∑n=1Nf(x)p(x)q(x)x∼q(x) \approx \frac{1}{N} \sum_{n=1}^N f(x) \frac{p(x)}{q(x)}_{x \sim q(x)} N1n=1Nf(x)q(x)p(x)xq(x)
    当直接计算 q(x)q(x)q(x) 下的期望不可行时,我们可以通过从 q(x)q(x)q(x) 中采样 NNN 个样本,并计算这些样本的平均值来近似期望值。

重要性采样通过从一个易于采样的分布 q(x)q(x)q(x) 中采样,并使用权重 p(x)q(x)\frac{p(x)}{q(x)}q(x)p(x) 来调整样本,从而估计目标分布 p(x)p(x)p(x) 下的期望值。这种方法在机器学习和统计学中非常有用,特别是在处理复杂分布或稀有事件时。

  1. 给损失函数增加重要性采样

图片中的内容涉及强化学习中策略梯度方法的一个重要概念,即如何利用重要性采样(Importance Sampling)来调整策略梯度估计。这通常出现在off-policy学习场景中,即学习策略(target policy)与生成数据的策略(behavior policy)不一致时。

原始公式:

1N∑n=1N∑t=1TnAθGAE(snt,ant)∇log⁡Pθ(ant∣snt) \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta}^{GAE}(s_n^t, a_n^t) \nabla \log P_{\theta}(a_n^t | s_n^t) N1n=1Nt=1TnAθGAE(snt,ant)logPθ(antsnt)

这个公式表示的是策略梯度的估计,其中:

  • NNN 是采样的轨迹数量。
  • TnT_nTn 是第 nnn 个轨迹的长度。
  • AθGAE(snt,ant)A_{\theta}^{GAE}(s_n^t, a_n^t)AθGAE(snt,ant) 是在状态 snts_n^tsnt 下采取动作 anta_n^tant 的广义优势估计值。
  • ∇log⁡Pθ(ant∣snt)\nabla \log P_{\theta}(a_n^t | s_n^t)logPθ(antsnt) 是在状态 snts_n^tsnt 下采取动作 anta_n^tant 的对数概率关于策略参数 θ\thetaθ 的梯度。

增加重要性采样的损失函数公式

图片展示了强化学习中使用梯度上升方法来优化策略参数的损失函数的推导过程。这个过程涉及到广义优势估计(Generalized Advantage Estimation, GAE)和重要性采样权重的使用。以下是详细的步骤解释:

  1. 初始损失函数
    1N∑n=1N∑t=1TnAθGAE(snt,ant)∇log⁡Pθ(ant∣snt) \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta}^{GAE}(s_n^t, a_n^t) \nabla \log P_{\theta}(a_n^t | s_n^t) N1n=1Nt=1TnAθGAE(snt,ant)logPθ(antsnt)
    这里,NNN 是采样的轨迹数量,TnT_nTn 是第 nnn 个轨迹的长度,AθGAE(snt,ant)A_{\theta}^{GAE}(s_n^t, a_n^t)AθGAE(snt,ant) 是在状态 snts_n^tsnt 下采取动作 anta_n^tant 的广义优势估计值,∇log⁡Pθ(ant∣snt)\nabla \log P_{\theta}(a_n^t | s_n^t)logPθ(antsnt) 是在状态 snts_n^tsnt 下采取动作 anta_n^tant 的对数概率关于策略参数 θ\thetaθ 的梯度。

  2. 引入重要性采样权重
    =1N∑n=1N∑t=1TnAθ′GAE(snt,ant)Pθ(ant∣snt)Pθ′(ant∣snt)∇log⁡Pθ(ant∣snt) = \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta'}^{GAE}(s_n^t, a_n^t) \frac{P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)} \nabla \log P_{\theta}(a_n^t | s_n^t) =N1n=1Nt=1TnAθGAE(snt,ant)Pθ(antsnt)Pθ(antsnt)logPθ(antsnt)
    这里,Pθ(ant∣snt)P_{\theta}(a_n^t | s_n^t)Pθ(antsnt) 是目标策略在状态 snts_n^tsnt 下采取动作 anta_n^tant 的概率,Pθ′(ant∣snt)P_{\theta'}(a_n^t | s_n^t)Pθ(antsnt) 是行为策略在状态 snts_n^tsnt 下采取动作 anta_n^tant 的概率。重要性采样权重用于调整由于策略变化导致的概率差异。

  3. 利用梯度和对数的关系
    =1N∑n=1N∑t=1TnAθ′GAE(snt,ant)Pθ(ant∣snt)Pθ′(ant∣snt)∇Pθ(ant∣snt)Pθ(ant∣snt) = \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta'}^{GAE}(s_n^t, a_n^t) \frac{P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)} \frac{\nabla P_{\theta}(a_n^t | s_n^t)}{P_{\theta}(a_n^t | s_n^t)} =N1n=1Nt=1TnAθGAE(snt,ant)Pθ(antsnt)Pθ(antsnt)Pθ(antsnt)Pθ(antsnt)
    这里,利用了 ∇log⁡f(x)=∇f(x)f(x)\nabla \log f(x) = \frac{\nabla f(x)}{f(x)}logf(x)=f(x)f(x) 的关系。

  4. 简化表达式
    =1N∑n=1N∑t=1TnAθ′GAE(snt,ant)∇Pθ(ant∣snt)Pθ′(ant∣snt) = \frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta'}^{GAE}(s_n^t, a_n^t) \frac{\nabla P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)} =N1n=1Nt=1TnAθGAE(snt,ant)Pθ(antsnt)Pθ(antsnt)

  5. 最终损失函数
    Loss=−1N∑n=1N∑t=1TnAθ′GAE(snt,ant)Pθ(ant∣snt)Pθ′(ant∣snt) \text{Loss} = -\frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta'}^{GAE}(s_n^t, a_n^t) \frac{P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)} Loss=N1n=1Nt=1TnAθGAE(snt,ant)Pθ(antsnt)Pθ(antsnt)
    最终的损失函数用于策略梯度方法中的梯度上升更新。通过最小化这个损失函数,我们可以优化策略参数 θ\thetaθ,使智能体在环境中的表现更好。

  6. 约束模型不要偏差太大

图片中展示了两种不同的损失函数,它们用于近端策略优化(Proximal Policy Optimization,简称PPO)算法中。PPO是一种强化学习算法,它通过限制策略更新的幅度来提高策略梯度方法的稳定性。以下是图片中两个损失函数的详细解释:

第一个损失函数(PPO1)(使用KL散度)

LossPPO1=−1N∑n=1N∑t=1TnAθ′GAE(snt,ant)Pθ(ant∣snt)Pθ′(ant∣snt)+βKL(Pθ,Pθ′) Loss_{PPO1} = -\frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} A_{\theta'}^{GAE}(s_n^t, a_n^t) \frac{P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)} + \beta KL(P_{\theta}, P_{\theta'}) LossPPO1=N1n=1Nt=1TnAθGAE(snt,ant)Pθ(antsnt)Pθ(antsnt)+βKL(Pθ,Pθ)

  • NNN 是采样的轨迹数量。
  • TnT_nTn 是第 nnn 个轨迹的长度。
  • Aθ′GAE(snt,ant)A_{\theta'}^{GAE}(s_n^t, a_n^t)AθGAE(snt,ant) 是在状态 snts_n^tsnt 下采取动作 anta_n^tant 的广义优势估计值,基于旧策略参数 θ′\theta'θ 计算。
  • Pθ(ant∣snt)P_{\theta}(a_n^t | s_n^t)Pθ(antsnt)Pθ′(ant∣snt)P_{\theta'}(a_n^t | s_n^t)Pθ(antsnt) 分别是新旧策略在状态 snts_n^tsnt 下采取动作 anta_n^tant 的概率。
  • βKL(Pθ,Pθ′)\beta KL(P_{\theta}, P_{\theta'})βKL(Pθ,Pθ) 是新旧策略之间的Kullback-Leibler散度,用于惩罚新策略与旧策略之间的差异过大,β\betaβ 是一个超参数,用于控制这种惩罚的强度。

第二个损失函数(PPO2)(使用裁剪)

LossPPO2=−1N∑n=1N∑t=1Tnmin⁡(Aθ′GAE(snt,ant)Pθ(ant∣snt)Pθ′(ant∣snt),clip(Pθ(ant∣snt)Pθ′(ant∣snt),1−ϵ,1+ϵ)Aθ′GAE(snt,ant)) Loss_{PPO2} = -\frac{1}{N} \sum_{n=1}^{N} \sum_{t=1}^{T_n} \min \left( A_{\theta'}^{GAE}(s_n^t, a_n^t) \frac{P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)}, \text{clip} \left( \frac{P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)}, 1 - \epsilon, 1 + \epsilon \right) A_{\theta'}^{GAE}(s_n^t, a_n^t) \right) LossPPO2=N1n=1Nt=1Tnmin(AθGAE(snt,ant)Pθ(antsnt)Pθ(antsnt),clip(Pθ(antsnt)Pθ(antsnt),1ϵ,1+ϵ)AθGAE(snt,ant))

  • 这个损失函数使用了裁剪(clip)操作来限制策略更新的幅度。裁剪函数 clip(x,1−ϵ,1+ϵ)\text{clip}(x, 1 - \epsilon, 1 + \epsilon)clip(x,1ϵ,1+ϵ) 确保比率 Pθ(ant∣snt)Pθ′(ant∣snt)\frac{P_{\theta}(a_n^t | s_n^t)}{P_{\theta'}(a_n^t | s_n^t)}Pθ(antsnt)Pθ(antsnt) 被限制在 1−ϵ1 - \epsilon1ϵ1+ϵ1 + \epsilon1+ϵ 之间,其中 ϵ\epsilonϵ 是一个超参数,通常取较小的值(如0.1或0.2)。

总结

PPO算法通过这两种损失函数来平衡策略的学习效率和更新稳定性。第一种损失函数通过KL散度来惩罚策略变化过大,而第二种损失函数通过裁剪操作直接限制策略更新的幅度。这两种方法都是为了确保策略更新不会过于激进,从而提高算法的稳定性和可靠性。

总结来说,引入重要性采样权重和利用梯度与对数的关系来推导和优化策略梯度方法中的损失函数。这种方法允许我们在off-policy学习场景中有效地利用旧数据进行学习。off-policy强化学习中使用重要性采样来调整策略梯度估计,以便利用与目标策略不同的行为策略生成的数据进行学习。我们可以用使用对另一个模型的评价来更新模型

Logo

开放原子旋武开源社区由开放原子开源基金会孵化及运营,业务方向涉及操作系统、终端设备、安全技术、基础软件等关键领域,致力于推动Rust编程语言在中国的开源生态建设与产业落地,面向开发者全面宣传和推广Rust。

更多推荐