Evolving Deep Architecture Generation with Residual Connections for Image Classification Using Particle Swarm Optimization

Tom Lawrence, Li Zhang*, Kay Rogage, Chee Peng Lim

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

1 Downloads (Pure)

Abstract

Automated deep neural architecture generation has gained increasing attention. However, exiting studies either optimize important design choices, without taking advantage of modern strategies such as residual/dense connections, or they optimize residual/dense networks but reduce search space by eliminating fine-grained network setting choices. To address the aforementioned weaknesses, we propose a novel particle swarm optimization (PSO)-based deep architecture generation algorithm, to devise deep networks with residual connections, whilst performing a thorough search which optimizes important design choices. A PSO variant is proposed which incorporates a new encoding scheme and a new search mechanism guided by non-uniformly randomly selected neighboring and global promising solutions for the search of optimal architectures. Specifically, the proposed encoding scheme is able to describe convolutional neural network architecture configurations with residual connections. Evaluated using benchmark datasets, the proposed model outperforms existing state-of-the-art methods for architecture generation. Owing to the guidance of diverse non-uniformly selected neighboring promising solutions in combination with the swarm leader at fine-grained and global levels, the proposed model produces a rich assortment of residual architectures with great diversity. Our devised networks show better capabilities in tackling vanishing gradients with up to 4.34% improvement of mean accuracy in comparison with those of existing studies.
Original languageEnglish
Article number7936
Pages (from-to)1-23
Number of pages23
JournalSensors
Volume21
Issue number23
DOIs
Publication statusPublished - 28 Nov 2021

Fingerprint

Dive into the research topics of 'Evolving Deep Architecture Generation with Residual Connections for Image Classification Using Particle Swarm Optimization'. Together they form a unique fingerprint.

Cite this