Enhancing Alzheimer's Disease Diagnosis with K-NN: A Study on Pre-processed MRI Data

Authors

  • Imroatul Khasanah

DOI:

https://doi.org/10.56705/ijaimi.v2i1.150

Keywords:

Alzheimer's Disease, K-Nearest Neighbor, MRI, Class Imbalance, Hu Moments

Abstract

Alzheimer's disease (AD) is a progressive neurodegenerative disorder characterized by cognitive decline and memory loss, necessitating early and accurate diagnosis for effective intervention. This study evaluates the performance of the K-Nearest Neighbor (K-NN) algorithm on a pre-processed Alzheimer MRI dataset, focusing on the challenge of imbalanced classes. The dataset, sourced from Kaggle, comprises 6400 MRI images resized to 128x128 pixels and categorized into four classes: Non-Demented, Mild Demented, Moderate Demented, and Very Mild Demented. Pre-processing involved segmentation using the Canny edge detection method and feature extraction through Hu Moments. The dataset was split into training (80%) and testing (20%) sets, with features scaled to a mean of 0 and variance of 1. The K-NN algorithm was evaluated using cross-validation with five different k values, revealing moderate performance metrics: accuracy ranging from 45.86% to 50.47%, precision from 41.87% to 47.00%, recall from 45.86% to 50.47%, F1-score from 42.42% to 47.58%, and ROC AUC from 55.18% to 58.87%. The results highlight the significant impact of class imbalance on the algorithm's performance, particularly for the underrepresented Moderate Demented class. This study underscores the need for techniques to address class imbalance to enhance classification accuracy. Future research should explore advanced methods such as data augmentation, re-sampling, and ensemble learning, as well as the evaluation of other machine learning models. These findings contribute to the field of medical image analysis and have practical implications for improving diagnostic tools for Alzheimer's disease.

References

N. Rismayanti, A. Naswin, U. Zaky, M. Zakariyah, and D. A. Purnamasari, “Evaluating Thresholding-Based Segmentation and Humoment Feature Extraction in Acute Lymphoblastic Leukemia Classification using Gaussian Naive Bayes,” Int. J. Artif. Intell. Med. Issues, vol. 1, no. 2, 2023.

A. Maulidinnawati, “Classification Optimization of Skin Cancer Using the Adaboost Algorithm,” … J. Artif. Intell. Med. …, 2023.

N. D. Mu’azu, “K-nearest neighbor based computational intelligence and RSM predictive models for extraction of Cadmium from contaminated soil,” Ain Shams Eng. J., vol. 14, no. 4, 2023, doi: 10.1016/j.asej.2022.101944.

R. Siddalingappa, “K-nearest-neighbor algorithm to predict the survival time and classification of various stages of oral cancer: a machine learning approach,” F1000Research, vol. 11, p. 70, 2022, doi: 10.12688/f1000research.75469.2.

M. Novitasari, “Classification of House Buildings Based on Land Size Using the K-Nearest Neighbor Algorithm,” AIP Conference Proceedings, vol. 2499. 2022, doi: 10.1063/5.0104960.

E. Alcaras, “Machine Learning Approaches for Coastline Extraction from Sentinel-2 Images: K-Means and K-Nearest Neighbour Algorithms in Comparison,” Communications in Computer and Information Science, vol. 1651. pp. 368–379, 2022, doi: 10.1007/978-3-031-17439-1_27.

C. Feng, “An Enhanced Quantum K-Nearest Neighbor Classification Algorithm Based on Polar Distance,” Entropy, vol. 25, no. 1, 2023, doi: 10.3390/e25010127.

R. Setiawan and H. Oumarou, “Classification of Rice Grain Varieties Using Ensemble Learning and Image Analysis Techniques,” Indones. J. Data …, 2024.

A. Aisyah and S. Anraeni, “Analisis penerapan metode K-Nearest Neighbor (K-NN) pada dataset citra penyakit malaria,” Indones. J. Data Sci., 2022.

