一种多核处理器-单图形处理器的深度强化学习加速方法
授权
摘要
本发明提出一种多核处理器‑单图形处理器的深度强化学习加速方法,能够在CPU+GPU的平台上,建立起特别基于PPO算法的深度强化学习框架,同时发明了一种加速收集采样与推理的流水线方法。在多核CPU上实现多环境进程模拟过程,每一核上安排多个环境进程。同时CPU起到控制数据的作用。在GPU上实现神经网络模型推理过程。CPU和GPU分别保存有动作网络、评估网络参数的内存空间,在环境和智能体交互的过程中,将每次的信息保存在CPU的经验信息池内存中,每隔一定循环,CPU根据算法的截断参数和筛选条件,从自身的经验信息池内存中提取batch规模的数据进行训练。本发明通过负载平衡,当CPU单核堆叠环境进程的模拟时间与神经网络模型推理时间相同时,能够实现环境采样与动作推理并行执行的流水线结构:通过并行半个CPU模拟器和半个GPU推理的运行,并在数据传输时准备好下一次半个CPU模拟器和半个GPU推理过程,可以进一步加速强化学习速度,达到比传统方法近2倍的整体训练速度。
基本信息
专利标题 :
一种多核处理器-单图形处理器的深度强化学习加速方法
专利标题(英):
暂无
公开(公告)号 :
CN112732436A
申请号 :
CN202011476497.6
公开(公告)日 :
2021-04-30
申请日 :
2020-12-15
授权号 :
CN112732436B
授权日 :
2022-04-22
发明人 :
阮爱武朱重阳
申请人 :
电子科技大学
申请人地址 :
四川省成都市高新区(西区)西源大道2006号
代理机构 :
代理人 :
优先权 :
CN202011476497.6
主分类号 :
G06F9/50
IPC分类号 :
G06F9/50 G06T1/20 G06N3/04 G06N3/063 G06N3/08
相关图片
IPC结构图谱
G
G部——物理
G06
计算;推算或计数
G06F
电数字数据处理
G06F9/00
程序控制装置,例如,控制单元
G06F9/06
应用存入的程序的,即应用处理设备的内部存储来接收程序并保持程序的
G06F9/46
多道程序装置
G06F9/50
资源分配,例如,中央处理单元的
法律状态
2022-04-22 :
授权
2021-05-21 :
实质审查的生效
IPC(主分类) : G06F 9/50
申请日 : 20201215
申请日 : 20201215
2021-04-30 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
1、
CN112732436A.PDF
PDF下载