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.

DRR: Global Context-Aware Neural Network Using Disease Relationship Reasoning and Attention-Based Feature Fusion

A novel interactive deep cascade spectral graph convolutional network with multi-relational graphs for disease prediction

Regional Brain Fusion: Graph Convolutional Network for Alzheimer's Disease Prediction and Analysis

ReHoGCNES-MDA: prediction of miRNA-disease associations using homogenous graph convolutional networks based on regular graph with random edge sampler

Deep Knowledge Reasoning Guided Disease Prediction.

Learning by Reasoning: an Explainable Hierarchical Association Regularized Deep Learning Method for Disease Prediction.

Context-Aware Health Event Prediction via Transition Functions on Dynamic Disease Graphs

Context-aware and Time-aware Attention-based Model for Disease Risk Prediction with Interpretability

Dementia analysis from functional connectivity network with graph neural networks

An effective drug-disease associations prediction model based on graphic representation learning over multi-biomolecular network

BertNDA: A Model Based on Graph-Bert and Multi-scale Information Fusion for Ncrna-Disease Association Prediction

DHGL: Dynamic hypergraph‐based deep learning model for disease prediction

Boosting drug-disease association prediction for drug repositioning via dual-feature extraction and cross-dual-domain decoding

Predicting potential microbe-disease associations based on dual branch graph convolutional network

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

Gndd: A Graph Neural Network-Based Method For Drug-Disease Association Prediction

RDGAN: Prediction of circRNA-Disease Associations via Resistance Distance and Graph Attention Network

Knowledge guided diagnosis prediction via graph spatial-temporal network

LGCDA: Predicting CircRNA-Disease Association Based on Fusion of Local and Global Features

A Knowledge-Guided Method for Disease Prediction Based on Attention Mechanism

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