Evolving Image Classification Architectures with Enhanced Particle Swarm Optimisation

Fielding, Ben and Zhang, Li (2018) Evolving Image Classification Architectures with Enhanced Particle Swarm Optimisation. IEEE Access, 6. pp. 68560-68575. ISSN 2169-3536

Text (Full text)
Fielding, Zhang - Evolving Image Classification Architectures with Enhanced Particle Swarm Optimisation OA.pdf - Published Version

Download (2MB) | Preview
Official URL: http://dx.doi.org/10.1109/ACCESS.2018.2880416


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.

Item Type: Article
Uncontrolled Keywords: Computer Vision, Convolutional Neural Networks, Deep Learning, Evolutionary Computation, Image Classification, Particle Swarm Optimisation
Subjects: G400 Computer Science
Department: Faculties > Engineering and Environment > Computer and Information Sciences
Depositing User: Paul Burns
Date Deposited: 29 Nov 2018 14:27
Last Modified: 01 Aug 2021 07:49
URI: http://nrl.northumbria.ac.uk/id/eprint/36965

Actions (login required)

View Item View Item


Downloads per month over past year

View more statistics