Zero-Shot Proxy with Incorporated-Score for Lightweight Deep Neural Architecture Search

Thi Trang Nguyen, Ji Hyeong Han

Research output: Contribution to journalArticlepeer-review

Abstract

Designing a high-performance neural network is a difficult task. Neural architecture search (NAS) methods aim to solve this process. However, the construction of a high-quality accuracy predictor, which is a key component of NAS, usually requires significant computation. Therefore, zero-shot proxy-based NAS methods have been actively and extensively investigated. In this work, we propose a new efficient zero-shot proxy, Incorporated-Score, to rank deep neural network architectures instead of using an accuracy predictor. The proposed Incorporated-Score proxy is generated by incorporating the zen-score and entropy information of the network, and it does not need to train any network. We then introduce an optimal NAS algorithm called Incorporated-NAS that targets the maximization of the Incorporated-Score of the neural network within the specified inference budgets. The experiments show that the network designed by Incorporated-NAS with Incorporated-Score outperforms the previously proposed Zen-NAS and achieves a new SOTAaccuracy on the CIFAR-10, CIFAR-100, and ImageNet datasets with a lightweight scale.

Original languageEnglish
Article number3325
JournalElectronics (Switzerland)
Volume13
Issue number16
DOIs
StatePublished - Aug 2024

Keywords

  • efficient model
  • neural architecture search (NAS)
  • zero-shot NAS

Fingerprint

Dive into the research topics of 'Zero-Shot Proxy with Incorporated-Score for Lightweight Deep Neural Architecture Search'. Together they form a unique fingerprint.

Cite this