Automated CNN architecture design with enhanced particle swarm optimization
摘要
Deep convolutional neural networks (CNNs) are powerful deep learning models for tackling various computer vision challenges, particularly image classification. However, designing an optimal CNN model is a complex, labor-intensive, and resource-demanding process that typically requires extensive manual effort and domain-specific knowledge. This paper introduces EPSO-CNN, an efficient framework leveraging particle swarm optimization (PSO) to automatically design optimal CNN architectures for image classification tasks. The proposed method utilizes a modified variable-length encoding scheme for flexible and efficient representation of CNN layers as particles, facilitating rapid architecture construction without conversion overhead. Novel strategies for velocity and position updates are incorporated to enhance search exploration and prevent premature convergence. Batch normalization and dropout techniques are employed to accelerate training and mitigate overfitting. Experimental results on eight widely used image classification datasets show that EPSO-CNN outperforms most of the 26 state-of-the-art models with notably lower error rates, and its superiority is further validated on NAS-Bench-101 and NAS-Bench-201, confirming its competitiveness across standardized NAS evaluation platforms. The best CNN architectures generated by EPSO-CNN are evaluated on facial expression recognition (FER) challenge dataset, showcasing their robustness and generalization ability in real-life application. The findings confirm the efficacy of EPSO-CNN in automating the design of deep CNN architectures, offering a valuable tool for advancing image classification methodologies.