Skip to content

15 强化学习

强化学习(Reinforcement Learning,简称RL)是机器学习的一个重要分支,前段时间人机大战的主角AlphaGo正是以强化学习为核心技术。在强化学习中,包含两种基本的元素:状态动作在某个状态下执行某种动作,这便是一种策略,学习器要做的就是通过不断地探索学习,从而获得一个好的策略。例如:在围棋中,一种落棋的局面就是一种状态,若能知道每种局面下的最优落子动作,那就攻无不克/百战不殆了。

若将状态看作为属性,动作看作为标记,易知:监督学习和强化学习都是在试图寻找一个映射,从已知属性/状态推断出标记/动作,这样强化学习中的策略相当于监督学习中的分类/回归器。但在实际问题中,强化学习并没有监督学习那样的标记信息,通常都是在尝试动作后才能获得结果,因此强化学习是通过反馈的结果信息不断调整之前的策略,从而算法能够学习到:在什么样的状态下选择什么样的动作可以获得最好的结果。

基本要素

强化学习任务通常使用马尔可夫决策过程(Markov Decision Process,简称MDP)来描述,具体而言:机器处在一个环境中,每个状态为机器对当前环境的感知;机器只能通过动作来影响环境,当机器执行一个动作后,会使得环境按某种概率转移到另一个状态;同时,环境会根据潜在的奖赏函数反馈给机器一个奖赏。综合而言,强化学习主要包含四个要素:状态、动作、转移概率以及奖赏函数。

  • 状态(X):机器对环境的感知,所有可能的状态称为状态空间;
  • 动作(A):机器所采取的动作,所有能采取的动作构成动作空间;
  • 转移概率(P):当执行某个动作后,当前状态会以某种概率转移到另一个状态;
  • 奖赏函数(R):在状态转移的同时,环境给反馈给机器一个奖赏。

因此,强化学习的主要任务就是通过在环境中不断地尝试,根据尝试获得的反馈信息调整策略,最终生成一个较好的策略π,机器根据这个策略便能知道在什么状态下应该执行什么动作。常见的策略表示方法有以下两种:

  • 确定性策略π(x)=a,即在状态x下执行a动作;
  • 随机性策略P=π(x,a),即在状态x下执行a动作的概率。

一个策略的优劣取决于长期执行这一策略后的累积奖赏,换句话说:可以使用累积奖赏来评估策略的好坏,最优策略则表示在初始状态下一直执行该策略后,最后的累积奖赏值最高。长期累积奖赏通常使用下述两种计算方法:

  • T步累积奖赏:E[1Tt=1Trt],即执行该策略T步的平均奖赏的期望值。
  • r折扣累积奖赏:E[t=0+γtrt+1],一直执行到最后,同时越往后的奖赏权重越低。

K摇摆赌博机

首先我们考虑强化学习最简单的情形:仅考虑一步操作,即在状态x下只需执行一次动作a便能观察到奖赏结果。易知:欲最大化单步奖赏,我们需要知道每个动作带来的期望奖赏值,这样便能选择奖赏值最大的动作来执行。若每个动作的奖赏值为确定值,则只需要将每个动作尝试一遍即可,但大多数情形下,一个动作的奖赏值来源于一个概率分布,因此需要进行多次的尝试。
单步强化学习实质上是K-摇臂赌博机(K-armed bandit)的原型,一般我们尝试动作的次数是有限的,那如何利用有限的次数进行有效地探索呢?这里有两种基本的想法:

  • 仅探索法: 将尝试的机会平均分给每一个动作,即轮流执行,最终将每个动作的平均奖赏作为期望奖赏的近似值。
  • 仅利用法: 将尝试的机会分给当前平均奖赏值最大的动作,隐含着让一部分人先富起来的思想。

可以看出:上述两种方法是相互矛盾的,仅探索法能较好地估算每个动作的期望奖赏,但是没能根据当前的反馈结果调整尝试策略;仅利用法在每次尝试之后都更新尝试策略,符合强化学习的思维,但容易找不到最优动作。因此需要在这两者之间进行折中。

ε-贪心

