A Weighted Ensemble Approach with Multiple Pre-Trained Deep Learning Models for Classification of Stroke

Authors

  • Rusul Ali Jabbar Alhatemi Department of Electronics and Computer Engineering, Çankırı Karatekin University, Türkiye
  • Serkan Savaş Department of Computer Engineering, Kırıkkale University, Türkiye https://orcid.org/0000-0003-3440-6271

DOI:

https://doi.org/10.47852/bonviewMEDIN32021963

Keywords:

ensemble learning, transfer learning, stroke, biomedical image classification, MR images, deep learning

Abstract

Stroke ranks as one of the deadliest diseases globally, emphasizing the crucial need for early diagnosis. This study aims to create a two-stage classification system for stroke and non-stroke images to support early clinical detection. Deep learning, a cornerstone of diagnosis, detection, and prompt treatment, is the primary methodology. Transfer learning adapts successful deep learning architectures for diverse problems, and ensemble learning combines multiple classifiers for enhanced results. These two techniques are applied to classify stroke using a dataset of stroke and normal images. In the initial stage, six pre-trained models are fine-tuned, with DenseNet, Xception, and EfficientNetB2 emerging as the top performers, achieving validation accuracies of 98.4%, 98.4%, and 98%, respectively. These models serve as base learners within an ensemble framework. A weighted average ensemble method combines them, resulting in a remarkable 99.84% accuracy on a reserved test dataset. This approach exhibits promise for stroke detection, a life-threatening condition, while also demonstrating the effectiveness of ensemble techniques in enhancing model performance.

 

Received: 27 October 2023 | Revised: 24 November 2023 | Accepted: 19 December 2023

 

Conflicts of Interest

The authors declare that they have no conflicts of interest to this work.

 

Data Availability Statement

Since the data used in the study were removed from the Kaggle platform after the study was conducted, no data link is provided. However, the study data can be sent to researchers who request it, with privacy and ethical restrictions.

Downloads

Published

2023-12-27

How to Cite

Alhatemi, R. A. J., & Savaş, S. (2023). A Weighted Ensemble Approach with Multiple Pre-Trained Deep Learning Models for Classification of Stroke. Medinformatics. https://doi.org/10.47852/bonviewMEDIN32021963

Issue

Section

Research Articles