文章目录
- 概览:RL方法分类
- 策略梯度(Policy Gradient)
- Basic Policy Gradient
- 目标函数1:平均状态值
- 目标函数2:平均单步奖励
- 🟡PG梯度计算
- 🟦REINFORCE
本系列文章介绍强化学习基础知识与经典算法原理,大部分内容来自西湖大学赵世钰老师的强化学习的数学原理课程(参考资料1),并参考了部分参考资料2、3的内容进行补充。
系列博文索引:
- 强化学习的数学原理学习笔记 - RL基础知识
- 强化学习的数学原理学习笔记 - 基于模型(Model-based)
- 强化学习的数学原理学习笔记 - 蒙特卡洛方法(Monte Carlo)
- 强化学习的数学原理学习笔记 - 时序差分学习(Temporal Difference)
- 强化学习的数学原理学习笔记 - 值函数近似(Value Function Approximation)
- 强化学习的数学原理学习笔记 - 策略梯度(Policy Gradient)
- 强化学习的数学原理学习笔记 - Actor-Critic
参考资料:
- 【强化学习的数学原理】课程:从零开始到透彻理解(完结)(主要)
- Sutton & Barto Book: Reinforcement Learning: An Introduction
- 机器学习笔记
*注:【】内文字为个人想法,不一定准确
概览:RL方法分类
*图源:https://zhuanlan.zhihu.com/p/36494307
策略梯度(Policy Gradient)
在先前的内容中,策略用表(tabular)的形式进行表达,其也可以用函数的形式进行表达(尤其是当状态空间或动作空间连续或非常大时),优势在于降低存储开销和提升泛化能力。
之前的方法(值函数近似)称之为Value-based,而策略梯度(Policy Gradient)和Actor-Critic均为Policy-based。Value-based方法围绕状态值/动作值设计,而Policy-based优化关于策略的目标函数,从而直接得到最优策略。
Basic Policy Gradient
将策略表示为参数化函数:
π
(
a
∣
s
,
θ
)
\pi(a|s, \theta)
π(a∣s,θ),其中
θ
∈
R
m
\theta \in \mathbb{R} ^m
θ∈Rm为参数向量,
π
\pi
π是关于
θ
\theta
θ的函数。
*其他写法:
π
(
a
,
s
,
θ
)
\pi(a,s, \theta)
π(a,s,θ),
π
θ
(
a
∣
s
)
\pi_\theta(a|s)
πθ(a∣s),
π
θ
(
a
,
s
)
\pi_\theta(a,s)
πθ(a,s)
与tabular representation的区别:
- 最优策略:不是能够最大化每个状态值的策略,而是能够最大化特定scalar metrics的策略
- 动作概率:不能直接获取,需要进行计算
- 策略更新:不能直接更新,需要通过改变参数 θ \theta θ来进行改变
策略梯度方法通过优化指定目标函数
J
(
θ
)
J(\theta)
J(θ),直接得到最优策略:
θ
t
+
1
=
θ
t
+
α
∇
θ
J
(
θ
t
)
\theta_{t+1} = \theta_t + \alpha \nabla_\theta J(\theta_t)
θt+1=θt+α∇θJ(θt)
目标函数
J
(
θ
)
J(\theta)
J(θ)通常有以下两种类型:平均状态值
v
ˉ
π
\bar{v}_\pi
vˉπ和平均单步奖励
r
ˉ
π
\bar{r}_\pi
rˉπ。实际上,当奖励折扣值
γ
<
1
\gamma<1
γ<1时,二者是等价的:
r
ˉ
π
=
(
1
−
γ
)
v
ˉ
π
\bar{r}_\pi = (1- \gamma) \bar{v}_\pi
rˉπ=(1−γ)vˉπ。
目标函数1:平均状态值
平均状态值(average state value / average value):
v
ˉ
π
=
∑
s
∈
S
d
(
s
)
v
π
(
s
)
=
E
[
v
π
(
S
)
]
\bar{v}_\pi = \sum_{s\in{\mathcal{S}}} d(s) v_\pi(s) = \mathbb{E}[v_\pi(S)]
vˉπ=s∈S∑d(s)vπ(s)=E[vπ(S)]
其中,
d
(
s
)
≥
0
d(s) \geq 0
d(s)≥0且
∑
s
∈
S
d
(
s
)
=
1
\textstyle\sum_{s\in{\mathcal{S}}} d(s) =1
∑s∈Sd(s)=1,因此
d
(
s
)
d(s)
d(s)既可以看作是状态
s
s
s的权重,也可以看作是随机变量
S
S
S的概率分布。
其他形式: v ˉ π = E [ ∑ t = 0 ∞ γ t R t + 1 ] \bar{v}_\pi = \mathbb{E} \Big[\sum_{t=0}^{\infin} \gamma^t R_{t+1} \Big] vˉπ=E[∑t=0∞γtRt+1]
向量形式: v ˉ π = d T v π \bar{v}_\pi = d^T v_\pi vˉπ=dTvπ
在常见的情况下,
d
d
d是取决于
π
\pi
π的平稳分布,即
d
π
(
s
)
d_\pi(s)
dπ(s),其具有以下性质:
d
π
T
P
π
=
d
π
T
d^T_\pi P_\pi = d^T_\pi
dπTPπ=dπT
其中,
P
π
P_\pi
Pπ是状态转移概率矩阵。
目标函数2:平均单步奖励
平均单步奖励(average one-step reward / average reward)
r
ˉ
π
=
∑
s
∈
S
d
(
s
)
r
π
(
s
)
=
E
[
r
π
(
S
)
]
\bar{r}_\pi = \sum_{s\in{\mathcal{S}}} d(s) r_\pi(s) = \mathbb{E}[r_\pi(S)]
rˉπ=s∈S∑d(s)rπ(s)=E[rπ(S)]
其中,
S
∼
d
π
S \sim d_\pi
S∼dπ,
d
π
d_\pi
dπ为平稳分布。
r
π
(
s
)
=
∑
a
∈
A
π
(
a
∣
s
)
r
(
s
,
a
)
r_\pi(s) = \sum_{a\in\mathcal{A}} \pi(a|s) r(s, a)
rπ(s)=∑a∈Aπ(a∣s)r(s,a)为策略
π
\pi
π在状态
s
s
s下取得的平均单步奖励,而
r
(
s
,
a
)
=
E
[
R
∣
s
,
a
]
=
∑
r
r
p
(
r
∣
s
,
a
)
r(s, a) = \mathbb{E} [R|s, a] = \sum_r r p(r | s, a)
r(s,a)=E[R∣s,a]=∑rrp(r∣s,a)。
另一种形式:
假设agent遵循一个策略生成了奖励为
(
R
t
+
1
,
R
t
+
2
,
⋯
)
(R_{t+1}, R_{t+2}, \cdots)
(Rt+1,Rt+2,⋯)的trajectory,其平均单步奖励为:
lim
n
→
∞
1
n
E
[
∑
k
=
1
n
R
t
+
k
∣
S
t
=
s
0
]
\lim_{n\rarr\infin} \frac{1}{n} \mathbb{E} \Big[ \sum_{k=1}^{n} R_{t+k} | S_t = s_0 \Big]
limn→∞n1E[∑k=1nRt+k∣St=s0]
其中,
s
0
s_0
s0为该trajectory的起始状态。考虑无穷多步的极限,上式等价于【似乎是与平稳随机过程有关,时间平均等于统计平均,不确定】:
lim
n
→
∞
1
n
E
[
∑
k
=
1
n
R
t
+
k
]
=
r
ˉ
π
\lim_{n\rarr\infin} \frac{1}{n} \mathbb{E} \Big[ \sum_{k=1}^{n} R_{t+k} \Big] = \bar{r}_\pi
limn→∞n1E[∑k=1nRt+k]=rˉπ
🟡PG梯度计算
策略梯度方法的梯度计算可以统一总结为下式:
∇
θ
J
(
θ
)
=
∑
s
∈
S
η
(
s
)
∑
a
∈
A
∇
θ
π
(
a
∣
s
,
θ
)
q
π
(
s
,
a
)
\nabla_\theta J(\theta) = \sum_{s\in\mathcal{S}} \eta (s) \sum_{a\in\mathcal{A}} \nabla_\theta \pi (a|s, \theta) q_\pi(s, a)
∇θJ(θ)=s∈S∑η(s)a∈A∑∇θπ(a∣s,θ)qπ(s,a)
其中:
- J ( θ ) J(\theta) J(θ)可以为 v ˉ π \bar{v}_\pi vˉπ、 r ˉ π \bar{r}_\pi rˉπ或 v ˉ π 0 \bar{v}_\pi^0 vˉπ0
- = = =可以为相等、约等 ≈ \approx ≈、成比例 ∝ \propto ∝
- η \eta η是状态的分布或权重(如上文中的 d π d_\pi dπ)
进一步地,可以基于下式计算梯度:
∇
θ
J
(
θ
)
=
E
[
∇
θ
ln
π
(
A
∣
S
,
θ
)
q
π
(
S
,
A
)
]
\nabla_\theta J(\theta) = \mathbb{E} [\nabla_\theta \ln\pi (A|S, \theta) q_\pi(S, A) ]
∇θJ(θ)=E[∇θlnπ(A∣S,θ)qπ(S,A)]
其中,
S
∼
η
S\sim\eta
S∼η且
A
∼
π
(
A
∣
S
,
θ
)
A\sim\pi(A|S, \theta)
A∼π(A∣S,θ)。通过随机采样的方式估计期望,则有:
∇
θ
J
(
θ
)
≈
∇
θ
ln
π
(
A
∣
S
,
θ
)
q
π
(
S
,
A
)
\nabla_\theta J(\theta) \approx \nabla_\theta \ln\pi (A|S, \theta) q_\pi(S, A)
∇θJ(θ)≈∇θlnπ(A∣S,θ)qπ(S,A)
注意:为了计算对数
ln
\ln
ln,对所有的
s
,
a
,
θ
s, a,\theta
s,a,θ,策略必须满足:
π
(
a
∣
s
,
θ
)
>
0
\pi(a|s, \theta) > 0
π(a∣s,θ)>0。即:策略必须是随机性(stochastic)的,且为探索性(exploratory)的。(*确定性策略见后续介绍Actor-Critic的博文中的DPG)
这可以通过softmax实现,将向量从
(
−
∞
,
+
∞
)
(-\infin,+\infin)
(−∞,+∞)限界至
(
0
,
1
)
(0,1)
(0,1)。softmax限界后的形式为:
π
(
a
∣
s
,
θ
)
=
e
h
(
s
,
a
,
θ
)
∑
a
′
∈
A
e
h
(
s
,
a
′
,
θ
)
\pi(a|s, \theta) = \frac{e^{h(s, a, \theta)}}{\textstyle\sum_{a' \in \mathcal{A}} e^{h(s, a', \theta)}}
π(a∣s,θ)=∑a′∈Aeh(s,a′,θ)eh(s,a,θ)
其中,
h
(
s
,
a
,
θ
)
h(s, a, \theta)
h(s,a,θ)类似于特征函数,具体由神经网络确定。
推导:
已知 d ln x d x = 1 x \frac{\mathrm{d} \ln x}{\mathrm{d} x} = \frac{1}{x} dxdlnx=x1,则 ∇ ln f ( x ) = ∇ f ( x ) f ( x ) \nabla \ln f(x) = \frac{\nabla f(x)}{f(x)} ∇lnf(x)=f(x)∇f(x),故有: ∇ θ ln π ( a ∣ s , θ ) = ∇ θ π ( a ∣ s , θ ) π ( a ∣ s , θ ) \nabla_\theta \ln \pi(a|s, \theta) = \frac{\nabla_\theta \pi(a|s, \theta)}{\pi(a|s, \theta)} ∇θlnπ(a∣s,θ)=π(a∣s,θ)∇θπ(a∣s,θ)
进一步地, π \pi π的梯度可以计算为: ∇ θ π ( a ∣ s , θ ) = π ( a ∣ s , θ ) ∇ θ ln π ( a ∣ s , θ ) {\nabla_\theta \pi(a|s, \theta)} = {\pi(a|s, \theta)} \nabla_\theta \ln \pi(a|s, \theta) ∇θπ(a∣s,θ)=π(a∣s,θ)∇θlnπ(a∣s,θ)
🟦REINFORCE
策略梯度(PG)方法基于梯度上升方法最大化目标函数:
θ
t
+
1
=
θ
t
+
α
E
[
∇
θ
ln
π
(
A
∣
S
,
θ
t
)
q
π
(
S
,
A
)
]
\theta_{t+1} = \theta_t + \alpha \mathbb{E} \big[ \nabla_\theta \ln\pi (A|S, \theta_t) q_\pi(S, A) \big]
θt+1=θt+αE[∇θlnπ(A∣S,θt)qπ(S,A)]
实际中,通过随机采样的方式估计期望与
q
π
(
s
t
,
a
t
)
q_\pi(s_t, a_t)
qπ(st,at),有:
θ
t
+
1
=
θ
t
+
α
∇
θ
ln
π
(
a
t
∣
s
t
,
θ
t
)
q
t
(
s
t
,
a
t
)
\theta_{t+1} = \theta_t + \alpha \nabla_\theta \ln\pi (a_t|s_t, \theta_t) q_t(s_t, a_t)
θt+1=θt+α∇θlnπ(at∣st,θt)qt(st,at)
注意: A ∼ π ( A ∣ S , θ ) A\sim\pi(A|S,\theta) A∼π(A∣S,θ), a t a_t at的采样依赖于状态 s t s_t st下的策略 π ( θ t ) \pi(\theta_t) π(θt),因此策略梯度是on-policy方法。
估计 q π ( s t , a t ) q_\pi(s_t,a_t) qπ(st,at)有两种方法:
- 蒙特卡洛(MC):REINFORCE(策略梯度的代表性算法)
- 时序差分(TD):Actor-Critic系列算法(见后续博文)
REINFORCE算法步骤(伪代码):
初始化:
π
(
a
∣
s
,
θ
)
\pi(a|s, \theta)
π(a∣s,θ),
γ
∈
(
0
,
1
)
\gamma \in (0,1)
γ∈(0,1),
α
>
0
\alpha >0
α>0
目标:最大化
J
(
θ
)
J(\theta)
J(θ)
步骤:在第
k
k
k次迭代中,选择策略
π
(
θ
k
)
\pi(\theta_k)
π(θk)的起始状态
s
0
s_0
s0,设其episode为
{
s
0
,
a
0
,
r
1
,
⋯
,
s
T
−
1
,
a
T
−
1
,
r
T
}
\{ s_0, a_0, r_1, \cdots, s_{T-1}, a_{T-1}, r_T \}
{s0,a0,r1,⋯,sT−1,aT−1,rT}
- 在每个时间步
t
=
0
,
1
,
⋯
,
T
−
1
t=0,1,\cdots,T-1
t=0,1,⋯,T−1:
- 值更新(蒙特卡洛方法): q t ( s t , a t ) = ∑ k = t + 1 T γ k − t − 1 r k q_t(s_t,a_t) = \textstyle \sum_{k=t+1}^T \gamma^{k-t-1} r_k qt(st,at)=∑k=t+1Tγk−t−1rk
- 策略更新:更新参数
θ
t
+
1
\theta_{t+1}
θt+1,公式见上
- *注意:蒙特卡洛是offline的,需要整个episode的数据,所以这里更新完参数后不立即使用策略去采集数据
- θ k = θ T \theta_k = \theta_T θk=θT,在下次迭代中生成下一组episode的数据