ε-贪心法基于一个概率来对探索和利用进行折中,具体而言:在每次尝试时,以ε的概率进行探索,即以均匀概率随机选择一个动作;以1ε的概率进行利用,即选择当前最优的动作。ε-贪心法只需记录每个动作的当前平均奖赏值与被选中的次数,便可以增量式更新。

输入:摇臂数K
奖赏函数R
尝试次数T
探索概率ϵ
过程:
  1: r=0
  2: i=1,2,,K:Q(i)=0,count(i)=0
  3: for t=1,2,,T do
  4: if rand()<ϵ then
  5: k=1,2,,K
  6: else
  7: k=argmaxiQ(i)
  8: end if
  9: v=R(k)
10: r=r+v
11: Q(k)=Q(k)×count(k)+vcount(k)+1增量式更新
12: count(k)=count(k)+1
13: end for
输出: 累积奖赏r
其中Q(i)为平均奖赏,count(i)为选中次数。

Softmax

Softmax算法则基于当前每个动作的平均奖赏值来对探索和利用进行折中,Softmax函数将一组值转化为一组概率,值越大对应的概率也越高,因此当前平均奖赏值越高的动作被选中的几率也越大。Softmax函数如下所示:

P(k)=eQ(k)τi=1KeQ(i)τ

其中τ表示温度,τ0越放大差距,τ越缩小差距。

输入:摇臂数K
奖赏函数R
尝试次数T
温度参数τ
过程:
  1: r=0
  2: i=1,2,,K:Q(i)=0,count(i)=0
  3: for t=1,2,,T do
  4: k=1,2,,K中根据Softmax函数随机选取
  5: v=R(k)
  6: r=r+v
  7: Q(k)=Q(k)×count(k)+vcount(k)+1
  8: count(k)=count(k)+1
  9: end for
输出:累积奖赏r

有模型学习

若学习任务中的四个要素都已知,即状态空间、动作空间、转移概率以及奖赏函数都已经给出,这样的情形称为“有模型学习”。假设状态空间和动作空间均为有限,即均为离散值,这样我们不用通过尝试便可以对某个策略进行评估。

策略评估

前面提到:在模型已知的前提下,我们可以对任意策略的进行评估(后续会给出演算过程)。一般常使用以下两种值函数来评估某个策略的优劣:

  • 状态值函数(V)V(x),即从状态x出发,使用π策略所带来的累积奖赏;
  • 状态-动作值函数(Q)Q(x,a),即从状态x出发,执行动作a后再使用π策略所带来的累积奖赏。

根据累积奖赏的定义,我们可以引入T步累积奖赏与r折扣累积奖赏:

