An improved transfer learning algorithm using integration method and its application in image segmentation and recognition

Chen, Yuchong
DOI: https://doi.org/10.1007/s11042-024-20290-z
IF: 2.577
2024-09-26
Multimedia Tools and Applications
Abstract:Images find frequent applications in numerous fields, such as healthcare and aerospace, and the demand for image recognition and classification is continually rising. Currently, research on image classification is primarily conducted within the realm of big data, where ample training images are available. However, in scenarios where image data acquisition is challenging, most studies rely on a single transfer learning model for experimentation. Building upon the examination of multiple transfer learning models, this paper delves into fusing these models and further exploring their optimization strategies. The findings reveal that in datasets with limited data, involving too many unfrozen convolution layers in weight updates can lead to overfitting in the transfer learning model. Conversely, fine-tuning only the fully connected layers achieves better training outcomes. The research outcomes indicate that the Stacking fusion model plays a pivotal role in enhancing model accuracy. Compared to VGG16, ResNet, and Inception V3, the fusion model exhibits accuracy improvements of 11.11%, 35.40%, and 20.64%, respectively. Furthermore, by substituting the InceptionV3 model with the DenseNet model and replacing the secondary classifier with the XGBoost algorithm, the overall model accuracy was further boosted by 3.65%, while the time consumption was reduced by 11.20%. The novelty of this paper lies in its utilization of the Stacking model fusion approach to integrate and optimize transfer learning models, thereby exploring avenues to enhance the ultimate performance of the model.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?