计算机研究与发展2023,Vol.60Issue(6) :1373-1384.DOI:10.7544/issn1000-1239.202111186

A3C深度强化学习模型压缩及知识抽取

A3C Deep Reinforcement Learning Model Compression and Knowledge Extraction

张晶 王子铭 任永功
计算机研究与发展2023,Vol.60Issue(6) :1373-1384.DOI:10.7544/issn1000-1239.202111186

A3C深度强化学习模型压缩及知识抽取

A3C Deep Reinforcement Learning Model Compression and Knowledge Extraction

张晶 1王子铭 1任永功1
扫码查看

作者信息

  • 1. 辽宁师范大学计算机与人工智能学院 辽宁大连 116081
  • 折叠

摘要

异步优势演员评论家(asynchronous advantage actor-critic,A3C)构建一主多从异步并行深度强化学习框架,其在最优策略探索中存在求解高方差问题,使主智能体难以保证全局最优参数更新及最佳策略学习.同时,利用百万计算资源构建的大规模并行网络,难以部署低功耗近端平台.针对上述问题,提出紧凑异步优势演员评论家(Compact_A3C)模型,实现模型压缩及知识抽取.该模型冻结并评价A3C框架中所有子智能体学习效果,将评价结果转化为主智能体更新概率,保证全局最优策略获取,提升大规模网络资源利用率.进一步,模型将优化主智能体作为"教师网络",监督小规模"学生网络"前期探索与策略引导,并构建线性衰减损失函数鼓励"学生网络"对复杂环境自由探索,强化自主学习能力,实现大规模A3C模型知识抽取及网络压缩.建立不同压缩比"学生网络",在流行Gym Classic Control与Atari 2600环境中达到了与大规模"教师网络"一致的学习效果.模型代码公布在https://github.com/meadewaking/Compact_A3C.

关键词

强化学习/深度强化学习/演员评论家模型/异步优势演员评论家模型/模型压缩

引用本文复制引用

基金项目

国家自然科学基金(61902165)

国家自然科学基金(61976109)

大连市科技创新基金(2018J12GX047)

大连市重点实验室专项()

出版年

2023
计算机研究与发展
中国科学院计算技术研究所 中国计算机学会

计算机研究与发展

CSTPCDCSCD北大核心
影响因子:2.649
ISSN:1000-1239
被引量1
参考文献量2
段落导航相关论文