Prediction of dementia using machine learning model and performance improvement with cuckoo algorithm

Abstract

Dementia is a brain disease that stays in the seventh position of death rate as per the report of the World Health Organization (WHO). Among the various types of dementia, Alzheimer’s disease has more than 70% of cases of dementia. The objective is to predict dementia disease from the open access series of imaging studies (OASIS) dataset using machine learning techniques. Also, the performance of the machine learning model is analyzed to improve the performance of the model using the cuckoo algorithm. In this paper, feature engineering has been focused and the prediction of dementia has been done using the OASIS dataset with the help of data mining techniques. Feature engineering is followed by prediction using the machine learning model Gaussian naïve Bayes (NB), support vector machine, and linear regression. Also, the best prediction model has been selected and done the validation. The evaluation metrics considered for validating the models are accuracy, precision, recall, and F1-Score and the highest values are 95%, 97%, 95%, and 95%. The Gaussian NB has been given these best results. The accuracy of the machine learning models has been increased by eliminating the factors which affect the performance of the models using the cuckoo algorithm.