Cargando…

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

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 counterp...

Descripción completa

Detalles Bibliográficos
Autores principales: Han, Tianyu, Nebelung, Sven, Pedersoli, Federico, Zimmermann, Markus, Schulze-Hagen, Maximilian, Ho, Michael, Haarburger, Christoph, Kiessling, Fabian, Kuhl, Christiane, Schulz, Volkmar, Truhn, Daniel
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8280105/
https://www.ncbi.nlm.nih.gov/pubmed/34262044
http://dx.doi.org/10.1038/s41467-021-24464-3
_version_ 1783722582125576192
author Han, Tianyu
Nebelung, Sven
Pedersoli, Federico
Zimmermann, Markus
Schulze-Hagen, Maximilian
Ho, Michael
Haarburger, Christoph
Kiessling, Fabian
Kuhl, Christiane
Schulz, Volkmar
Truhn, Daniel
author_facet Han, Tianyu
Nebelung, Sven
Pedersoli, Federico
Zimmermann, Markus
Schulze-Hagen, Maximilian
Ho, Michael
Haarburger, Christoph
Kiessling, Fabian
Kuhl, Christiane
Schulz, Volkmar
Truhn, Daniel
author_sort Han, Tianyu
collection PubMed
description 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.
format Online
Article
Text
id pubmed-8280105
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-82801052021-07-20 Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization Han, Tianyu Nebelung, Sven Pedersoli, Federico Zimmermann, Markus Schulze-Hagen, Maximilian Ho, Michael Haarburger, Christoph Kiessling, Fabian Kuhl, Christiane Schulz, Volkmar Truhn, Daniel Nat Commun Article 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. Nature Publishing Group UK 2021-07-14 /pmc/articles/PMC8280105/ /pubmed/34262044 http://dx.doi.org/10.1038/s41467-021-24464-3 Text en © The Author(s) 2021 https://creativecommons.org/licenses/by/4.0/Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons license, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons license and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Han, Tianyu
Nebelung, Sven
Pedersoli, Federico
Zimmermann, Markus
Schulze-Hagen, Maximilian
Ho, Michael
Haarburger, Christoph
Kiessling, Fabian
Kuhl, Christiane
Schulz, Volkmar
Truhn, Daniel
Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization
title Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization
title_full Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization
title_fullStr Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization
title_full_unstemmed Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization
title_short Advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization
title_sort advancing diagnostic performance and clinical usability of neural networks via adversarial training and dual batch normalization
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8280105/
https://www.ncbi.nlm.nih.gov/pubmed/34262044
http://dx.doi.org/10.1038/s41467-021-24464-3
work_keys_str_mv AT hantianyu advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT nebelungsven advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT pedersolifederico advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT zimmermannmarkus advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT schulzehagenmaximilian advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT homichael advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT haarburgerchristoph advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT kiesslingfabian advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT kuhlchristiane advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT schulzvolkmar advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization
AT truhndaniel advancingdiagnosticperformanceandclinicalusabilityofneuralnetworksviaadversarialtraininganddualbatchnormalization