Research on Skin Lesion Classification Using the Improved EfficientNet Network Based on Transfer Learning
In view of the problems of large network model parameters and low classification accuracy in current skin disease auxiliary classification technology,an improved EfficientNet skin disease classification method based on transfer learning is proposed.This method applies the idea of transfer learning to improve the lightweight deep convolutional neural network EfficientNet,including adding global average pooling layers,freezing different layers and fine-tuning the model to form TL-EfficientNet network.The experimental results show that the accuracy of TL-EfficientNetB0 on the ISIC2018 skin lesion dataset after class weight preprocessing reaches 85.07%,Macro_P reach-es 0.82,and the number of the network parameters is only 4.49 M,which is suitable for mobile deployment.
transfer learninglightweight convolutional neural networkEfficientNetskin lesion classification
万方数据
维普
