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...
Autores principales: | , , , , , , , , , , |
---|---|
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 |