期刊
SWARM AND EVOLUTIONARY COMPUTATION
卷 67, 期 -, 页码 -出版社
ELSEVIER
DOI: 10.1016/j.swevo.2021.100955
关键词
Neural architecture search; Convolutional neural networks; Encoding scheme; Genetic algorithm; Image classification
资金
- National Natural Science Foundation of China [61976225, 62041208]
- Research and Development program in Key Areas of Hunan Province [2019SK2143]
- Natural Science Foundation of Changsha [Kq2014294]
- Foundational Research Funds for the Central Universities of Central South University [2020zzts132]
This paper introduces a genetic algorithm with a simple encoding scheme (SEECNN) for evolving CNNs to address image classification problems, and achieves effective results through a stable search strategy to update the population.
Automatically searched architectures by neural architecture search (NAS) methods have shown promising performance in various visual recognition tasks. Among NAS methods, evolutionary neural architecture methods are popular because of their potential to find the global optimal convolutional neural networks (CNNs). These methods usually use an individual to represent a CNN, while often facing two challenges: 1) since a CNN has numerous encoded parameters and weights, the length of each individual is long, which causes a large search space, and 2) due to the unknown optimal depth of a CNN, it is necessary to deal with a variable-length optimization problem, which leads to the search in a messy way since search spaces with different dimensions may have different optimal solutions. In this paper, we propose a genetic algorithm with a simple encoding scheme (SEECNN) for evolving CNNs to address image classification problems. In our encoding scheme, the parameters and weights of each layer are encoded into an individual and the whole population represents an entire CNN. Over the course of evolution, three offspring subpopulations are separately produced by genetic operators on three subpopulations. In each subpopulation, the lengths of individuals are short and equal. Afterward, we design a stable search strategy to update the population based on the performance improvement, where we only insert, replace, and remove one individual to generate candidate populations. SEECNN is compared with 24 well-known algorithms on nine benchmark datasets and five state-of-the-art methods on a real-world skin disease image classification case. The results demonstrate its effectiveness.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据