强化学习基础篇(三十一)策略梯度(3)Actor-Critic算

2020-11-06  本文已影响0人  Jabes

强化学习基础篇(三十一)策略梯度(3)Actor-Critic算法

1.引入Baseline

在使用策略梯度方法更新过程中,降低方差的另一种方法是使用baseline。

在REINFORCE算法得到的更新方式为:
\nabla_{\theta} \mathbb{E}_{\tau \sim \pi_{\theta}}[R]=\mathbb{E}_{\tau}\left[\sum_{t=0}^{T-1} G_{t} \cdot \nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right)\right]
其中的G_{t}=\sum_{t^{\prime}=t}^{T-1} r_{t}是由轨迹产生的回报,具有很高的方差,如果考虑其上减去一个baseline b(s)
\nabla_{\theta} \mathbb{E}_{\tau \sim \pi_{\theta}}[R]=\mathbb{E}_{\tau}\left[\sum_{t=0}^{T-1}\left(G_{t}-b\left(s_{t}\right)\right) \cdot \nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right)\right]
一般而言,baseline的选择可以是回报的期望:
b\left(s_{t}\right)=\mathbb{E}\left[r_{t}+r_{t+1}+\ldots+r_{T-1}\right]
Baseline的引入可以降低方差,但是有baseline不含有参数\theta,所以不会改变更新过程的梯度:
\mathbb{E}_{\tau}\left[\nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right) b\left(s_{t}\right)\right]=0

E_{\tau}\left[\nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right)\left(G_{t}-b\left(s_{t}\right)\right)\right]=E_{\tau}\left[\nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right) G_{t}\right]

\operatorname{Var}_{\tau}\left[\nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right)\left(G_{t}-b\left(s_{t}\right)\right)\right]<\operatorname{Var}_{\tau}\left[\nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right) G_{t}\right]

这里的baseline的选择还可以是一个另一个被w参数化的函数。
\nabla_{\theta} J(\theta)=\mathbb{E}_{\tau}\left[\sum_{t=0}^{T-1}\left(G_{t}-b_{w}\left(s_{t}\right)\right) \cdot \nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right)\right]

2、Vanilla Policy Gradient算法

通过加入baseline,我们可以得到Vanilla Policy Gradient算法:

image.png

3、使用Critic降低方差

在实际中\nabla_{\theta} J(\theta)=\mathbb{E}_{\pi_{\theta}}\left[\sum_{t=0}^{T-1} G_{t} \cdot \nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right)\right]更新过程的G_t可以使用动作值函数代替Q^{\pi_{\theta}}\left(s_{t}, a_{t}\right),动作值函数作为Critic可以由参数化的函数近似:
Q_{w}(s, a) \approx Q^{\pi_{\theta}}(s, a)
所以策略梯度更新可以修改为:
\nabla_{\theta} J(\theta)=\mathbb{E}_{\pi_{\theta}}\left[\sum_{t=0}^{T-1} Q_{w}\left(s_{t}, a_{t}\right) \cdot \nabla_{\theta} \log \pi_{\theta}\left(a_{t} \mid s_{t}\right)\right]
这样就可以形成Actor-Critic算法,其中:

如果使用线性函数进行Q函数的近似Q_{w}(s, a)=\psi(s, a)^{T} \mathbf{w},然后使用TD(0)的方法更新Critic的参数w,使用PG更新Actor的参数\theta,这样就有简单的QAC算法:

image.png

4、Actor-Critc函数近似

在AC算法中,我们需要维护两组参数,在实现过程中可以由两种网络的设计,一种是分别使用神经网络拟合两组参数,第一组输出价值函数,第二组输出策略。

image.png

另一种方法是让两个输出共享同一个网络:

image.png

5、使用Baseline降低AC的方差

我们到Q函数的形式为:
Q^{\pi, \gamma}(s, a)=\mathbb{E}_{\pi}\left[r_{1}+\gamma r_{2}+\ldots \mid s_{1}=s, a_{1}=a\right]
价值函数为:
\begin{aligned} V^{\pi, \gamma}(s) &=\mathbb{E}_{\pi}\left[r_{1}+\gamma r_{2}+\ldots \mid s_{1}=s\right] \\ &=\mathbb{E}_{a \sim \pi}\left[Q^{\pi, \gamma}(s, a)\right] \end{aligned}
如果将价值函数作为一个baseline,可以定义优势函数如下:
A^{\pi, \gamma}(s, a)=Q^{\pi, \gamma}(s, a)-V^{\pi, \gamma}(s)
这样使用Advantage funtion的策略梯度就为:
\nabla_{\theta} J(\theta)=\mathbb{E}_{\pi_{\theta}}\left[\nabla_{\theta} \log \pi_{\theta}(s, a) A^{\pi, \gamma}(s, a)\right]

使用N-step 近似

我们之前使用的是MC的回报G_t,但也可以使用TD的方法进行更新,或者n-step方法进行更新:

比如:
\begin{array}{rl}n=1(T D) & G_{t}^{(1)}=r_{t+1}+\gamma v\left(s_{t+1}\right) \\ n=2 & G_{t}^{(2)}=r_{t+1}+\gamma r_{t+2}+\gamma^{2} v\left(s_{t+2}\right) \\ n=\infty(M C) & G_{t}^{(\infty)}=r_{t+1}+\gamma r_{t+2}+\ldots+\gamma^{T-t-1} r_{T}\end{array}
使用了n-step方法的优势函数可以为:
\begin{aligned} \hat{A}_{t}^{(1)} &=r_{t+1}+\gamma v\left(s_{t+1}\right)-v\left(s_{t}\right) \\ \hat{A}_{t}^{(2)} &=r_{t+1}+\gamma r_{t+2}+\gamma^{2} v\left(s_{t+2}\right)-v\left(s_{t}\right) \\ \hat{A}_{t}^{(\infty)} &=r_{t+1}+\gamma r_{t+2}+\ldots+\gamma^{T-t-1} r_{T}-v\left(s_{t}\right) \end{aligned}
这里\hat A^{(1)}具有低variance,但是高的bias,相反\hat{A}_{t}^{(\infty)}具有高variance,但是低的bias。

上一篇下一篇

猜你喜欢

热点阅读