Citation
Ahmed, Md. Kowsar and Rahman, Anichur and Kundu, Dipanjali and Alam, Md. Alve Authi and Ovi, Maharabe Uddin Sarker and Mahir, Shahariar Hossain and Tashrif, Md Tanjum An and Miah, Abu Saleh Musa and Al Farid, Fahmid and Mansor, Sarina (2026) An explainable framework for dementia risk prediction based on lifestyle data using machine learning and deep learning. Discover Artificial Intelligence, 6 (1). ISSN 2731-0809|
Text
s44163-025-00782-z.pdf - Published Version Restricted to Repository staff only Download (4MB) |
Abstract
Dementia is a progressive neurodegenerative disease that affects millions worldwide and is a challenge for early detection due to the cost, invasiveness, and limited access of available clinical methods. This study proposes an explainable Artificial Intelligence (AI) system for the prediction of risk for dementia based solely on lifestyle parameters, thus providing a non-invasive and economical option. We tested a spectrum of machine learning methods (Decision Tree, Random Forest, K-Nearest Neighbor, Naïve Bayes, Support Vector Machine, and Multilayer Perceptron) and also deep learning models (Hybrid Neural Network). Out of those, Random Forest exhibited the best accuracy of 93.1% on lifestyle data and 95.2% on an independent external validation dataset, exhibiting excellent generalizability. To combat the crucial issue of model explainability, SHapley Additive exPlanations (SHAP) was utilized for providing both the global and local interpretability and highlighting the importance of modifiable lifestyle parameters, e.g., physical activity, smoking, and comorbidities, in the determination of dementia risk. Class imbalance was overcome using methods like SMOTE, Gaussian augmentation, and scaling, thus enabling reliable detection of minority classes. Results confirm that the combination of explainable AI and efficient ML/DL methods provides accurate, interpretable, and clinically significant prediction of dementia risk. The constructed system has the potential for supporting active healthcare interventions, especially in resource-constrained environments where advanced diagnosis equipment is not available.
| Item Type: | Article |
|---|---|
| Uncontrolled Keywords: | Dementia prediction, Artificial intelligence, Explainable AI, Data analysis, Lifestyle factors, Shapley additive explanations, Neural networks, Hybrid neural network, Machine learning, Data imbalance, Proactive healthcare |
| Subjects: | Q Science > QA Mathematics > QA71-90 Instruments and machines |
| Divisions: | Faculty of Artificial Intelligence & Engineering (FAIE) |
| Depositing User: | Ms Suzilawati Abu Samah |
| Date Deposited: | 31 Jul 2026 07:14 |
| Last Modified: | 31 Jul 2026 07:14 |
| URII: | http://shdl.mmu.edu.my/id/eprint/16430 |
Downloads
Downloads per month over past year
Edit (login required) |
