基于深度学习的翼状胬肉自动分类系统研究

Research on the automatic classification system of pterygium based on deep learning

摘要:

目的：评估基于深度学习的翼状胬肉自动分类诊断系统的应用价值。

方法：在2020-05/2021-04期间，从南京医科大学附属眼科医院共收集750张翼状胬肉正常、观察期和手术期眼前节图片。在原始数据集和增强数据集上分别训练7个三分类模型。测试临床470张图片，比较数据增强前后模型的泛化能力，确定可用于翼状胬肉自动分类系统的最好模型。

结果：在原始数据集上训练最好模型的灵敏度平均值为92.55%，特异度平均值为96.86%，AUC平均值为94.70%。数据增强后，不同模型灵敏度、特异度和AUC平均提升3.7%、1.9%和2.7%。在增强数据集上训练的EfficientNetB7模型灵敏度平均值为93.63%，特异度平均值为97.34%，AUC平均值为95.47%。

结论：在增强数据集上训练的EfficientNetB7模型取得最好的分类效果，可用于翼状胬肉自动分类系统。该自动分类系统能较好地诊断翼状胬肉疾病，有望成为基层医疗的有效筛查工具，也为翼状胬肉的细化分级研究提供参考。

Abstract:

AIM: To evaluate the application value of the automatic classification and diagnosis system of pterygium based on deep learning.

METHODS: A total of 750 images of normal, observational and operative anterior sections of pterygium were collected from the Affiliated Eye Hospital of Nanjing Medical University between May 2020 and April 2021. Seven triclassification models were respectively trained with original data set and enhanced data set. Totally 470 clinical images were tested, and the generalization ability of the model before and after data enhancement was compared to determine the best model for the automatic classification system of pterygium.

RESULTS:The average sensitivity, specificity and AUC of the best model trained on the original data set were 92.55%, 96.86% and 94.70% respectively. After data was enhanced, the sensitivity, specificity and AUC of different models were increased by 3.7%, 1.9% and 2.7% on average. The sensitivity, specificity and AUC of the EfficientNetB7 model trained on the enhanced data set were 93.63%, 97.34% and 95.47% respectively.

CONDLUSION: The EfficientNetB7 model, which was trained on the enhanced data set, achieves the best classification result and can be used in the automatic classification system of pterygium.This automatic classification system can diagnose diseases about pterygium better, and it is expected to be an effective screening tool for primary medical care. It also provides reference for the research on the refinement and grading of pterygium.