DOI: 10.2478/ijssis-2024-0013 ISSN: 1178-5608

Explainable AI for binary and multi-class classification of leukemia using a modified transfer learning ensemble model

Nilkanth Mukund Deshpande, Shilpa Gite, Biswajeet Pradhan
  • Electrical and Electronic Engineering
  • Control and Systems Engineering

Abstract

In leukemia diagnosis, automating the process of decision-making can reduce the impact of individual pathologists' expertise. While deep learning models have demonstrated promise in disease diagnosis, combining them can yield superior results. This research introduces an ensemble model that merges two pre-trained deep learning models, namely, VGG-16 and Inception, using transfer learning. It aims to accurately classify leukemia subtypes using real and standard dataset images, focusing on interpretability. Therefore, the use of Local Interpretable Model-Agnostic Explanations (LIME) is employed to achieve interpretability. The ensemble model achieves an accuracy of 83.33% in binary classification, outperforming individual models. In multi-class classification, VGG-16 and Inception reach accuracies of 83.335% and 93.33%, respectively, while the ensemble model reaches an accuracy of 100%.

More from our Archive