Ensemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edema

Abstract Background Diabetic macular edema (DME) is a leading cause of vision loss in diabetes, with variable responses to anti-vascular endothelial growth factor (anti-VEGF) therapy in DME patients. Current diagnosis relies on optical coherence tomography (OCT) imaging, but manual interpretation is...

Full description

Saved in:
Bibliographic Details
Main Authors: Yu Fang, Jianwei Lin, Peiwen Xie, Huishan Zhu, Tsz Kin Ng, Guihua Zhang
Format: Article
Language:English
Published: BMC 2025-07-01
Series:BMC Ophthalmology
Subjects:
Online Access:https://doi.org/10.1186/s12886-025-04181-x
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1849769101382123520
author Yu Fang
Jianwei Lin
Peiwen Xie
Huishan Zhu
Tsz Kin Ng
Guihua Zhang
author_facet Yu Fang
Jianwei Lin
Peiwen Xie
Huishan Zhu
Tsz Kin Ng
Guihua Zhang
author_sort Yu Fang
collection DOAJ
description Abstract Background Diabetic macular edema (DME) is a leading cause of vision loss in diabetes, with variable responses to anti-vascular endothelial growth factor (anti-VEGF) therapy in DME patients. Current diagnosis relies on optical coherence tomography (OCT) imaging, but manual interpretation is limited. This study aims to integrate 3D-OCT features and clinical variables to develop machine learning (ML) models for predicting anti-VEGF treatment outcomes. Methods and analysis Medical records and 3D-OCT images of DME patients were included in this study. The 3D-OCT images were categorized into good and poor visual response groups based on the best corrected visual acuity at one month after three consecutive anti-VEGF treatments. The images and clinical features were subjected to assessment by 11 automatic classification models for anti-VEGF treatment responses in DME patients. The top 3 performing models were selected to build an ensemble model, and evaluated in the test dataset. Results This study included 142 patients with 3D-OCT images of 170 eyes. A total of 20 image and clinical features were selected for the model construction and test in DME patients responded to anti-VEGF therapy. Adaptive boosting (AdaBoost), GradientBoosting, and light gradient boosting machine (LightGBM) exhibited better performances than the remaining 8 models. The ensemble model constructed achieved a sensitivity of 0.941, specificity of 0.882, and accuracy of 0.912 in the test dataset, with an area under the receiver operating characteristic curve of 0.976. Conclusion This study established an ensemble ML algorithm based on 3D-OCT images and clinical features for automatic detection of treatment responses to anti-VEGF treatment in DME patients to predict the efficacy of anti-VEGF treatment in DME patients and assist clinicians in optimal treatment decisions.
format Article
id doaj-art-ffbbcf0cfbac4d9496c3ce64ebccc7bd
institution DOAJ
issn 1471-2415
language English
publishDate 2025-07-01
publisher BMC
record_format Article
series BMC Ophthalmology
spelling doaj-art-ffbbcf0cfbac4d9496c3ce64ebccc7bd2025-08-20T03:03:34ZengBMCBMC Ophthalmology1471-24152025-07-012511910.1186/s12886-025-04181-xEnsemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edemaYu Fang0Jianwei Lin1Peiwen Xie2Huishan Zhu3Tsz Kin Ng4Guihua Zhang5Joint Shantou International Eye Center of Shantou University and the Chinese University of Hong KongJoint Shantou International Eye Center of Shantou University and the Chinese University of Hong KongJoint Shantou International Eye Center of Shantou University and the Chinese University of Hong KongJoint Shantou International Eye Center of Shantou University and the Chinese University of Hong KongJoint Shantou International Eye Center of Shantou University and the Chinese University of Hong KongJoint Shantou International Eye Center of Shantou University and the Chinese University of Hong KongAbstract Background Diabetic macular edema (DME) is a leading cause of vision loss in diabetes, with variable responses to anti-vascular endothelial growth factor (anti-VEGF) therapy in DME patients. Current diagnosis relies on optical coherence tomography (OCT) imaging, but manual interpretation is limited. This study aims to integrate 3D-OCT features and clinical variables to develop machine learning (ML) models for predicting anti-VEGF treatment outcomes. Methods and analysis Medical records and 3D-OCT images of DME patients were included in this study. The 3D-OCT images were categorized into good and poor visual response groups based on the best corrected visual acuity at one month after three consecutive anti-VEGF treatments. The images and clinical features were subjected to assessment by 11 automatic classification models for anti-VEGF treatment responses in DME patients. The top 3 performing models were selected to build an ensemble model, and evaluated in the test dataset. Results This study included 142 patients with 3D-OCT images of 170 eyes. A total of 20 image and clinical features were selected for the model construction and test in DME patients responded to anti-VEGF therapy. Adaptive boosting (AdaBoost), GradientBoosting, and light gradient boosting machine (LightGBM) exhibited better performances than the remaining 8 models. The ensemble model constructed achieved a sensitivity of 0.941, specificity of 0.882, and accuracy of 0.912 in the test dataset, with an area under the receiver operating characteristic curve of 0.976. Conclusion This study established an ensemble ML algorithm based on 3D-OCT images and clinical features for automatic detection of treatment responses to anti-VEGF treatment in DME patients to predict the efficacy of anti-VEGF treatment in DME patients and assist clinicians in optimal treatment decisions.https://doi.org/10.1186/s12886-025-04181-xDiabetic macular edemaAnti-VEGF treatmentMachine learningOptical coherence tomography.
spellingShingle Yu Fang
Jianwei Lin
Peiwen Xie
Huishan Zhu
Tsz Kin Ng
Guihua Zhang
Ensemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edema
BMC Ophthalmology
Diabetic macular edema
Anti-VEGF treatment
Machine learning
Optical coherence tomography.
title Ensemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edema
title_full Ensemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edema
title_fullStr Ensemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edema
title_full_unstemmed Ensemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edema
title_short Ensemble machine learning algorithm for anti-VEGF treatment efficacy prediction in diabetic macular edema
title_sort ensemble machine learning algorithm for anti vegf treatment efficacy prediction in diabetic macular edema
topic Diabetic macular edema
Anti-VEGF treatment
Machine learning
Optical coherence tomography.
url https://doi.org/10.1186/s12886-025-04181-x
work_keys_str_mv AT yufang ensemblemachinelearningalgorithmforantivegftreatmentefficacypredictionindiabeticmacularedema
AT jianweilin ensemblemachinelearningalgorithmforantivegftreatmentefficacypredictionindiabeticmacularedema
AT peiwenxie ensemblemachinelearningalgorithmforantivegftreatmentefficacypredictionindiabeticmacularedema
AT huishanzhu ensemblemachinelearningalgorithmforantivegftreatmentefficacypredictionindiabeticmacularedema
AT tszkinng ensemblemachinelearningalgorithmforantivegftreatmentefficacypredictionindiabeticmacularedema
AT guihuazhang ensemblemachinelearningalgorithmforantivegftreatmentefficacypredictionindiabeticmacularedema