Deep learning for predicting myopia severity classification method
摘要
Myopia is a major cause of vision impairment. To improve the efficiency of myopia screening, this paper proposes a deep learning model, X-ENet, which combines the advantages of depthwise separable convolution and dynamic convolution to classify different severities of myopia. The proposed model not only enables precise extraction of detailed features from fundus images but also achieves lightweight processing, thereby improving both computational efficiency and classification accuracy.
ApproachFirst, fundus images are enhanced and preprocessed to improve feature extraction effectiveness and enhance the model’s generalization capability. Then, the model is trained using fivefold cross-validation, leveraging dynamic convolution and depthwise separable convolution to extract features from each fundus image and classify the severity of myopia. Next, Grad-CAM is employed to visualize the model’s decision-making process, highlighting the regions contributing to classification. Finally, a user-friendly GUI interface is developed to intuitively present the classification results, thereby enhancing the system’s usability and practical applicability.
ResultsThe experimental results show that the proposed method achieves an accuracy of 0.9104, a precision of 0.8154, a recall of 0.8177, an F1-score of 0.8147, and a specificity of 0.9376 in the classification of myopia severity.
SignificanceThe model significantly outperforms existing conventional deep learning models in terms of accuracy, demonstrating strong effectiveness and reliability.