Comparative Machine Learning Models for Dementia Prediction Using SMOTE

Authors

  • Rahma Puspitasari Universitas Muslim Indonesia
  • Tazkirah Amaliah Universitas Muslim Indonesia
  • Herdianti Darwis Universitas Muslim Indonesia

DOI:

https://doi.org/10.56705/ijaimi.v3i2.351

Keywords:

Dementia Prediction, SMOTE, XGBoost, Random Forest, SVM, CDSS

Abstract

Dementia is a progressive neurodegenerative disorder that leads to cognitive decline and significantly affects patients' quality of life. Early detection is crucial for determining appropriate medical interventions and slowing disease progression. This study aims to develop a machine learning-based dementia prediction model and compare the performance of three algorithms: Support Vector Machine (SVM), Random Forest (RF), and XGBoost. The dataset, obtained from the Kaggle platform, consists of 373 MRI-based patient records categorized into three diagnosis groups: Converted, Demented, and Nondemented. To address class imbalance, the Synthetic Minority Oversampling Technique (SMOTE) was applied. Experimental results show that the XGBoost algorithm achieved the best performance, with an accuracy of 93.86%, precision of 94%, recall of 94%, and F1-score of 94%, outperforming SVM and Random Forest. The application of SMOTE improved the model’s sensitivity to minority classes. The combination of XGBoost and SMOTE demonstrates high accuracy in dementia prediction and holds potential for integration into clinical decision support systems (CDSS) to assist early diagnosis.

References

S. V. Birajdar, M. Mulchandani, F. Mazahir, and A. K. Yadav, “Chapter 1 - Dementia and neurodegenerative disorder: An introduction,” U. Gupta and P. B. T.-N.-B. A. for the T. of D. Kesharwani, Eds., Academic Press, 2023, pp. 1–36. doi: https://doi.org/10.1016/B978-0-12-824331-2.00007-8.

Z. He et al., “The current status, trends, and challenges of Alzheimer’s disease and other dementias in Asia (1990–2036),” Front. Public Heal., vol. 13, no. June, 2025, doi: 10.3389/fpubh.2025.1583339.

A. Javeed, A. L. Dallora, J. S. Berglund, A. Ali, L. Ali, and P. Anderberg, “Machine Learning for Dementia Prediction: A Systematic Review and Future Research Directions.,” J. Med. Syst., vol. 47, no. 1, p. 17, Feb. 2023, doi: 10.1007/s10916-023-01906-7.

F. Yi et al., “XGBoost-SHAP-based interpretable diagnostic framework for alzheimer’s disease.,” BMC Med. Inform. Decis. Mak., vol. 23, no. 1, p. 137, Jul. 2023, doi: 10.1186/s12911-023-02238-9.

ShashwatWork, “Dementia Prediction Dataset.” Kaggle, 2021. [Online]. Available: https://www.kaggle.com/datasets/shashwatwork/dementia-prediction-dataset

H. Darwis, A. N. P. Pagala, S. Anraeni, T. Amaliah, I. As’ad, and A. U. Tenripada, “Analysis of Public Sentiment about Childfree in Indonesia using Support Vector Machine Methods,” in 2025 19th International Conference on Ubiquitous Information Management and Communication (IMCOM), 2025, pp. 1–8. doi: 10.1109/IMCOM64595.2025.10857551.

Herman, H. Darwis, Nurfauziyah, R. Puspitasari, D. Widyawati, and A. Faradibah, “Comparative Analysis of Anxiety Disorder Classification Using Algorithm Naïve Bayes, Decision Tree and K-NN,” in 2025 19th International Conference on Ubiquitous Information Management and Communication (IMCOM), 2025, pp. 1–6. doi: 10.1109/IMCOM64595.2025.10857485.

R. R. Raja and H. Darwis, “A Comparative Study of Public Opinion on Indonesian Police : Examining Cases in the Aftermath of the Kanjuruhan Football Disaster,” vol. 6, no. 1, pp. 260–270, 2025, doi: https://doi.org/10.56705/ijodas.v6i2.235.

A. L. H. A. H. H. D. A. Kurnia, “SMOTE Technique Utilization in Cirrhosis Classification: A Comparison of Gradient Boosting and XGBoost,” JICO Int. J. Informatics Comput., no. Vol. 1 No. 1 (2025): May 2025, pp. 34–40, 2025, [Online]. Available: https://iaico.org/index.php/JICO/article/view/5/5

