multiprocessing与threading模块相关踩坑记录

Background


多线程/多进程/分布式编程在深度学习/强化学习的应用中是很常见的问题,本文的问题就是在实现DPPO的时候遇到的。

在开始复现这个算法之前我已经参考运行了Morvan大神的demo实现,这份代码的可读性非常棒,它的并行是按照读者-写者模式执行的,其中模型更新的master线程是读者,收集数据的worker线程是写者,双方的操作严格互斥:写者与环境交互得到训练数据放入队列,读者从队列中取出数据进行训练。后续测试中发现代码虽然确实实现了并行,然而运行效率并不是很高,运行时的CPU利用率始终保持在一个比较低的水平,经分析后原因主要有以下几点

  • Python GIL的限制
  • 对于读者而言,每次执行完一次更新后都会触发同步;对于写者而言,每次收集到一个batch的数据后也都会触发同步,因此很多时间会浪费在操作系统/线程级别的IO上
  • 与A3C的实现不同,每次一个worker获取到数据时,不管其他worker处于什么状态,它们的buffer都会被清空,这就导致不管你开了多少个worker,最终只会有一个worker收集到的数据有效并传递给master线程。按照我个人的理解,这样实现的目的在于维持训练的稳定性——每当一个worker推送数据给master时,master都会进行模型参数的更新,而更新后的policy已经不是其他worker收集数据时的policy了,由于PPO方法只能用于on-policy,这部分数据理论上来讲应当舍弃

那么既然在异步调度中会有这么多限制,首先一个问题,是否可以抛弃一部分理论上的严谨性,把程序实现变成纯粹异步的?

答案是否定的,原因在于模型更新这一步无论如何都必须要进行同步,否则如果模型正在更新参数的时候worker运行,那么worker得到的trajectories就会是脏数据,因为这些trajectories从概率分布上讲既不服从旧的policy distribution,也不服从更新后的policy distribution。

那么进一步,是否可以只对于模型更新操作进行同步,剩余操作全部异步呢?

理论上来说似乎是可行的,然而后续的实验中发现,由于模型更新速度比worker收集trajectory快,大部分时间里master都会抢占掉锁,全局队列中的元素长期很少,这反而使得程序在操作系统/线程级别的IO上花费了更多的时间效率。

因此我最后选择了一个折中的方案,并将这份代码改成了自己的风格,这里总结下修改过的地方

  • 设置一个队列大小的上限阈值MAX_QSIZE,同步操作仅发生在队列大小达到上限或队列为空时
    • 当队列大小达到上限,阻塞worker,进行模型更新直到队列为空
    • 当队列为空,阻塞master,启动所有worker异步收集数据
  • 经验上来讲,PPO本就是TRPO的近似,而TRPO方法中每步更新的KL divergence upper bound是有理论保障的,因此每步更新policy distribution不会有太大变化,每次模型参数更新后可以不清空其他worker的buffer
  • 考虑Python GIL的问题根深蒂固,用multiprocessing代替threading模块是更好的选择
  • 子线程/进程的运行不阻塞主线程/进程,主线程实时进行evaluation和render
  • 实例化一个Event类成员来管理训练的迭代停止,防止程序无法正常结束的情况

multiprocessing

Python的multiprocessing库提供了与threading非常接近的API,且是由强变量类型的Python实现的,非常人性化,以下是几种使用multiprocessing创建进程的方式

简单进程的创建

import multiprocessing

def worker(num):
    """thread worker function"""
    print 'Worker:', num
    return

if __name__ == '__main__':
    jobs = []
    for i in range(5):
        p = multiprocessing.Process(target=worker, args=(i,))
        jobs.append(p)
        p.start()

继承派生

import multiprocessing

class Worker(multiprocessing.Process):
    def __init(self):
        super(Worker, self).__init__()

    def run(self):
        print 'In %s' % self.name
        return

if __name__ == '__main__':
    jobs = []
    for i in range(5):
        p = Worker()
        jobs.append(p)
        p.start()
    for j in jobs:
        j.join()

即使如此,如果你认为可以用与多线程并行的相同方式实现多进程并行,那将是调bug噩梦的开始。

<img src="http://img.99danji.com/uploadfile/2016/0419/20160419034745372.jpg" width="200px;">

所谓基础不牢地动山摇,如果你不明白其中原因,请重复仔细阅读下面这两句话:

线程是操作系统调度的最小单位,进程是操作系统中资源分配的最小单位

换个说法

线程之间资源可以共享,进程则不然

  • 具体来说,如果每个子进程执行需要消耗的时间非常短,则不必使用多进程,因为进程的启动关闭也会耗费资源
  • 使用多进程往往是用来处理CPU密集型的需求,如果是IO密集型则使用多线程去处理更加合适

threading

Our approach

References

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

推荐阅读更多精彩内容