Author:
Müller-Franzes Gustav,Niehues Jan Moritz,Khader Firas,Arasteh Soroosh Tayebi,Haarburger Christoph,Kuhl Christiane,Wang Tianci,Han Tianyu,Nolte Teresa,Nebelung Sven,Kather Jakob Nikolas,Truhn Daniel
Abstract
AbstractAlthough generative adversarial networks (GANs) can produce large datasets, their limited diversity and fidelity have been recently addressed by denoising diffusion probabilistic models, which have demonstrated superiority in natural image synthesis. In this study, we introduce Medfusion, a conditional latent DDPM designed for medical image generation, and evaluate its performance against GANs, which currently represent the state-of-the-art. Medfusion was trained and compared with StyleGAN-3 using fundoscopy images from the AIROGS dataset, radiographs from the CheXpert dataset, and histopathology images from the CRCDX dataset. Based on previous studies, Progressively Growing GAN (ProGAN) and Conditional GAN (cGAN) were used as additional baselines on the CheXpert and CRCDX datasets, respectively. Medfusion exceeded GANs in terms of diversity (recall), achieving better scores of 0.40 compared to 0.19 in the AIROGS dataset, 0.41 compared to 0.02 (cGAN) and 0.24 (StyleGAN-3) in the CRMDX dataset, and 0.32 compared to 0.17 (ProGAN) and 0.08 (StyleGAN-3) in the CheXpert dataset. Furthermore, Medfusion exhibited equal or higher fidelity (precision) across all three datasets. Our study shows that Medfusion constitutes a promising alternative to GAN-based models for generating high-quality medical images, leading to improved diversity and less artifacts in the generated images.
Publisher
Springer Science and Business Media LLC
Reference50 articles.
1. Samala, R. K. et al. Breast cancer diagnosis in digital breast tomosynthesis: Effects of training sample size on multi-stage transfer learning using deep neural nets. IEEE Trans. Med Imaging 38(3), 686–696. https://doi.org/10.1109/TMI.2018.2870343 (2019).
2. Cho, J., Lee, K., Shin, E., Choy, G., Do, S. How much data is needed to train a medical image deep learning system to achieve necessary high accuracy? arXiv (accessed 11 November 2022)
3. http://arxiv.org/abs/1511.06348 (2016).
4. Wang, T. et al. A review on medical imaging synthesis using deep learning and its clinical applications. J. Appl. Clin. Med. Phys. 22(1), 11–36. https://doi.org/10.1002/acm2.13121 (2021).
5. Han, T. et al. Breaking medical data sharing boundaries by using synthesized radiographs. Sci. Adv. 6(49), 7973. https://doi.org/10.1126/sciadv.abb7973 (2020).
Cited by
32 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献