Published January 1, 2023
| Version v1
Journal article
Open
Fully automatic CNN design with inception and ResNet blocks
Creators
- 1. Nazarbayev Univ, Sch Engn & Digital Sci, Dept Comp Sci, Nur Sultan, Kazakhstan
Description
Although convolutional neural networks (CNNs) are widely used in image classification tasks and have demonstrated promising classification accuracy results, designing a CNN architecture requires a manual adjustment of parameters through a series of experiments as well as sufficient knowledge both in the problem domain and CNN architecture design. Therefore, it is difficult for users without prior experience to design a CNN for specific purposes. In this paper, we propose a framework for the automatic construction of CNN architectures based on ResNet, DenseNet, and Inception blocks and the roulette wheel selection method with a dynamic learning rate. Compared with the state of the art, the proposed approach has a significant improvement in the domain of image classification. Experimental evaluation of our approach including a comparison with the previous works on three benchmark datasets demonstrates the effectiveness of the overall method. The proposed algorithm not only improves the previous algorithm but also keeps the advantages of automatic CNN construction without requiring manual interventions. The source code of the our framework can be found at https://github.com/ btogzhan2000/ea-cnn-complab.
Files
bib-6a0ebd28-3f5b-47a2-8b18-ea18750e6d0e.txt
Files
(149 Bytes)
| Name | Size | Download all |
|---|---|---|
|
md5:6a13930d3c0dc2cf7b27a3ef5de4900e
|
149 Bytes | Preview Download |