Enhancing Alzheimer's disease detection: An explainable machine learning approach with ensemble techniques

Citation

Mahamud, Eram and Assaduzzaman, Md and Islam, Jahirul and Fahad, Nafiz and Hossen, Md Jakir and Ramanathan, Thirumalaimuthu Thirumalaiappan (2025) Enhancing Alzheimer's disease detection: An explainable machine learning approach with ensemble techniques. Intelligence-Based Medicine, 11. p. 100240. ISSN 26665212

[img] Text
4.pdf - Published Version
Restricted to Repository staff only

Download (9MB)

Abstract

Alzheimer's Disease (AD) is a progressive neurodegenerative disorder that necessitates early and accurate diagnosis for effective intervention. This study presents a novel machine learning (ML)-driven predictive framework for AD diagnosis, integrating Explainable Artificial Intelligence (XAI) methodologies to enhance interpretability. The dataset, sourced from Kaggle, comprises 2149 patient records with 34 distinct attributes, representing a comprehensive range of demographic, clinical, and lifestyle-related factors. To improve model robustness, rigorous data preprocessing techniques were employed, including mean/mode imputation for missing values, feature scaling using min-max normalization, and class balancing via SMOTE, SMOTEENN, and ADASYN. Feature selection technique was performed using Chi-Square and Recursive Feature Elimination (RFE) to retain the most relevant predictors. Various ML models—including Naïve Bayes, Decision Tree, Random Forest, Logistic Regression, AdaBoost, XGBoost, K-Nearest Neighbors (KNN), and Gradient Boosting—were assessed using accuracy, precision, recall, F1-score, and AUC (Area Under the Curve). The proposed ensemble model, combining LightGBM (LGBM) and Random Forest (RF) with Chi-Square feature selection and utilizing soft voting, achieved the highest test accuracy of 96.35 %, surpassing existing models. Additionally, SHAP (SHapley Additive exPlanations) and LIME (Local Interpretable Model-agnostic Explanations) were utilized to interpret the model's decision-making process, identifying key risk factors and improving transparency for clinical applications. These findings highlight the potential of ML and XAI in advancing AD diagnosis, with future work aiming to validate the model on larger, more diverse datasets and integrate it into real-world clinical workflows.

Item Type: Article
Subjects: Q Science > QA Mathematics > QA71-90 Instruments and machines
Divisions: Faculty of Engineering and Technology (FET)
Depositing User: Ms Suzilawati Abu Samah
Date Deposited: 30 Apr 2025 01:14
Last Modified: 30 Apr 2025 01:14
URII: http://shdl.mmu.edu.my/id/eprint/13708

Downloads

Downloads per month over past year

View ItemEdit (login required)