H. Darwis, R. Puspitasari, Purnawansyah, W. Astuti, D. Atmajaya, and M. Hasnawi, “A Deep Learning Approach for Improving Waste Classification Accuracy with ResNet50 Feature Extraction,” in 2025 19th International Conference on Ubiquitous Information Management and Communication (IMCOM), 2025, pp. 1–8. doi: 10.1109/IMCOM64595.2025.10857536.

A. R. Manga’, A. Nanda, A. N. Handayani, H. W. Herwanto, R. A. Asmara, and D. Lantara, “Comparison Between Single-Input and Multi-Input Classification with the Application of Canny Feature Extraction and Classification Algorithms on the Toraja Buffalo Dataset,” in 2025 19th International Conference on Ubiquitous Information Management and Communication (IMCOM), 2025, pp. 1–8. doi: 10.1109/IMCOM64595.2025.10857554.

Purnawansyah, A. P. Wibawa, T. Widiyaningtyas, Haviluddin, H. Darwis, and H. Azis, “An in-depth exploration of supervised and semi-supervised learning on face recognition,” vol. 15, no. 1, 2025, doi: doi:10.1515/comp-2025-0029.

A. R. Manga, Nirmala, H. Azis, F. Fattah, Y. Salim, and H. Darwis, “ResNet-50 for Flower Image Classification: A Comparative Study of Segmentation and Non-Segmentation Approaches,” in 2025 19th International Conference on Ubiquitous Information Management and Communication (IMCOM), 2025, pp. 1–6. doi: 10.1109/IMCOM64595.2025.10857520.

Y. Salim, A. F. Rakasyah, H. Darwis, Harlinda, Irawati, and A. R. Manga, “Comparative Analysis of Machine Learning Algorithms and Ensemble Techniques for Diverse Image Classification Tasks,” in 2025 19th International Conference on Ubiquitous Information Management and Communication (IMCOM), 2025, pp. 1–7. doi: 10.1109/IMCOM64595.2025.10857517.

H. Azis, P. Purnawansyah, and N. Alfiyyah, “Multiclass Classification on Nominal Value of Rupiah Banknotes Based on Image Processing,” Ilk. J. Ilm., vol. 16, pp. 87–99, Apr. 2024, doi: 10.33096/ilkom.v16i1.1784.87-99.

A. Gaffar, N. Azmi, E. Alwi, S. Abdullah, and D. Widyawati, Optimizing Brain Tumor Classification with ResNet-50 Feature Extraction and Machine Learning Algorithms. 2025. doi: 10.1109/IMCOM64595.2025.10857486.

A. R. Manga’, A. N. Handayani, H. W. Herwanto, R. A. Asmara, Y. I. Sulistya, and K. Kasmira, “Analysis of the Ensemble Method Classifier’s Performance on Handwritten Arabic Characters Dataset,” Ilk. J. Ilm., vol. 15, no. 1, pp. 186–192, 2023, doi: 10.33096/ilkom.v15i1.1357.186-192.

B. A. Maulana and N. Hidayati, “Churn Prediction in Credit Customers Using Random Forest and XGBoost Methods,” Indones. J. Data Sci., vol. 6, no. 1, pp. 82–90, 2025, doi: 10.56705/ijodas.v6i1.215.

R. Rahmaddeni, M. K. Anam, Y. Irawan, S. Susanti, and M. Jamaris, “Comparison of Support Vector Machine and XGBSVM in Analyzing Public Opinion on Covid-19 Vaccination,” Ilk. J. Ilm., vol. 14, no. 1, pp. 32–38, 2022, doi: 10.33096/ilkom.v14i1.1090.32-38.

G. Titaley, N. Rismayanti, A. Handayani, and J. Ardiansah, “Performance Comparison of Ensemble Learning Models for Brain Tumor Detection on Augmented MRI Datasets,” Ilk. J. Ilm., vol. 17, pp. 86–97, Aug. 2025, doi: 10.33096/ilkom.v17i2.2523.86-97.

Downloads

Published

2025-11-29

Issue

Vol. 3 No. 2 (2025): International Journal of Artificial Intelligence in Medical Issues

Section

Articles