Skip to main content
Article
Neural architecture search as sparse supernet
The Proceedings of 35th AAAI Conference on Artificial Intelligence, 2021 Feb 2-9
  • Y. WU
  • A. LIU
  • Zhiwu HUANG, Singapore Management University
  • S. ZHANG
  • Gool L. VAN
Publication Type
Conference Proceeding Article
Version
publishedVersion
Publication Date
2-2021
Abstract

This paper aims at enlarging the problem of Neural Architecture Search (NAS) from Single-Path and Multi-Path Search to automated Mixed-Path Search. In particular, we model the NAS problem as a sparse supernet using a new continuous architecture representation with a mixture of sparsity constraints. The sparse supernet enables us to automatically achieve sparsely-mixed paths upon a compact set of nodes. To optimize the proposed sparse supernet, we exploit a hierarchical accelerated proximal gradient algorithm within a bi-level optimization framework. Extensive experiments on Convolutional Neural Network and Recurrent Neural Network search demonstrate that the proposed method is capable of searching for compact, general and powerful neural architectures.

Publisher
AAAI Press
City or Country
Virtual
Creative Commons License
Creative Commons Attribution-NonCommercial-No Derivative Works 4.0 International
Citation Information
Y. WU, A. LIU, Zhiwu HUANG, S. ZHANG, et al.. "Neural architecture search as sparse supernet" The Proceedings of 35th AAAI Conference on Artificial Intelligence, 2021 Feb 2-9 (2021) p. 10379 - 10387
Available at: http://works.bepress.com/alanliu/154/