机器学习入门——实战篇之强化学习

这是本篇文章是《机器学习入门》系列文章的第五篇,该系列有如下文章:
《机器学习入门——基础篇》
《机器学习入门——实战篇之监督学习》
《机器学习入门——实战篇之非监督学习》
《机器学习入门——实战篇之深度学习》
《机器学习入门——实战篇之强化学习》

在《机器学习入门——基础篇》中已经介绍过强化学习的概念,也介绍了动态规划和蒙特卡洛方法。因为这两部分内容都是可以写成专著的大块头,本文不做详述,咱们直接进入实战,用一个简单的栗子来进入强化学习的世界。

有这么一个经典案例,就是“机器人走迷宫”,通过强化学习算法,让机器人自己学会寻找走出迷宫的办法,并不断优化这个路径,最后得到较优解。

在该项目中,使用强化学习算法,实现一个自动走迷宫机器人。

14-robot.png

1、如上图所示,智能机器人显示在右上角。在我们的迷宫中,有陷阱(红色炸弹)及终点(蓝色的目标点)两种情景。机器人要尽量避开陷阱、尽快到达目的地。
2、机器人可执行的动作包括:向上走 u、向右走 r、向下走 d、向左走 l。
3、执行不同的动作后,根据不同的情况会获得不同的奖励,具体而言,有以下几种情况。
撞到墙壁:-10
走到终点:50
走到陷阱:-30
其余情况:-0.1

正式开始代码之前,我们先来理一理思路,强化学习作为机器学习算法的一种,其模式是让智能体在“训练”中学到“经验”,以实现给定的任务。但不同于监督学习与非监督学习,在强化学习的框架中,我们更侧重通过让被训练的对象与环境的交互来学习。通常在监督学习和非监督学习任务中,被训练对象往往需要通过给定的训练集,辅之以既定的训练目标(如最小化损失函数),通过给定的学习算法来实现这一目标。然而在强化学习中,被训练对象则是通过其与环境交互得到的奖励进行学习。这个环境可以是虚拟的(如虚拟的迷宫),也可以是真实的(自动驾驶汽车在真实道路上收集数据)。

在强化学习中有五个核心组成部分,它们分别是:
环境(Environment)
智能体(Agent)
状态(State)
动作(Action)
奖励(Reward)

在某一时间节点 t:
智能体在从环境中感知其所处的状态: s_t
智能体根据某些准则选择动作: a_t
环境根据智能体选择的动作,向智能体反馈奖励 r_{t+1}
通过合理的学习算法,智能体将在这样的问题设置下,成功学到一个在状态s_t选择动作a_t的策略

对应“机器人走迷宫”这个题目,我们就能得到以下名词的含义:

环境 : 迷宫的各条路径,迷宫路径上的陷阱,最终迷宫的出口也就是目标位置

状态 : 为了便于将机器人走迷宫的问题抽象成一个数学上好操作的模型,我们在迷宫中加入了左边,例如(0,0)这样的点,这样机器人在任何一个时间点都会对应一个位置坐标,同时,这个坐标会结合环境具体情况带给机器人一个激励,有正向激励,也有负向激励,这些激励会直接被带入我们的算法公式,算出具体值,来刺激机器人做出下一步动作的决策;比如终点坐标带来的激励就是很大的一个正向值50,比如陷阱坐标所带来的就是一个很大的负向值-30,而如果不是什么特别的位置的坐标,也就是其余情况,就会给一个恰当的负向激励,刺激机器人继续寻找目标;以上描述就是这个模型的状态

动作 : 向上,向下,向左,向右各种方向上的移动

