Evaluating neural network robustness for melanoma classification using mutual information

Molly O'Brien; Julia Bukowski; Greg Hager; Aria Pezeshk; Mathias Unberath

doi:10.1117/12.2612192

4 April 2022 Evaluating neural network robustness for melanoma classification using mutual information

Molly O'Brien, Julia Bukowski, Greg Hager, Aria Pezeshk, Mathias Unberath

Proceedings Volume 12032, Medical Imaging 2022: Image Processing; 120320P (2022) https://doi.org/10.1117/12.2612192
Event: SPIE Medical Imaging, 2022, San Diego, California, United States

Abstract

Deep Neural Networks (DNN) have made rapid progress in medical image analysis, and it is now common to see algorithms proposed in the literature that claim to meet or exceed clinician performance. However, DNNs typically operate like black box systems. Deployment of these algorithms for safety-critical tasks, such as medical applications, is therefore challenging without methods to characterize and identify the robustness of DNNs in a generalized setting. Furthermore, DNNs are known to be sensitive to small changes in input data, and may rely on spurious correlations seen in the training samples which precludes them from generalizing to data with a different distribution. We previously proposed an attribute-ranking algorithm that ranks discrete data attributes by how informative they are about the DNN performance using Mutual Information. In this study we leverage this algorithm in a novel way to determine if the data attributes that impact the DNN performance are clinically relevant and thereby whether the DNN predictions are robust. We demonstrate the applicability of this method on melanoma classification with a DNN trained on the publicly available HAM 10,000 dataset and achieve 0.855 AUC on the held-out HAM test set. Our analysis identifies that image saturation, which is not a clinically relevant feature, is highly indicative of both whether an image is melanomic in the HAM training data and whether the DNN prediction is correct. Further testing reveals that when the classifier is tested on the SIIM-ISIC Melanoma dataset, where the correlation between image saturation and melanoma is not present, the classifier achieves only an AUC of 0.591, confirming that the DNN is not robust.

Conference Presentation

Citation Download Citation

Molly O'Brien, Julia Bukowski, Greg Hager, Aria Pezeshk, and Mathias Unberath "Evaluating neural network robustness for melanoma classification using mutual information", Proc. SPIE 12032, Medical Imaging 2022: Image Processing, 120320P (4 April 2022); https://doi.org/10.1117/12.2612192

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