ModuleNet:知识继承的神经架构搜索

ModuleNet: Knowledge-Inherited Neural Architecture Search

IEEE Transactions on Cybernetics · 2021
被引 40
ABS 3

中文导读

提出ModuleNet算法,通过将现有卷积神经网络分解为带权重的模块构成知识库,再利用NSGA-II算法搜索新架构,无需调参即可在多个数据集上取得更好性能。

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.

计算机科学人工智能卷积神经网络神经架构搜索知识重用