Efficient Accelerator/Network Co-Search with Circular Greedy Reinforcement Learning
Liu, Zejian2,3; Li, Gang1; Cheng, Jian2,3
刊名IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS II: EXPRESS BRIEFS
2023-02
页码1-5
关键词Accelerator/Network Co-Search Reinforcement Learning Performance Estimation Multi-objective Optimization
DOI10.1109/TCSII.2023.3237912
英文摘要

Recently, accelerator/network co-search has shown great promise in reducing the complexity of co-design and achieving higher model accuracy. Unlike the manual design methodology, it automatically learns the optimal network architecture and corresponding accelerator under the given constraints. However, prior works do not consider the direct feedback of searched accelerators on the network search in each step, which leads to low converge speed and sub-optimal solutions. To address this issue, we propose DAN, a reinforcement learning-based framework for fast and accurate accelerator/network co-search. The fundamental idea is to model the co-search as interleaved network-aware accelerator search (AS) and accelerator-aware network search (NS) using separate RL agents, which improves the performance of AS and NS, and encourages a tight interaction between AS and NS. Experimental results show that our proposed method consistently outperforms single-agent based method in terms of converge speed and performance.

语种英语
内容类型期刊论文
源URL[http://ir.ia.ac.cn/handle/173211/52035]  
专题类脑芯片与系统研究
通讯作者Cheng, Jian
作者单位1.The Department of Computer Science and Engineering, Shanghai Jiao Tong University
2.The School of Future Technology, University of Chinese Academy of Sciences
3.The Institute of Automation, Chinese Academy of Sciences
推荐引用方式
GB/T 7714
Liu, Zejian,Li, Gang,Cheng, Jian. Efficient Accelerator/Network Co-Search with Circular Greedy Reinforcement Learning[J]. IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS II: EXPRESS BRIEFS,2023:1-5.
APA Liu, Zejian,Li, Gang,&Cheng, Jian.(2023).Efficient Accelerator/Network Co-Search with Circular Greedy Reinforcement Learning.IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS II: EXPRESS BRIEFS,1-5.
MLA Liu, Zejian,et al."Efficient Accelerator/Network Co-Search with Circular Greedy Reinforcement Learning".IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS II: EXPRESS BRIEFS (2023):1-5.
个性服务
查看访问统计
相关权益政策
暂无数据
收藏/分享
所有评论 (0)
暂无评论
 

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。


©版权所有 ©2017 CSpace - Powered by CSpace