奖励 : 撞到墙壁:-10; 走到终点:50;走到陷阱:-30;其余情况:-0.1
这个项目中,我们采用基于 Q-Learning 的强化学习算法。Q-Learning 是一个值迭代(Value Iteration)算法。与策略迭代(Policy Iteration)算法不同,值迭代算法会计算每个”状态“或是”状态-动作“的值(Value)或是效用(Utility),然后在执行动作的时候,会设法最大化这个值。因此,对每个状态值的准确估计,是我们值迭代算法的核心。通常我们会考虑最大化动作的长期奖励,即不仅考虑当前动作带来的奖励,还会考虑动作长远的奖励。

在 Q-Learning 算法中,我们把这个长期奖励记为 Q 值,我们会考虑每个 ”状态-动作“ 的 Q 值,具体而言,它的计算公式为:

q(s_t, a) = R_{t+1} + \gamma \times maxq((a, s_t))

也就是对于当前的“状态-动作” (s_t, 𝑎) ,我们考虑执行动作 𝑎 后环境给我们的奖励 𝑅_{𝑡+1} ,以及执行动作 𝑎 到达 s_{t+1} 后,执行任意动作能够获得的最大的Q值 maxq(a,s_{t+1}) , \gamma为折扣因子。

不过一般地,我们使用更为保守地更新 Q 表的方法,即引入松弛变量 𝑎𝑙𝑝ℎ𝑎,按如下的公式进行更新,使得 Q 表的迭代变化更为平缓。

q(s_t, a) = (1-\alpha) \times q(s_t, a) + \alpha \times (R_{t+1}, \gamma \times maxq(a, s_{t+1}))

在强化学习中,「探索-利用」问题是非常重要的。具体来说,根据上面的定义,我们会尽可能地让机器人在每次选择最优的决策,来最大化长期奖励。但是这样做有如下的弊端:

在初步的学习中,我们的 Q 值会不准确,如果在这个时候都按照 Q 值来选择,那么会造成错误。

学习一段时间后,机器人的路线会相对固定,则机器人无法对环境进行有效的探索。

因此我们需要一种办法,来解决如上的问题,增加机器人的探索。由此我们考虑使用 epsilon-greedy 算法(是一种贪心算法,加入了随机因子epsilon的贪心算法),即在小车选择动作的时候,以一部分的概率随机选择动作,以一部分的概率按照最优的 Q 值选择动作。同时,这个选择随机动作的概率应当随着训练的过程逐步减小。

简单的实现以下这个贪心算法epsilon-greedy,有如下代码:

import random
# 动作列表如下
actions = ['u','r','d','l']
qline = {'u':1.2, 'r':-2.1, 'd':-24.5, 'l':27}
# epsilon以0.3的概率进行随机选择
epsilon = 0.3 
def choose_action(epsilon):
    action = None
    # 以前面定义好的一个epsilon进行选择
    if epsilon < random.random(): 
        # 随机选择一个动作
        action = random.choice(actions)  
else: 
    # 否则选择具有最大 Q 值的动作
        action = max(qline, key=qline.get) 
return action

接下来我们就可以生成迷宫啦,本文开始的时候看到的迷宫太过简单,我们给一个稍微有点难度的。

from Maze import Maze
myMaze = Maze(maze_size=(10,11), trap_number=5)

Python是不是很温馨又简单,居然有Maze这个东西。

好了,我们生成好了这么一个迷宫,准备去训练小机器人。接下来,就是最核心的部分,我们的小机器人啦,也就是要一个Robot类。

Robot类中,我们需要实现诸多功能,以使得我们成功实现一个强化学习智能体。总体来说,之前我们是人为地在环境中移动了机器人,但是现在通过实现 Robot 这个类,机器人将会自己移动。通过实现学习函数,Robot 类将会学习到如何选择最优的动作,并且更新强化学习中对应的参数。

首先 Robot 有多个输入,其中 alpha=0.5, gamma=0.9, epsilon0=0.5 表征强化学习相关的各个参数的默认值,这些在之前我们已经了解到,Maze 应为机器人所在迷宫。

随后需要一个 Robot.update 函数,它指明了在每次执行动作时,Robot 需要执行的指令。下面我们来看一看Robot这个类。

