4.7 Article

Completely Automated CNN Architecture Design Based on Blocks

Journal

Publisher

IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/TNNLS.2019.2919608

Keywords

Automatic architecture design; convolutional neural networks (CNNs); evolutionary deep learning; genetic algorithms (GAs); neural networks

Funding

  1. National Natural Science Foundation of China [61803277]
  2. Fundamental Research Funds for the Central Universities
  3. National Natural Science Fund of China for Distinguished Young Scholar [61625204]
  4. Marsden Fund of New Zealand Government [VUW1209, VUW1509, VUW1615]
  5. Huawei Industry Fund [E2880/3663]
  6. University Research Fund at the Victoria University of Wellington [209862/3580, 213150/3662]

Ask authors/readers for more resources

The performance of convolutional neural networks (CNNs) highly relies on their architectures. In order to design a CNN with promising performance, extensive expertise in both CNNs and the investigated problem domain is required, which is not necessarily available to every interested user. To address this problem, we propose to automatically evolve CNN architectures by using a genetic algorithm (GA) based on ResNet and DenseNet blocks. The proposed algorithm is completely automatic in designing CNN architectures. In particular, neither preprocessing before it starts nor postprocessing in terms of CNNs is needed. Furthermore, the proposed algorithm does not require users with domain knowledge on CNNs, the investigated problem, or even GAs. The proposed algorithm is evaluated on the CIFAR10 and CIFAR100 benchmark data sets against 18 state-of-the-art peer competitors. Experimental results show that the proposed algorithm outperforms the state-of-the-art CNNs hand-crafted and the CNNs designed by automatic peer competitors in terms of the classification performance and achieves a competitive classification accuracy against semiautomatic peer competitors. In addition, the proposed algorithm consumes much less computational resource than most peer competitors in finding the best CNN architectures.

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