Evolving Image Classification Architectures with Enhanced Particle Swarm Optimisation

Benjamin Fielding, Li Zhang

Research output: Contribution to journalArticlepeer-review

69 Citations (Scopus)
51 Downloads (Pure)

Abstract

Convolutional Neural Networks (CNNs) have become the de facto technique for image feature extraction in recent years, however their design and construction remains a complicated task. As more developments are made in progressing the internal components of CNNs, the task of assembling them effectively from core components becomes even more arduous. To overcome these barriers, we propose Swarm Optimised Block Architecture (SOBA), combined with an enhanced adaptive Particle Swarm Optimisation (PSO) algorithm for deep CNN model evolution. The enhanced PSO model employs adaptive acceleration coefficients generated using several cosine annealing mechanisms to overcome stagnation. Specifically, we propose a combined training and structure optimisation process for deep CNN model generation, where the proposed PSO model is utilised to explore a bespoke search space defined by a simplified block-based structure. The proposed PSO model not only devises deep networks specifically for image classification, but also builds and pre-trains models for transfer learning tasks. To significantly reduce the hardware and computational cost of the search, the devised CNN model is optimised and trained simultaneously, using a weight sharing mechanism and a final fine-tuning process. Our system compares favourably with related research for optimised deep network generation. It achieves an error rate of 4.78% on the CIFAR-10 image classification task, with 34 hours of combined optimisation and training, and an error rate of 25.42% on the CIFAR-100 image data set in 36 hours. All experiments were performed on a single NVIDIA GTX 1080Ti consumer GPU.
Original languageEnglish
Pages (from-to)68560-68575
Number of pages16
JournalIEEE Access
Volume6
Early online date13 Nov 2018
DOIs
Publication statusE-pub ahead of print - 13 Nov 2018

Keywords

  • image classification
  • convolutional neural networks
  • computer vision

Fingerprint

Dive into the research topics of 'Evolving Image Classification Architectures with Enhanced Particle Swarm Optimisation'. Together they form a unique fingerprint.

Cite this