keyboard_arrow_up
No Masks Needed: Explainable AI for Deriving Segmentation from Classification

Authors

Mosong Ma1, Tania Stathaki1, and Michalis Lazarou2, 1Imperial College London, UK, 2University of Surrey, UK

Abstract

Medical image segmentation is vital for modern healthcare and is a key element of computer-aided diagnosis. While recent advancements in computer vision have explored unsu- pervised segmentation using pre-trained models, these methods have not been translated well to the medical imaging domain. In this work, we introduce a novel approach that fine-tunes pre- trained models specifically for medical images, achieving accurate segmentation with extensive processing. Our method integrates Explainable AI to generate relevance scores, enhancing the segmentation process. Unlike traditional methods that excel in standard benchmarks but falter in medical applications, our approach achieves improved results on datasets like CBIS-DDSM, NuInsSeg and Kvasir-SEG.

Keywords

Medical Image Segmentation, Explainable AI, Transfer Learning.

Full Text  Volume 15, Number 17