Abstract:Background: Alzheimer disease and related dementias (ADRD) rank as the sixth leading cause of death in the United States, underlining the importance of accurate ADRD risk prediction. While recent advancements in ADRD risk prediction have primarily relied on imaging analysis, not all patients undergo medical imaging before an ADRD diagnosis. Merging machine learning with claims data can reveal additional risk factors and uncover interconnections among diverse medical codes. Objective: The study aims to use graph neural networks (GNNs) with claim data for ADRD risk prediction. Addressing the lack of human-interpretable reasons behind these predictions, we introduce an innovative, self-explainable method to evaluate relationship importance and its influence on ADRD risk prediction. Methods: We used a variationally regularized encoder-decoder GNN (variational GNN [VGNN]) integrated with our proposed relation importance method for estimating ADRD likelihood. This self-explainable method can provide a feature-important explanation in the context of ADRD risk prediction, leveraging relational information within a graph. Three scenarios with 1-year, 2-year, and 3-year prediction windows were created to assess the model's efficiency, respectively. Random forest (RF) and light gradient boost machine (LGBM) were used as baselines. By using this method, we further clarify the key relationships for ADRD risk prediction. Results: In scenario 1, the VGNN model showed area under the receiver operating characteristic (AUROC) scores of 0.7272 and 0.7480 for the small subset and the matched cohort data set. It outperforms RF and LGBM by 10.6% and 9.1%, respectively, on average. In scenario 2, it achieved AUROC scores of 0.7125 and 0.7281, surpassing the other models by 10.5% and 8.9%, respectively. Similarly, in scenario 3, AUROC scores of 0.7001 and 0.7187 were obtained, exceeding 10.1% and 8.5% than the baseline models, respectively. These results clearly demonstrate the significant superiority of the graph-based approach over the tree-based models (RF and LGBM) in predicting ADRD. Furthermore, the integration of the VGNN model and our relation importance interpretation could provide valuable insight into paired factors that may contribute to or delay ADRD progression. Conclusions: Using our innovative self-explainable method with claims data enhances ADRD risk prediction and provides insights into the impact of interconnected medical code relationships. This methodology not only enables ADRD risk modeling but also shows potential for other image analysis predictions using claims data.

An application of the Bayesian network model based on the EN-ESL-GA algorithm: Exploring the predictors of heart disease in middle-aged and elderly people in China

A Machine Learning-Based Data Mining in Medical Examination Data: a Biological Features-Based Biological Age Prediction Model

Improving Cardiovascular Risk Prediction Through Machine Learning Modelling of Irregularly Repeated Electronic Health Records

Predicting all-cause mortality and premature death using interpretable machine learning among a middle-aged and elderly Chinese population

Risk prediction of hypertension complications based on the intelligent algorithm optimized Bayesian network

Bayesian network analysis of factors influencing type 2 diabetes, coronary heart disease, and their comorbidities

Study on the risk of coronary heart disease in middle-aged and young people based on machine learning methods: a retrospective cohort study

An explainable machine learning based prediction model for Alzheimer's disease in China longitudinal aging study

Individualized prediction of chronic kidney disease for the elderly in longevity areas in China: Machine learning approaches

Self-Explainable Graph Neural Network for Alzheimer Disease and Related Dementias Risk Prediction: Algorithm Development and Validation Study

An Ensemble Heart Disease Prediction Model Bagged with Logistic Regression, Naïve Bayes and K Nearest Neighbour.

Disability risk prediction model based on machine learning among Chinese healthy older adults: results from the China Health and Retirement Longitudinal Study

An improved ensemble learning approach for the prediction of heart disease risk

Effective Data Decision-Making and Transmission System Based on Mobile Health for Chronic Disease Management in the Elderly

What factors preventing the older adults in China from living longer: a machine learning study

A case for the use of deep learning algorithms for individual and population level assessments of mental health disorders: Predicting depression among China's elderly

Using machine learning algorithms for predicting cognitive impairment and identifying modifiable factors among Chinese elderly people

Integrating multi-task and cost-sensitive learning for predicting mortality risk of chronic diseases in the elderly using real-world data

Prediction model for the risk of osteoporosis incorporating factors of disease history and living habits in physical examination of population in Chongqing, Southwest China: based on artificial neural network

Self-explainable Graph Neural Network for Alzheimer's Disease And Related Dementias Risk Prediction

Establishment of a Mild Cognitive Impairment Risk Model in Middle-Aged and Older Adults: a Longitudinal Study