Denoising diffusion adversarial models for unconditional medical image generation

Abstract

Unconditional medical image synthesis is the task of generating realistic and diverse medical images from random noise without any prior information or constraints. Synthesizing realistic medical images can enrich the quality and diversity of medical imaging datasets, which in turn, enhance the performance and generalization of deep learning models for medical imaging. Prevalent approach for synthesizing medical images involves generative adversarial networks (GAN) or denoising diffusion probabilistic models (DDPM). However, GAN models that implicitly learn the image distribution are prone to limited sample fidelity and diversity. On the other hand, diffusion models suffer from slow sampling speed due to small diffusion steps. In this paper, we propose a novel diffusion-based method for unconditional medical image synthesis, Diff-Med-Synth, that generates realistic and diverse medical images from random noise. Diff-Med-Synth combines the advantages of denoising diffusion probabilistic models and GANs to achieve fast and efficient image sampling. We evaluate our method on two multi-contrast MRI datasets and show that it outperforms state-of-the-art methods in terms of quality, diversity, and fidelity of the synthesized images.

Koşulsuz tıbbi görüntü sentezi, önceden bilgi veya kısıt olmaksızın rastgele gürültüden gerçekçi ve çeşitli tıbbi görüntüler üretme görevidir. Gerçekçi tıbbi görüntüler sentezlemek, tıbbi görüntüleme veri kümelerinin kalitesini ve çeşitliliğini zenginleştirebilir, bu da sırayla tıbbi görüntüleme için derin öğrenme modellerinin performansını ve genelleştirilmesini artırabilir. Tıbbi görüntüler sentezlemek için yaygın yaklaşım, üretici çekişmeli ağlar (GAN) veya gürültüyü giderme difüzyon olasılık modelleri (DDPM) içerir. Ancak, GAN modelleri görüntü dağılımını dolaylı olarak öğrenir ve sınırlı örnek sadakati ve çeşitliliğe eğilimlidir. Öte yandan, difüzyon modelleri küçük difüzyon adımları nedeniyle yavaş örnekleme hızından muzdariptir. Bu çalışmada, rastgele gürültüden gerçekçi ve hızlı şekilde tıbbi görüntüler üreten yeni bir difüzyon tabanlı yöntem olan Diff-Med-Synth’i öneriyoruz. Diff-Med-Synth, hızlı ve verimli görüntü örnekleme elde etmek için gürültüyü giderme difüzyon olasılık modellerinin ve GAN’ların avantajlarını birleştirir. Yöntemimizi iki çok kontrastlı MRG veri kümesinde değerlendiriyoruz ve sentezlenen görüntülerin kalitesi, çeşitliliği ve sadakati açısından son teknik yöntemleri geride bıraktığını gösteriyoruz.