ModuleNet: Knowledge-Inherited Neural Architecture Search

Yaran Chen, Ruiyuan Gao, Fenggang Liu, Dongbin Zhao*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

23 Citations (Scopus)

Abstract

Although neural the architecture search (NAS) can bring improvement to deep models, it always neglects precious knowledge of existing models. The computation and time costing property in NAS also means that we should not start from scratch to search, but make every attempt to reuse the existing knowledge. In this article, we discuss what kind of knowledge in a model can and should be used for a new architecture design. Then, we propose a new NAS algorithm, namely, ModuleNet, which can fully inherit knowledge from the existing convolutional neural networks. To make full use of the existing models, we decompose existing models into different modules, which also keep their weights, consisting of a knowledge base. Then, we sample and search for a new architecture according to the knowledge base. Unlike previous search algorithms, and benefiting from inherited knowledge, our method is able to directly search for architectures in the macrospace by the NSGA-II algorithm without tuning parameters in these modules. Experiments show that our strategy can efficiently evaluate the performance of a new architecture even without tuning weights in convolutional layers. With the help of knowledge we inherited, our search results can always achieve better performance on various datasets (CIFAR10, CIFAR100, and ImageNet) over original architectures.

Original languageEnglish
Pages (from-to)11661-11671
Number of pages11
JournalIEEE Transactions on Cybernetics
Volume52
Issue number11
DOIs
Publication statusPublished - 1 Nov 2022

Keywords

  • Evaluation algorithm
  • knowledge inherited
  • neural architecture search (NAS)

Fingerprint

Dive into the research topics of 'ModuleNet: Knowledge-Inherited Neural Architecture Search'. Together they form a unique fingerprint.

Cite this