Research Article Open Access

A 2-Tier Stacking Ensemble Classifier for Disease Classification

K. Sukanya Varshini1 and R. Annie Uthra1
  • 1 Department of Computational Intelligence, Faculty of Engineering and Technology, SRM Institute of Science and Technology, Kattankulathur, Chennai, India

Abstract

Diarrhea, dysentery, and dehydration are top of the list of high mortality-causing diseases in children under the age group of five. Despite the tremendous growth of machine learning in the field of medical research, still, some areas remain untouched, especially, the pediatric department. Since medical data is used in the proposed work, even a slight increase in the accuracy of the model will be of great importance. In this study, we propose a 2-tier stacking ensemble method for disease identification. Initially, the data is pre-processed and sent to train the tier-1 machine learning models. Based on the majority voting the metadata is selected and sent to the next tier and finally, at the meta-classifier level disease classification happens. Performance metrics like the accuracy, precision, recall, F1-score and mean absolute error of the individual machine learning algorithms were analyzed and were used to compare with the proposed stacking ensemble method. The results proved that the 2-tier stacking model proposed in the work shows an accuracy of 95.41%, a precision of 94.47%, a recall value of 92.78%, and an F1-score of 93.50%. The proposed model achieved a high accuracy value when compared to the other machine learning models.

Journal of Computer Science
Volume 19 No. 7, 2023, 796-811

DOI: https://doi.org/10.3844/jcssp.2023.796.811

Submitted On: 20 January 2023 Published On: 7 June 2023

How to Cite: Varshini, K. S. & Uthra, R. A. (2023). A 2-Tier Stacking Ensemble Classifier for Disease Classification. Journal of Computer Science, 19(7), 796-811. https://doi.org/10.3844/jcssp.2023.796.811

  • 1,828 Views
  • 1,062 Downloads
  • 0 Citations

Download

Keywords

  • Machine Learning
  • Disease Identification
  • Ensemble Stacking Model
  • Dysentery
  • Diarrhea
  • Dehydration
  • Pediatric