需要一个初始化函数将参数引入:

def __init__(self, maze, alpha=0.5, gamma=0.9, epsilon0=0.5):

还需要一个更新函数,每次把参数做一个更新:

def update_parameter(self):
    self.t = self.t + 1
    if self.testing:
        # 测试
        self.epsilon = self.epsilon0
    else:
        # 学习过程中更新参数,让epsilon逐渐变小
        # 从而实现参数的收敛
        if self.epsilon0 - self.t * 0.03 > 0:
            self.epsilon = self.epsilon0 - self.t * 0.03
        else:
            self.epsilon = 0.01
    return self.epsilon

然后有这么一个选择动作的函数:

def choose_action(self):
    def is_random_exploration():
        if random.random() < self.epsilon:
            return True
        else:
            return False
    if self.learning:
        if is_random_exploration():
            return self.valid_actions[random.randint(0, 3)]
        else:
            return max(self.Qtable[self.state], key = self.Qtable[self.state].get)
    elif self.testing:
        return max(self.Qtable[self.state], key = self.Qtable[self.state].get)
    else:
        return self.valid_actions[random.randint(0, 3)]

当然了,必须要有一个更新知识表的函数:

def update_Qtable(self, r, action, next_state):
    if self.learning:
        q_with_cur_state = self.Qtable[self.state][action]
        q_with_next_state = r + self.gamma * max(self.Qtable[next_state].values())
        self.Qtable[self.state][action] += self.alpha * (q_with_next_state - q_with_cur_state)

最后有一个整体的更新函数:

def update(self):
    self.state = self.sense_state() # 取得目前的状态
    self.create_Qtable_line(self.state) # 生成Qtable的记录        
    action = self.choose_action() # 选择一个动作
    reward = self.maze.move_robot(action) # 移动机器人
    next_state = self.sense_state() # 下一个状态
    self.create_Qtable_line(next_state) # 下一个记录
    if self.learning and not self.testing:
        # 更新Qtable,注意,这个表是机器人的“经验”所在
        # 初始化的时候这个表里面都设置成0
        # 随着训练的深入,可以逐步更新这个表,从而指导机器人移动
        self.update_Qtable(reward, action, next_state)
        self.update_parameter() # 更新参数
    return action, reward

上述部分并不是全部的代码,我们略去了一些小的方法,只讲了比较核心的几个函数,有心的读者可以自己去补充并实现这个项目。

在实现了上述内容之后,我们就可以开始对我们 Robot 进行训练并调参了。这里我们使用了另一个类Runner用来迭代这个模型,从而实现整个训练过程及可视化。使用如下的代码,可以成功对机器人进行训练。并且在当前文件夹中生成一个视频,记录整个训练的过程。通过观察该视频,我们能够发现训练过程中的问题,并且优化代码及参数。

# 可选的参数:
epoch = 40
epsilon0 = 0.5
alpha = 0.5
gamma = 0.9

maze_size = (10,11)
trap_number = 5

from Runner import Runner
g = Maze(maze_size=maze_size,trap_number=trap_number)
r = Robot(g,alpha=alpha, epsilon0=epsilon0, gamma=gamma)
r.set_status(learning=True, testing=False)
runner = Runner(r, g)
runner.run_training(epoch, display_direction=True)
# 我们姑且把整个训练的过程记录成process.mp4文件吧
runner.generate_movie(filename = "process.mp4") 

至此,我们这个项目的主要代码就介绍完了,希望大家可以对强化学习有一个直观的认识:)

这是本篇文章是《机器学习入门》系列文章的第五篇,该系列有如下文章:
《机器学习入门——基础篇》
《机器学习入门——实战篇之监督学习》
《机器学习入门——实战篇之非监督学习》
《机器学习入门——实战篇之深度学习》
《机器学习入门——实战篇之强化学习》

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270