Optimizing convolutional neural networks architecture using a modified particle swarm optimization for image classification - Archive ouverte HAL Accéder directement au contenu
Article Dans Une Revue Expert Systems with Applications Année : 2023

Optimizing convolutional neural networks architecture using a modified particle swarm optimization for image classification

Résumé

Although Convolutional Neural Networks (CNNs) have been shown to be highly effective in image classification tasks, designing their architecture to achieve optimal results is often challenging. This process is time consuming, requires significant effort and expertise, and is complicated by the large number of hyperparameters. To address this problem, in this work we propose an approach that reduces human intervention and automatically generates the best CNN design. Our approach uses a variant of Particle Swarm Optimization (PSO), called Particle Swarm Optimization without Velocity (PSWV), to speed up convergence and reduce the number of iterations required to determine the optimal CNN hyperparameters. We developed a novel strategy to determine the updated position of each particle using a linear combination of the best position of the particle and the best position of the swarm without relying on the velocity equation. Our algorithm harnesses the power of the variable-length encoding strategy to represent particles within the population, thereby providing swift convergence towards the best architecture. We evaluate our proposed algorithm against several recent algorithms in the literature by using nine benchmark datasets for classification tasks and comparing it to 27 other algorithms, including state-of-the-art ones. Our experimental results show that our proposed method, pswvCNN, is able to quickly find effective CNN architectures that provide comparable performance to the best currently available designs, indicating its significant potential.
Fichier non déposé

Dates et versions

hal-04115021 , version 1 (02-06-2023)

Identifiants

Citer

D. Elhani, A.C. Megherbi, A. Zitouni, F. Dornaika, S. Sbaa, et al.. Optimizing convolutional neural networks architecture using a modified particle swarm optimization for image classification. Expert Systems with Applications, 2023, 229, part A, 120411. ⟨10.1016/j.eswa.2023.120411⟩. ⟨hal-04115021⟩
31 Consultations
0 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More