I. P. Putri, “Analisis Performa Metode K- Nearest Neighbor (KNN) dan Crossvalidation pada Data Penyakit Cardiovascular,” Indones. J. Data Sci., vol. 2, no. 1, pp. 21–28, 2021, doi: 10.33096/ijodas.v2i1.25.

M. R. Sharan, “Classification of Medicinal Leaf by Using Canny Edge Detection and SVM Classifier,” 2022 Int. Conf. Futur. Technol. INCOFT 2022, 2022, doi: 10.1109/INCOFT55651.2022.10094461.

S. K. T. Hwa, “Tuberculosis detection using deep learning and contrast-enhanced canny edge detected X-Ray images,” IAES Int. J. Artif. Intell., vol. 9, no. 4, pp. 713–720, 2020, doi: 10.11591/ijai.v9.i4.pp713-720.

Z. Hu, “Canny Algorithm Enabling Precise Offline Line Edge Roughness Acquisition in High-Resolution Lithography,” ACS Omega, 2022, doi: 10.1021/acsomega.2c06769.

Y. Jusman, “Classification System of Malaria Disease with Hu Moment Invariant and Support Vector Machines,” Proc. - 2022 2nd Int. Conf. Electron. Electr. Eng. Intell. Syst. ICE3IS 2022, pp. 365–368, 2022, doi: 10.1109/ICE3IS56585.2022.10010304.

Y. Jusman, “Classification System for Leukemia Cell Images based on Hu Moment Invariants and Support Vector Machines,” Proc. - 2021 11th IEEE Int. Conf. Control Syst. Comput. Eng. ICCSCE 2021, pp. 137–141, 2021, doi: 10.1109/ICCSCE52189.2021.9530974.

B. P. Sari, “Classification System for Cervical Cell Images based on Hu Moment Invariants Methods and Support Vector Machine,” 2021 Int. Conf. Intell. Technol. CONIT 2021, 2021, doi: 10.1109/CONIT51480.2021.9498353.

N. Rismayanti and A. P. Utami, “Improving Multi-Class Classification on 5-Celebrity-Faces Dataset using Ensemble Classification Methods,” Indones. J. Data …, 2023.

Q. Zhang, “Enhanced Harris hawks optimization-based fuzzy k-nearest neighbor algorithm for diagnosis of Alzheimer’s disease,” Comput. Biol. Med., vol. 165, 2023, doi: 10.1016/j.compbiomed.2023.107392.

D. C. E. Saputra, “K-Nearest Neighbor of Beta Signal Brainwave to Accelerate Detection of Concentration on Student Learning Outcomes,” Eng. Lett., vol. 30, no. 1, pp. 318–324, 2022.

D. Lu, “Effective detection of Alzheimer’s disease by optimizing fuzzy K-nearest neighbors based on salp swarm algorithm,” Comput. Biol. Med., vol. 159, 2023, doi: 10.1016/j.compbiomed.2023.106930.

H. Azis, L. Syafie, F. Fattah, and ..., “Unveiling Algorithm Classification Excellence: Exploring Calendula and Coreopsis Flower Datasets with Varied Segmentation Techniques,” 2024 18th Int. …, 2024.

M. M. Baharuddin, T. Hasanuddin, and H. Azis, “Analisis Performa Metode K-Nearest Neighbor untuk Identifikasi Jenis Kaca,” Ilk. J. Ilm., vol. 11, no. 28, pp. 269–274, 2019.

H. Azis, F. Fattah, and P. Putri, “Performa Klasifikasi K-NN dan Cross-validation pada Data Pasien Pengidap Penyakit Jantung,” Ilk. J. Ilm., vol. 12, no. 2, pp. 81–86, 2020.

H. Azis, F. T. Admojo, and E. Susanti, “Analisis Perbandingan Performa Metode Klasifikasi pada Dataset Multiclass Citra Busur Panah,” Techno.Com, vol. 19, no. 3, 2020.

Downloads

Published

2024-05-30

Issue

Vol. 2 No. 1 (2024): International Journal of Artificial Intelligence in Medical Issues

Section

Articles