Performance Optimization of Convolutional Neural Networks in Image Classification
Haoyu Ma · International Journal of Computer Science and Information Technology · 2024
With the development of deep learning, CNNs have become mainstream in image classification. However, CNN performance is affected by factors like loss function choice and optimization algorithm. This paper aims to improve CNN performance in image classification by optimizing SGD. Firstly, we analyze different loss functions and propose a suitable optimization strategy. Second, we explore SGD and its variants' roles in CNN training, focusing on accelerating convergence and improving accuracy. Experiments show the proposed scheme effectively enhances CNN performance, providing a reliable solution for image classification. This study supports CNN applications in image recognition and lays a foundation for exploring more efficient optimization methods.