Efficient Accelerator/Network Co-Search with Circular Greedy Reinforcement Learning
文献类型:期刊论文
作者 | Liu, Zejian2,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 |
DOI | 10.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. |
入库方式: OAI收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。