中文

统一并提升基于梯度的免训练神经架构搜索

机器学习 2022-10-13 v2 人工智能

摘要

神经架构搜索(NAS)因其自动化神经架构设计的能力而广受欢迎。最近,人们提出了许多免训练指标,以在无需训练的情况下实现NAS,从而使NAS更具可扩展性。尽管这些免训练指标在经验上表现出竞争力,但缺乏对其统一的理论理解。因此,(a) 这些指标之间的关系尚不清楚,(b) 对其经验性能缺乏理论解释,以及 (c) 现有的免训练NAS可能存在未开发的潜力,这些潜力或许可以通过统一的理论理解来揭示。为此,本文对基于梯度的免训练NAS进行了统一的理论分析,这使我们能够 (a) 从理论上研究它们的关系,(b) 从理论上保证它们的泛化性能,以及 (c) 利用我们统一的理论理解,开发一个名为混合NAS(HNAS)的新颖框架,该框架以一种原则性的方式持续提升免训练NAS的性能。值得注意的是,HNAS可以同时享受免训练NAS(即卓越的搜索效率)和基于训练的NAS(即卓越的搜索效果)的优势,我们已通过大量实验证明了这一点。

关键词

引用

@article{arxiv.2201.09785,
  title  = {Unifying and Boosting Gradient-Based Training-Free Neural Architecture Search},
  author = {Yao Shu and Zhongxiang Dai and Zhaoxuan Wu and Bryan Kian Hsiang Low},
  journal= {arXiv preprint arXiv:2201.09785},
  year   = {2022}
}

备注

Published as a conference paper at NeurIPS 2022