A benchmark for neural network robustness in skin cancer classification

Publikation: Beitrag in FachzeitschriftForschungsartikelBeigetragenBegutachtung

Beitragende

Abstract

Background: One prominent application for deep learning–based classifiers is skin cancer classification on dermoscopic images. However, classifier evaluation is often limited to holdout data which can mask common shortcomings such as susceptibility to confounding factors. To increase clinical applicability, it is necessary to thoroughly evaluate such classifiers on out-of-distribution (OOD) data. Objective: The objective of the study was to establish a dermoscopic skin cancer benchmark in which classifier robustness to OOD data can be measured. Methods: Using a proprietary dermoscopic image database and a set of image transformations, we create an OOD robustness benchmark and evaluate the robustness of four different convolutional neural network (CNN) architectures on it. Results: The benchmark contains three data sets—Skin Archive Munich (SAM), SAM-corrupted (SAM-C) and SAM-perturbed (SAM-P)—and is publicly available for download. To maintain the benchmark's OOD status, ground truth labels are not provided and test results should be sent to us for assessment. The SAM data set contains 319 unmodified and biopsy-verified dermoscopic melanoma (n = 194) and nevus (n = 125) images. SAM-C and SAM-P contain images from SAM which were artificially modified to test a classifier against low-quality inputs and to measure its prediction stability over small image changes, respectively. All four CNNs showed susceptibility to corruptions and perturbations. Conclusions: This benchmark provides three data sets which allow for OOD testing of binary skin cancer classifiers. Our classifier performance confirms the shortcomings of CNNs and provides a frame of reference. Altogether, this benchmark should facilitate a more thorough evaluation process and thereby enable the development of more robust skin cancer classifiers.

Details

OriginalspracheEnglisch
Seiten (von - bis)191-199
Seitenumfang9
FachzeitschriftEuropean journal of cancer
Jahrgang155
PublikationsstatusVeröffentlicht - Sept. 2021
Peer-Review-StatusJa

Externe IDs

ORCID /0000-0002-2164-4644/work/148607192
Scopus 85110706464
PubMed 34388516
ORCID /0000-0003-4340-9706/work/159608222

Schlagworte

Ziele für nachhaltige Entwicklung

Schlagwörter

  • Artificial intelligence, Benchmarking, Deep learning, Dermatology, Melanoma, Nevus, Skin neoplasms