Improved Method of ResNet50 Image Classification Based on Transfer Learning
Pubblicato online: 16 giu 2025
Pagine: 1 - 9
DOI: https://doi.org/10.2478/ijanmc-2025-0011
Parole chiave
© 2025 Tao Shi et al., published by Sciendo
This work is licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
Aiming at the issues of high computational cost and limited generalization ability of ResNet50 in classifying images, this study advances an optimization strategy based on transfer learning. The model is initialized with transfer learning to reduce computational burden, and data augmentation techniques are employed to enhance generalization ability. Additionally, label smoothing is introduced to optimize the cross-entropy loss, thereby reducing sensitivity to noisy labels. The training process is further optimized using cosine annealing learning rate decay. Experimental findings reveal that the optimized ResNet50 model achieves a 6.25% improvement in classification accuracy on the CIFAR-10 dataset, validating the validity of the suggested methods.