Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization

Nat Commun. 2021 Jul 14;12(1):4315. doi: 10.1038/s41467-021-24464-3.

Abstract

Unmasking the decision making process of machine learning models is essential for implementing diagnostic support systems in clinical practice. Here, we demonstrate that adversarially trained models can significantly enhance the usability of pathology detection as compared to their standard counterparts. We let six experienced radiologists rate the interpretability of saliency maps in datasets of X-rays, computed tomography, and magnetic resonance imaging scans. Significant improvements are found for our adversarial models, which are further improved by the application of dual-batch normalization. Contrary to previous research on adversarially trained models, we find that accuracy of such models is equal to standard models, when sufficiently large datasets and dual batch norm training are used. To ensure transferability, we additionally validate our results on an external test set of 22,433 X-rays. These findings elucidate that different paths for adversarial and real images are needed during training to achieve state of the art results with superior clinical interpretability.

MeSH terms

  • Humans
  • Machine Learning
  • Neural Networks, Computer*
  • Radiographic Image Interpretation, Computer-Assisted / methods*
  • Reproducibility of Results