Automatic design of CNN architecture based on genetic algorithm and particle swarm optimization
摘要
Convolutional neural networks (CNNs) are indispensable in computer vision, especially for image classification, but designing optimal CNN architectures is challenging and often requires extensive domain-specific knowledge. This paper presents a novel method that combines genetic algorithms (GAs) and particle swarm optimization (PSO) to automate CNN architecture design. The proposed algorithm integrates the global search capability of GAs with the local optimization efficiency of PSO, effectively addressing the limitations of each. Tested on the CIFAR-10 benchmark dataset, the algorithm achieved a classification accuracy of 94.47%, outperforming most existing manually designed and evolutionary algorithm-designed architectures. Statistical analysis confirmed the significance of the improvements. This approach not only enhances the accuracy of CNN architectures but also demonstrates the potential of hybrid metaheuristic algorithms in automated design, making it more accessible for researchers across various fields.