Robustness of convolutional neural networks in recognition of pigmented skin lesions

Research output: Contribution to journalResearch articleContributedpeer-review

Contributors

  • Roman C Maron - , German Cancer Research Center (DKFZ) (Author)
  • Sarah Haggenmüller - , German Cancer Research Center (DKFZ) (Author)
  • Christof von Kalle - , Charité – Universitätsmedizin Berlin (Author)
  • Jochen S Utikal - , SRH University of Applied Sciences Heidelberg (Author)
  • Friedegund Meier - , University Hospital Carl Gustav Carus Dresden (Author)
  • Frank F Gellrich - , Department of Dermatology, National Center for Tumor Diseases (NCT) Dresden, University Hospital Carl Gustav Carus Dresden (Author)
  • Axel Hauschild - , University Hospital Schleswig-Holstein Campus Kiel (Author)
  • Lars E French - , Hospital de Basurto (Author)
  • Max Schlaak - , Hospital de Basurto (Author)
  • Kamran Ghoreschi - , Department of Dermatology (Author)
  • Heinz Kutzner - , Dermatopathology Laboratory (Author)
  • Markus V Heppt - , State Vocational Colleges at the University Hospital Erlangen (Author)
  • Sebastian Haferkamp - , University Hospital Regensburg (Author)
  • Wiebke Sondermann - , University Hospital Essen (Author)
  • Dirk Schadendorf - , University Hospital Essen (Author)
  • Bastian Schilling - , University Hospital of Würzburg (Author)
  • Achim Hekler - , German Cancer Research Center (DKFZ) (Author)
  • Eva Krieghoff-Henning - , German Cancer Research Center (DKFZ) (Author)
  • Jakob N Kather - , Else Kröner Fresenius Center for Digital Health, National Center for Tumor Diseases (NCT) Heidelberg, German Cancer Research Center (DKFZ) (Author)
  • Stefan Fröhling - , German Cancer Research Center (DKFZ) (Author)
  • Daniel B Lipka - , German Cancer Research Center (DKFZ) (Author)
  • Titus J Brinker - , German Cancer Research Center (DKFZ) (Author)

Abstract

BACKGROUND: A basic requirement for artificial intelligence (AI)-based image analysis systems, which are to be integrated into clinical practice, is a high robustness. Minor changes in how those images are acquired, for example, during routine skin cancer screening, should not change the diagnosis of such assistance systems.

OBJECTIVE: To quantify to what extent minor image perturbations affect the convolutional neural network (CNN)-mediated skin lesion classification and to evaluate three possible solutions for this problem (additional data augmentation, test-time augmentation, anti-aliasing).

METHODS: We trained three commonly used CNN architectures to differentiate between dermoscopic melanoma and nevus images. Subsequently, their performance and susceptibility to minor changes ('brittleness') was tested on two distinct test sets with multiple images per lesion. For the first set, image changes, such as rotations or zooms, were generated artificially. The second set contained natural changes that stemmed from multiple photographs taken of the same lesions.

RESULTS: All architectures exhibited brittleness on the artificial and natural test set. The three reviewed methods were able to decrease brittleness to varying degrees while still maintaining performance. The observed improvement was greater for the artificial than for the natural test set, where enhancements were minor.

CONCLUSIONS: Minor image changes, relatively inconspicuous for humans, can have an effect on the robustness of CNNs differentiating skin lesions. By the methods tested here, this effect can be reduced, but not fully eliminated. Thus, further research to sustain the performance of AI classifiers is needed to facilitate the translation of such systems into the clinic.

Details

Original languageEnglish
Pages (from-to)81-91
Number of pages11
JournalEuropean journal of cancer
Volume145
Publication statusPublished - Mar 2021
Peer-reviewedYes

External IDs

Scopus 85099254625
ORCID /0000-0002-2164-4644/work/148607198

Keywords

Sustainable Development Goals

Keywords

  • Dermoscopy, Diagnosis, Computer-Assisted, Diagnosis, Differential, Humans, Image Interpretation, Computer-Assisted, Melanoma/pathology, Neural Networks, Computer, Nevus/pathology, Predictive Value of Tests, Reproducibility of Results, Skin Neoplasms/pathology