Journal Article

Machine learning models for risk prediction of age-related macular degeneration in Fujian eye study

Luo, Xiangdong, Zhang, Mingqin, Li, Yang, Li, Xiaoxin, Hu, Qinrui, Wang, Bin

2025

Overview

Age-related macular degeneration (AMD) is a retinal disorder that significantly impairs vision. This study investigates various machine learning models for predicting AMD risk, laying the groundwork for further research using big data and determining the most effective predictive model. Utilizing data from 8211 records with 39 features from the Fujian Eye Study, a cross-sectional epidemiological investigation, several machine learning models were developed and assessed. The models included logistic regression (LR), K-nearest neighbors (KNN), support vector machine (SVM), decision tree (DT), random forest (RF), light gradient boosting machine (LightGBM), and extreme gradient boosting (XGBoost). Data preprocessing, feature selection, and model training were all key components of the study. After evaluating multiple models, the logistic regression model emerged as the most accurate, achieving a balanced accuracy of 0.6364. Among the predictive features, educational background had the highest influence on the model's predictions, with an average SHAP (SHapley Additive exPlanations) value of 0.8199. Other significant factors included outdoor time and left eye spherical equivalent (OSSE), with SHAP values of 0.6474 and 0.6377, respectively. This study confirms that logistic regression is the most effective machine learning model for predicting AMD risk, with educational background identified as the most critical risk factor.

• Publisher: Public Library of Science,Public Library of Science (PLoS)
• Subject: Accuracy / Age / Age related diseases / Aged / Aged, 80 and over / Big Data / Blood pressure / Body mass index / Cross-Sectional Studies / Datasets / Decision trees / Diagnosis / Education / Epidemiology / Ethics / Eye / Eye diseases / Female / Health aspects / Humans / Impact factors / Influence / Learning algorithms / Logistic Models / Machine Learning / Macular degeneration / Macular Degeneration - diagnosis / Macular Degeneration - epidemiology / Male / Middle Aged / Prediction models / Regression analysis / Regression models / Risk Factors / Support Vector Machine / Support vector machines / Type 2 diabetes / Value analysis / Variables / Visual acuity