Class activation mapping and uncertainty estimation in multi-organ segmentation

M. S. Sadique; W. Farzana; A. Temtam; K. M. Iftekharuddin

doi:10.1117/12.2655508

7 April 2023 Class activation mapping and uncertainty estimation in multi-organ segmentation

M. S. Sadique, W. Farzana, A. Temtam, K. M. Iftekharuddin

Proceedings Volume 12465, Medical Imaging 2023: Computer-Aided Diagnosis; 124650V (2023) https://doi.org/10.1117/12.2655508
Event: SPIE Medical Imaging, 2023, San Diego, California, United States

Abstract

Deep learning (DL)-based medical imaging and image segmentation algorithms achieve impressive performance on many benchmarks. Yet the efficacy of deep learning methods for future clinical applications may become questionable due to the lack of ability to reason with uncertainty and interpret probable areas of failures in prediction decisions. Therefore, it is desired that such a deep learning model for segmentation classification is able to reliably predict its confidence measure and map back to the original imaging cases to interpret the prediction decisions. In this work, uncertainty estimation for multiorgan segmentation task is evaluated to interpret the predictive modeling in DL solutions. We use the state-of-the-art nnU-Net to perform segmentation of 15 abdominal organs (spleen, right kidney, left kidney, gallbladder, esophagus, liver, stomach, aorta, inferior vena cava, pancreas, right adrenal gland, left adrenal gland, duodenum, bladder, prostate/uterus) using 200 patient cases for the Multimodality Abdominal Multi-Organ Segmentation Challenge 2022. Further, the softmax probabilities from different variants of nnU-Net are used to compute the knowledge uncertainty in the deep learning framework. Knowledge uncertainty from ensemble of DL models is utilized to quantify and visualize class activation map for two example segmented organs. The preliminary result of our model shows that class activation maps may be used to interpret the prediction decision made by the DL model used in this study.

Conference Presentation

Citation Download Citation

M. S. Sadique, W. Farzana, A. Temtam, and K. M. Iftekharuddin "Class activation mapping and uncertainty estimation in multi-organ segmentation", Proc. SPIE 12465, Medical Imaging 2023: Computer-Aided Diagnosis, 124650V (7 April 2023); https://doi.org/10.1117/12.2655508

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available