Auto-DenseNet: DenseNet Optimization Method Based on Evolutionary Algorithm

Leilei Zhai, Dianwei Wang, Jie Fang, Zhijie Xu

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

Abstract

To address the issue of time-consuming and laborious design of network structure parameters and connection methods, in this paper we propose an evolutionary algorithm-based DenseNet network optimization method (Auto Dense Convolutional Network, Auto-DenseNet). Firstly, the internal parameters of DenseNet are encoded to generate an initial population, and crossover and variation operators are designed to ensure the effectiveness of the evolution of offspring. Secondly, the dataset is imported into the decoded individual evaluation fitness. And then, the superior individuals are selected to enter the next iteration. Finally, the optimized network individuals are selected for the image classification task. After tuning validation on the MNIST-RD dataset, tests were conducted on three widely used benchmark image classification datasets. The experimental results demonstrate that the network structure searched by the Auto-DenseNet algorithm outperforms most existing structures in terms of classification performance, and the number of network parameters.

Original languageEnglish
Title of host publication5th International Conference on Intelligent Autonomous Systems, ICoIAS 2022
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages387-393
Number of pages7
ISBN (Electronic)9781665498388
DOIs
Publication statusPublished - 2022
Externally publishedYes
Event5th International Conference on Intelligent Autonomous Systems, ICoIAS 2022 - Dalian, China
Duration: 23 Sept 202225 Sept 2022

Publication series

Name5th International Conference on Intelligent Autonomous Systems, ICoIAS 2022

Conference

Conference5th International Conference on Intelligent Autonomous Systems, ICoIAS 2022
Country/TerritoryChina
CityDalian
Period23/09/2225/09/22

Keywords

  • dense connection network
  • evolutionary algorithm
  • image classification
  • Neural network structure optimization

Fingerprint

Dive into the research topics of 'Auto-DenseNet: DenseNet Optimization Method Based on Evolutionary Algorithm'. Together they form a unique fingerprint.

Cite this