4.7 Article

Simple-Encoded evolving convolutional neural network and its application to skin disease image classification

Journal

SWARM AND EVOLUTIONARY COMPUTATION
Volume 67, Issue -, Pages -

Publisher

ELSEVIER
DOI: 10.1016/j.swevo.2021.100955

Keywords

Neural architecture search; Convolutional neural networks; Encoding scheme; Genetic algorithm; Image classification

Funding

  1. National Natural Science Foundation of China [61976225, 62041208]
  2. Research and Development program in Key Areas of Hunan Province [2019SK2143]
  3. Natural Science Foundation of Changsha [Kq2014294]
  4. Foundational Research Funds for the Central Universities of Central South University [2020zzts132]

Ask authors/readers for more resources

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.

Authors

I am an author on this paper
Click your name to claim this paper and add it to your profile.

Reviews

Primary Rating

4.7
Not enough ratings

Secondary Ratings

Novelty
-
Significance
-
Scientific rigor
-
Rate this paper

Recommended

No Data Available
No Data Available