Abstract:Denoising diffusion models have found applications in image segmentation by generating segmented masks conditioned on images. Existing studies predominantly focus on adjusting model architecture or improving inference, such as test-time sampling strategies. In this work, we focus on improving the training strategy and propose a novel recycling method. During each training step, a segmentation mask is first predicted given an image and a random noise. This predicted mask, which replaces the conventional ground truth mask, is used for denoising task during training. This approach can be interpreted as aligning the training strategy with inference by eliminating the dependence on ground truth masks for generating noisy samples. Our proposed method significantly outperforms standard diffusion training, self-conditioning, and existing recycling strategies across multiple medical imaging data sets: muscle ultrasound, abdominal CT, prostate MR, and brain MR. This holds for two widely adopted sampling strategies: denoising diffusion probabilistic model and denoising diffusion implicit model. Importantly, existing diffusion models often display a declining or unstable performance during inference, whereas our novel recycling consistently enhances or maintains performance. We show that, under a fair comparison with the same network architectures and computing budget, the proposed recycling-based diffusion models achieved on-par performance with non-diffusion-based supervised training. By ensembling the proposed diffusion and the non-diffusion models, significant improvements to the non-diffusion models have been observed across all applications, demonstrating the value of this novel training method. This paper summarizes these quantitative results and discusses their values, with a fully reproducible JAX-based implementation, released at <a class="link-external link-https" href="https://github.com/mathpluscode/ImgX-DiffSeg" rel="external noopener nofollow">this https URL</a>.

Edge-and-Mask Integration-Driven Diffusion Models for Medical Image Segmentation

MedSegDiff: Medical Image Segmentation with Diffusion Probabilistic Model

Accelerating denoising diffusion probabilistic model via truncated inverse processes for medical image segmentation

Explicit-implicit priori knowledge-based diffusion model for generative medical image segmentation

BerDiff: Conditional Bernoulli Diffusion Model for Medical Image Segmentation

Conditional Diffusion Models for Weakly Supervised Medical Image Segmentation

Diffusion model-based text-guided enhancement network for medical image segmentation

Enhancing Label-efficient Medical Image Segmentation with Text-guided Diffusion Models

ERSegDiff: a diffusion-based model for edge reshaping in medical image segmentation

Importance of Aligning Training Strategy with Evaluation for Diffusion Models in 3D Multiclass Segmentation

HiDiff: Hybrid Diffusion Framework for Medical Image Segmentation

A generic plug & play diffusion-based denosing module for medical image segmentation

PGDiffSeg: Prior-Guided Denoising Diffusion Model with Parameter-Shared Attention for Breast Cancer Segmentation

Introducing Shape Prior Module in Diffusion Model for Medical Image Segmentation

DiffuseExpand: Expanding dataset for 2D medical image segmentation using diffusion models

Analysing Diffusion Segmentation for Medical Images

Denoising Diffusion Semantic Segmentation with Mask Prior Modeling

A Recycling Training Strategy for Medical Image Segmentation with Diffusion Denoising Models

Distribution Aligned Diffusion and Prototype-guided network for Unsupervised Domain Adaptive Segmentation

FDiff-Fusion:Denoising diffusion fusion network based on fuzzy learning for 3D medical image segmentation

Enhancing Medical Image Segmentation with Deep Learning and Diffusion Models