{VTπ(x)=Eπ[1Tt=1Tτt|x0=x],T步累积奖赏Vγπ(x)=Eπ[t=0+γtτt+1|x0=x],γ折扣累积奖赏{QTπ(x,a)=Eπ[1Tt=1Tτt|x0=x,a0=a]Qγπ(x,a)=Eπ[t=0+γtrt+1|x0=x,a0=a]

由于MDP具有马尔可夫性,即现在决定未来,将来和过去无关,我们很容易找到值函数的递归关系:

T步累积奖赏:

VTπ(x)=Eπ[1Tt=1Tτt|x0=x]=Eπ[1Tτ1+T1T1T1t=2Tτt|x0=x]=aAπ(x,a)xXPxxa(1TRxxa+T1TEπ[1T1t=1T1τt|x0=x])=aAπ(x,a)xXPxxa(1TRxxa+T1TVT1π(x))

类似地,对于r折扣累积奖赏可以得到:

Vγπ(x)=aAπ(x,a)xXPxxa(Rxxa+γVγπ(x))

易知:当模型已知时,策略的评估问题转化为一种动态规划问题,即以填表格的形式自底向上,先求解每个状态的单步累积奖赏,再求解每个状态的两步累积奖赏,一直迭代逐步求解出每个状态的T步累积奖赏。算法流程如下所示:

输入:MDP四元组E=X,A,P,R
被评估的策略π
累积奖赏参数T
  1: xX:V(x)=0即最优一个状态的值函数,由于不再执行动作/转移,因此值函数为0
  2: for t=1,2, do
  3: xX:V(x)=aAπ(x,a)xXPxxa(1tRxxa+t1tV(x))
  4: if t=T+1 then
  5: break
  6: else
  7: V=V记录每个状态的t步累积奖赏值
  8: end if
  9: end for
输出:状态值函数V

对于状态-动作值函数,只需通过简单的转化便可得到:

{QTπ(x,a)=xXPxxa(1TRxxa+T1TVT1π(x))Qγπ(x,a)=xXPxxa(Rxxa+γVγπ(x))

策略改进

理想的策略应能使得每个状态的累积奖赏之和最大,简单来理解就是:不管处于什么状态,只要通过该策略执行动作,总能得到较好的结果。因此对于给定的某个策略,我们需要对其进行改进,从而得到最优的值函数

{VT(x)=maxaAxXPxxa(1TRxxa+T1TVT1(x))Vγ(x)=maxaAxXPxxa(Rxxa+γVγ(x))

选择当前最优的动作

{QT(x,a)=xXPxxa(1TRxxa+T1TmaxaAQT1(x,a))Qγ(x,a)=xXPxxa(Rxxa+γmaxaAQγ(x,a))

最优Bellman等式改进策略的方式为:将策略选择的动作改为当前最优的动作,而不是像之前那样对每种可能的动作进行求和。易知:选择当前最优动作相当于将所有的概率都赋给累积奖赏值最大的动作,因此每次改进都会使得值函数单调递增。$$\displaystyle \pi'(x)=\underset{a \in A}{\arg \max } Q^{\pi}(x, a)$$

将策略评估与策略改进结合起来,我们便得到了生成最优策略的方法:先给定一个随机策略,现对该策略进行评估,然后再改进,接着再评估/改进一直到策略收敛、不再发生改变。这便是策略迭代算法,算法流程如下所示:

输入:MDP四元组E=X,A,P,R
累积奖赏参数T
过程:
  1: xX:V(x)=0,π(x,a)=1|A(x)|随机策略
  2: loop
  3: for t=1,2, do
  4: xX:V(x)=aAπ(x,a)xXPxxa(1tRxxa+t1tV(x))使用动态规划法计算
  5: if t=T+1 then
  6: break
  7: else
  8: V=V
  9: end if
10: end for得到当前策略的T步状态值函数
11: xX:π(x)=argmaxaAQ(x,a)依据状态值函数更新策略
12: if x:π(x)=π(x) then
13: break
14: else
15: π=π
16: end for
17: end loop
输出:最优策略π

可以看出:策略迭代法在每次改进策略后都要对策略进行重新评估,因此比较耗时。若从最优化值函数的角度出发,即先迭代得到最优的值函数,再来计算如何改变策略,这便是值迭代算法,算法流程如下所示:

输入:MDP四元组E=X,A,P,R
累积奖赏参数T
收敛阈值θ
过程:
  1: xX:V(x)=0
  2: for t=1,2, do
  3: xX:V(x)=maxaAxXPxxa(1tRxxa+t1tV(x))每次都选择最优动作
  4: if maxxX|V(x)V(x)|<θ then
  5: break
  6: else
  7: V=V
  8: end if
  9: end for得到了最优的值函数
输出:策略π(x)=argmaxaAQ(x,a)根据最优值函数阿里改变策略

蒙特卡罗强化学习

在现实的强化学习任务中,环境的转移函数与奖赏函数往往很难得知,因此我们需要考虑在不依赖于环境参数的条件下建立强化学习模型,这便是免模型学习。蒙特卡罗强化学习便是其中的一种经典方法。
由于模型参数未知,状态值函数不能像之前那样进行全概率展开,从而运用动态规划法求解。一种直接的方法便是通过采样来对策略进行评估/估算其值函数,蒙特卡罗强化学习正是基于采样来估计状态-动作值函数:对采样轨迹中的每一对状态-动作,记录其后的奖赏值之和,作为该状态-动作的一次累积奖赏,通过多次采样后,使用累积奖赏的平均作为状态-动作值的估计,并引入ε-贪心策略保证采样的多样性

输入:环境E
动作空间A
起始状态x0
策略执行步数T
过程:
  1: Q(x,a)=0,count(x,a)=0,π(x,a)=1|A(x)|随机策略)   2: for s=1,2, do
  3: 在E中执行策略π产生轨迹:<x0,a0,r1,x1,a1,r2,,xT1,aT1,rT,xT>
  4: for t=0,1,,T1 do
  5: R=1Tti=t+1Tri状态-动作的Tt步累积奖赏
  6: Q(xt,at)=Q(xt,at)×count(xt,at)+Rcount(xt,at)+1更新状态-动作值函数
  7: count(xt,at)=count(xt,at)+1
  8: end for
  9: 对所有已见状态xπ(x,a)={argmaxaQ(x,a),1ϵA,ϵ
10: end for
输出:策略π

在上面的算法流程中,被评估和被改进的都是同一个策略,因此称为同策略蒙特卡罗强化学习算法。引入ε-贪心仅是为了便于采样评估,而在使用策略时并不需要ε-贪心,那能否仅在评估时使用ε-贪心策略,而在改进时使用原始策略呢?这便是异策略蒙特卡罗强化学习算法

输入:环境E
动作空间A
起始状态x0
策略执行步数T
过程:
  1: Q(x,a)=0,count(x,a)=0,π(x,a)=1|A(x)|
  2: for s=1,2, do
  3: 在E中执行πϵ-贪心策略产生轨迹:<x0,a0,r1,x1,a1,r2,,xT1,aT1,rT,xT>
  4: pi={1ϵ+ϵ/|A|,ai=π(x);()ϵ/|A|,aiπ(x),()
  5:for t=0,1,,T1 do
  6:R=1Tti=t+1T(ri×j=iT11pj)修正的累积奖赏
  7:Q(xt,at)=Q(xt,at)×count(xt,at)+Rcount(xt,at)+1
  8:count(xt,at)=count(xt,at)+1
  9:end for
10: π(x)=argmaxaQ(x,a)根据估计的状态-动作值函数改变策略
11: end for
输出:策略π

AlphaGo原理浅析

本篇一开始便提到强化学习是AlphaGo的核心技术之一,刚好借着这个东风将AlphaGo的工作原理了解一番。正如人类下棋那般“手下一步棋,心想三步棋”,Alphago也正是这个思想,当处于一个状态时,机器会暗地里进行多次的尝试/采样,并基于反馈回来的结果信息改进估值函数,从而最终通过增强版的估值函数来选择最优的落子动作。
其中便涉及到了三个主要的问题:(1)如何确定估值函数(2)如何进行采样(3)如何基于反馈信息改进估值函数,这正对应着AlphaGo的三大核心模块:深度学习蒙特卡罗搜索树强化学习

1. 深度学习(拟合估值函数)
由于围棋的状态空间巨大,像蒙特卡罗强化学习那样通过采样来确定值函数就行不通了。在围棋中,状态值函数可以看作为一种局面函数,状态-动作值函数可以看作一种策略函数,若我们能获得这两个估值函数,便可以根据这两个函数来完成:(1)衡量当前局面的价值;(2)选择当前最优的动作。那如何精确地估计这两个估值函数呢?这就用到了深度学习,通过大量的对弈数据自动学习出特征,从而拟合出估值函数。

2. 蒙特卡罗搜索树(采样)
蒙特卡罗树是一种经典的搜索框架,它通过反复地采样模拟对局来探索状态空间。具体表现在:从当前状态开始,利用策略函数尽可能选择当前最优的动作,同时也引入随机性来减小估值错误带来的负面影响,从而模拟棋局运行,使得棋盘达到终局或一定步数后停止。

3. 强化学习(调整估值函数)
在使用蒙特卡罗搜索树进行多次采样后,每次采样都会反馈后续的局面信息(利用局面函数进行评价),根据反馈回来的结果信息自动调整两个估值函数的参数,这便是强化学习的核心思想,最后基于改进后的策略函数选择出当前最优的落子动作。