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.

Explainable artificial intelligence and domain adaptation for predicting HIV infection with graph neural networks

Identifying influential neighbors in social networks and venue affiliations among young MSM: a data science approach to predict HIV infection

Disease prediction with edge-variational graph convolutional networks

HIV-1/HBV Coinfection Accurate Multitarget Prediction Using a Graph Neural Network-Based Ensemble Predicting Model

Application of a long short-term memory neural network: a burgeoning method of deep learning in forecasting HIV incidence in Guangxi, China

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

Predicting unseen antibodies' neutralizability via adaptive graph neural networks

Application of machine-learning techniques in classification of HIV medical care status for people living with HIV in South Carolina

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

A graph neural network-based model with Out-of-Distribution Robustness for enhancing Antiretroviral Therapy Outcome Prediction for HIV-1

Generalizable and explainable prediction of potential miRNA-disease associations based on heterogeneous graph learning

Graph Message Passing with Cross-location Attentions for Long-term ILI Prediction

Identifying HIV-related digital social influencers using an iterative deep learning approach

Prediction of Interpersonal Personality for People Living with HIV Using Machine Learning

Integrating gated recurrent unit in graph neural network to improve infectious disease prediction: an attempt

Study on Prediction Model of HIV Incidence Based on GRU Neural Network Optimized by MHPSO.

Classifying Genetic Interactions Using an HIV Experimental Study

Explainable AI for Tuberculosis Detection using Deep Learning

Predictive model and risk analysis for coronary heart disease in people living with HIV using machine learning

Graph Neural Networks for UnsupervisedDomain Adaptation of Histopathological ImageAnalytics

Lightweight federated learning for STIs/HIV prediction