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.
