Automatically Designing CNN Architectures Using the Genetic Algorithm for Image Classification

Convolutional neural networks (CNNs) have gained remarkable success on many image classification tasks in recent years. However, the performance of CNNs highly relies upon their architectures. For the most state-of-the-art CNNs, their architectures are often manually designed with expertise in both CNNs and the investigated problems. Therefore, it is difficult for users, who have no extended expertise in CNNs, to design optimal CNN architectures for their own image classification problems of interest. In this article, we propose an automatic CNN architecture design method by using genetic algorithms, to effectively address the image classification tasks. The most merit of the proposed algorithm remains in its…

• NN
  National Natural Science Foundation of China

  Awards: 61803277, 61625204
• FR
  Fundamental Research Funds for the Central Universities