Causal Explanation from Mild Cognitive Impairment Progression using Graph Neural Networks.

Abstract

Mild Cognitive Impairment (MCI) is a transitional stage between normal cognitive aging and dementia. Some individuals with MCI revert to normal, while others progress to dementia. There are limited studies using explainable artificial intelligence on longitudinal data, particularly including genotypes, biomarkers and chronic diseases, to explore these differences. This study introduces a novel approach to understanding MCI progression using explainable graph neural networks. Utilizing longitudinal temporal data, we constructed a comprehensive graph representation of each individual in the study cohort. Our temporal graph convolutional network achieved 72.4% accuracy in predicting MCI transitions, while our causal explanation method outperformed existing explanation techniques in stability, accuracy, and faithfulness. We identified a causal subgraph with informative variables including hypertension, arrhythmia, congestive heart failure, coronary artery disease, stroke, lipid-related issues, and sex.