Background: Early detection and personalized care for Alzheimer's Disease (AD) mitigate the devastating consequences for millions of people around the globe. In the current scenario, there is a lack of user-friendly AI applications for predicting and understanding the progression of AD. The application should address the critical need for a predictive analytics tool that offers timely and transparent insights by utilizing the patient data. The research aims to develop an AI-driven application for predicting and interpreting the disease progression.
Method: The application utilizes ADNI dataset, where the available patient data has been oversampled to 13752 to balance the target class. The Random Forest Classifier, a powerful machine learning technique was used to train the set of selected features including age, cognitive scores from standardized tests, and brain imaging data. Based on these inputs, the model accurately predicts an individual's current cognitive state, categorizing them as Normal, Mild Cognitive Impairment (MCI), or Dementia. The application also utilizes the patient's historical and current data to predict the progression from MCI to Dementia, or Normal to MCI, or remaining in the current cognitive state. The application uses LIME (Local Interpretable Model-agnostic Explanations) and SHAP (SHapley Additive exPlanations) for transparent explanations and providing clear insights into the rationale behind interpretability in predicting each class using different visualizations.
Result: The application (https://alzheimer-disease-prediction-progression-explanations.streamlit.app/) demonstrates accuracy in both initial diagnosis and progression prediction (Figure 1). Explainable AI techniques provide user-friendly insights into model reasoning, promoting trust and understanding (Figure 2 and 3). The user-friendly interface facilitates data input and interpretation for both healthcare professionals and individuals.
Conclusion: The AI-driven application emerges as a valuable tool in AD prediction and analysis. The transparent and interpretable nature of the model, facilitated by LIME and SHAP, enhances trust and understanding. Early detection and progression prediction capabilities allow healthcare professionals with timely interventions to care the patients.
© 2024 The Alzheimer's Association. Alzheimer's & Dementia published by Wiley Periodicals LLC on behalf of Alzheimer's Association.