Deep Learning Ensemble for Predicting Diabetic Macular Edema Onset Using Ultra-Wide Field Color Fundus Image

Pengyao Qin,Arun J. Thirunavukarasu,Le Zhang
2024-10-09
Abstract:Diabetic macular edema (DME) is a severe complication of diabetes, characterized by thickening of the central portion of the retina due to accumulation of fluid. DME is a significant and common cause of visual impairment in diabetic patients. Center-involved DME (ci-DME) is the highest risk form of disease as fluid extends close to the fovea which is responsible for sharp central vision. Earlier diagnosis or prediction of ci-DME may improve treatment outcomes. Here, we propose an ensemble method to predict ci-DME onset within a year using ultra-wide-field color fundus photography (UWF-CFP) images provided by the DIAMOND Challenge. We adopted a variety of baseline state-of-the-art classification networks including ResNet, DenseNet, EfficientNet, and VGG with the aim of enhancing model robustness. The best performing models were Densenet 121, Resnet 152 and EfficientNet b7, and these were assembled into a definitive predictive model. The final ensemble model demonstrates a strong performance with an Area Under Curve (AUC) of 0.7017, an F1 score of 0.6512, and an Expected Calibration Error (ECE) of 0.2057 when deployed on a synthetic dataset. The performance of this ensemble model is comparable to previous studies despite training and testing in a more realistic setting, indicating the potential of UWF-CFP combined with a deep learning classification system to facilitate earlier diagnosis, better treatment decisions, and improved prognostication in ci-DME.
Image and Video Processing,Artificial Intelligence,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the early prediction of diabetic macular edema (DME), especially center - involved diabetic macular edema (ci - DME). DME is a serious diabetic complication, manifested as thickening of the central part of the retina due to fluid accumulation, and is one of the main causes of visual impairment in diabetic patients. Center - involved DME (ci - DME) is the most dangerous form because the fluid accumulates close to the fovea centralis of the macula, which is responsible for sharp central vision. Early diagnosis or prediction of ci - DME can improve the treatment effect. Therefore, this paper proposes a deep - learning ensemble method based on ultra - wide - field color fundus photography (UWF - CFP) to predict the occurrence of ci - DME within one year. Specifically, the researchers adopted a variety of advanced convolutional neural networks (CNNs), such as ResNet, DenseNet, EfficientNet and VGG. They used data augmentation strategies to improve the robustness of the model and utilized the ensemble method to combine the best - performing models (Densenet 121, Resnet 152 and EfficientNet b7) in order to enhance the overall accuracy and reliability of ci - DME prediction. The performance metrics of the final ensemble model on the synthetic dataset are: AUC 0.7017, F1 score 0.6512, and expected calibration error (ECE) 0.2057, indicating that the model has high performance and potential practical application value.