Adapting Classification Neural Network Architectures for Medical Image Segmentation Using Explainable AI
No Thumbnail Available
Date
2025-02-13
Journal Title
Journal ISSN
Volume Title
Publisher
Abstract
Segmentation neural networks are widely used in medical imaging to identify anomalies that may impact patient health. Despite their effectiveness, these networks face significant challenges, including the need for extensive annotated patient data, time-consuming manual segmentation processes and restricted data access due to privacy concerns. In contrast, classification neural networks, similar to segmentation neural networks, capture essential parameters for identifying objects during training. This paper leverages this characteristic, combined with explainable artificial intelligence (XAI) techniques, to address the challenges of segmentation. By adapting classification neural networks for segmentation tasks, the proposed approach reduces dependency on manual segmentation. To demonstrate this concept, the Medical Segmentation Decathlon ‘Brain Tumours’ dataset was utilised. A ResNet classification neural network was trained, and XAI tools were applied to generate segmentation-like outputs. Our findings reveal that GuidedBackprop is among the most efficient and effective methods, producing heatmaps that closely resemble segmentation masks by accurately highlighting the entirety of the target object.
Description
Publisher Copyright: © 2025 by the authors.
Keywords
medical imaging, classification models, image segmentation, explainable artificial intelligence, neural networks, 1.2 Computer and information sciences, 2.6 Medical engineering, 1.1. Scientific article indexed in Web of Science and/or Scopus database
Citation
Nikulins, A, Edelmers, E, Sudars, K & Polaka, I 2025, 'Adapting Classification Neural Network Architectures for Medical Image Segmentation Using Explainable AI', Journal of Imaging, vol. 11, no. 2, 55. https://doi.org/10.3390/jimaging11020055