arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.06138cs.LGcs.AIcs.NE
收藏

AdvantageNAS: Efficient Neural Architecture Search with Credit Assignment

University of Tsukuba(筑波大学);RIKEN AIP(理化学研究所革新智能统合研究中心)

Rei Sato, Jun Sakuma, Youhei Akimoto

英文摘要:

Neural architecture search (NAS) is an approach for automatically designing a neural network architecture without human effort or expert knowledge. However, the high computational cost of NAS limits its use in commercial applications. Two recent NAS paradigms, namely one-shot and sparse propagation, which reduce the time and space complexities, respectively, provide clues for solving this problem. In this paper, we propose a novel search strategy for one-shot and sparse propagation NAS, namely AdvantageNAS, which further reduces the time complexity of NAS by reducing the number of search iterations. AdvantageNAS is a gradient-based approach that improves the search efficiency by introducing credit assignment in gradient estimation for architecture updates. Experiments on the NAS-Bench-201 and PTB dataset show that AdvantageNAS discovers an architecture with higher performance under a limited time budget compared to existing sparse propagation NAS. To further reveal the reliabilities of AdvantageNAS, we investigate it theoretically and find that it monotonically improves the expected loss and thus converges.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