Open Access Te Herenga Waka-Victoria University of Wellington
Browse

Completely Automated CNN Architecture Design Based on Blocks

Download (2.9 MB)
journal contribution
posted on 2020-10-29, 00:53 authored by Y Sun, Bing XueBing Xue, Mengjie ZhangMengjie Zhang, GG Yen
© 2019 IEEE. 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.

History

Preferred citation

Sun, Y., Xue, B., Zhang, M. & Yen, G. G. (2020). Completely Automated CNN Architecture Design Based on Blocks. IEEE Transactions on Neural Networks and Learning Systems, 31(4), 1242-1254. https://doi.org/10.1109/TNNLS.2019.2919608

Journal title

IEEE Transactions on Neural Networks and Learning Systems

Volume

31

Issue

4

Publication date

2020-04-01

Pagination

1242-1254

Publisher

Institute of Electrical and Electronics Engineers (IEEE)

Publication status

Published

ISSN

2162-237X

eISSN

2162-2388

Language

en

Usage metrics

    Journal articles

    Categories

    No categories selected

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC