I3 - Understanding and Visualizing Generalization UNets

Abhejit Rajagopal, Vamshi Chowdary Madala, Thomas A Hope, Peder Larson

Show abstract - Show schedule - Proceedings - PDF - Reviews

Fully-convolutional neural networks, such as the 2D or 3D UNet, are now pervasive in medical imaging for semantic segmentation, classification, image denoising, domain translation, and reconstruction. However, evaluation of UNet performance, as with most CNNs, has mostly been relegated to evaluation of a few performance metrics (e.g. accuracy, IoU, SSIM, etc.) using the network\'s final predictions, which provides little insight into important issues such as dataset shift that occur in clinical application. In this paper, we propose techniques for understanding and visualizing the generalization performance of UNets in image classification and regression tasks, giving rise to metrics that are indicative of performance on a withheld test-set without the need for groundtruth annotations.
Hide abstract

Friday 9th July
I1-3 (long): Interpretability and Explainable AI - 13:00 - 13:30 (UTC+2)
Hide schedule


Can't display slides, your browser doesn't support embedding PDFs.

Download slides