Lee Ki-Sun, Kim Jae Young, Jeon Eun-Tae, Choi Won Suk, Kim Nan Hee, Lee Ki Yeol
Medical Science Research Center, Ansan Hospital, Korea University College of Medicine, Ansan si 15355, Korea.
Division of Infectious Diseases, Department of Internal Medicine, Ansan Hospital, Korea University College of Medicine, Ansan si 15355, Korea.
J Pers Med. 2020 Nov 7;10(4):213. doi: 10.3390/jpm10040213.
According to recent studies, patients with COVID-19 have different feature characteristics on chest X-ray (CXR) than those with other lung diseases. This study aimed at evaluating the layer depths and degree of fine-tuning on transfer learning with a deep convolutional neural network (CNN)-based COVID-19 screening in CXR to identify efficient transfer learning strategies. The CXR images used in this study were collected from publicly available repositories, and the collected images were classified into three classes: COVID-19, pneumonia, and normal. To evaluate the effect of layer depths of the same CNN architecture, CNNs called VGG-16 and VGG-19 were used as backbone networks. Then, each backbone network was trained with different degrees of fine-tuning and comparatively evaluated. The experimental results showed the highest AUC value to be 0.950 concerning COVID-19 classification in the experimental group of a fine-tuned with only 2/5 blocks of the VGG16 backbone network. In conclusion, in the classification of medical images with a limited number of data, a deeper layer depth may not guarantee better results. In addition, even if the same pre-trained CNN architecture is used, an appropriate degree of fine-tuning can help to build an efficient deep learning model.
根据最近的研究,与其他肺部疾病患者相比,新冠肺炎患者的胸部X光(CXR)具有不同的特征。本研究旨在评估基于深度卷积神经网络(CNN)的新冠肺炎胸部X光筛查中迁移学习的层深度和微调程度,以确定有效的迁移学习策略。本研究中使用的胸部X光图像是从公开可用的存储库中收集的,收集的图像分为三类:新冠肺炎、肺炎和正常。为了评估同一CNN架构的层深度的影响,将称为VGG-16和VGG-19的CNN用作主干网络。然后,每个主干网络以不同程度的微调进行训练并进行比较评估。实验结果表明,在仅对VGG16主干网络的2/5个块进行微调的实验组中,新冠肺炎分类的最高AUC值为0.950。总之,在数据量有限的医学图像分类中,更深的层深度可能无法保证更好的结果。此外,即使使用相同的预训练CNN架构,适当程度的微调也有助于构建高效的深度学习模型。