IAES International Journal of Artificial Intelligence (IJ-AI)
Vol 14, No 1: February 2025

Balancing and metaheuristic techniques for improving machine learning models in brain stroke prediction

Aouragh, Abd Allah (Unknown)
Bahaj, Mohamed (Unknown)
Toufik, Fouad (Unknown)



Article Info

Publish Date
01 Feb 2025

Abstract

A brain stroke, medically referred to as a stroke, represents a critical condition triggered by the disruption of blood flow to a region of the brain. Early detection of stroke is crucial to prevent fatal complications. In this study, we worked with an unbalanced dataset of 4981 entries on stroke, which we balanced using the K-means synthetic minority over-sampling technique (KMeansSMOTE) algorithm. We then employed five machine learning algorithms: decision tree, random forest, support vector machine, K-nearest neighbors, and gradient boosting. We compared the hyperparameter optimization of these algorithms using four metaheuristic techniques: gray wolf optimization, particle swarm optimization, genetic algorithm, and artificial bee colony. The models' effectiveness was evaluated using multiple metrics, such as accuracy, recall, precision, F1-score, and area under the receiver operating characteristic curve. Our findings indicate that the random forest optimized by the genetic algorithm achieved the best performance, with an accuracy of 97.39% and an F1-score of 97.35%. This study highlights the effectiveness of balancing and metaheuristics techniques in optimizing machine learning models for stroke forecasting.

Copyrights © 2025






Journal Info

Abbrev

IJAI

Publisher

Subject

Computer Science & IT Engineering

Description

IAES International Journal of Artificial Intelligence (IJ-AI) publishes articles in the field of artificial intelligence (AI). The scope covers all artificial intelligence area and its application in the following topics: neural networks; fuzzy logic; simulated biological evolution algorithms (like ...