Abstract:Medical image segmentation is crucial in accurately identifying and delineating regions of interest in medical images, which can inform the diagnosis and treatment of various diseases. Therefore, developing high-performance computer-aided diagnosis systems for medical image segmentation has become a prominent focus within computer vision. With the development of deep learning, diffusion models have exhibited remarkable performance in medical image segmentation tasks. However, traditional segmentation diffusion models typically employ random Gaussian noise to generate segmentation masks, resulting in non-unique segmentation masks that fail to guarantee the reproducibility of segmentation results. To tackle this issue, this paper introduces a novel method, Cold SegDiffusion, for general medical image segmentation based on the diffusion model. In this method, medical image segmentation is conceptualized as a denoising problem. The segmentation masks covering medical images serve as input for the segmentation encoder, addressing the challenge of generating non-unique masks due to noise randomness. Additionally, the contrast enhancement module is designed to translate features into the frequency domain, addressing the issues of low contrast and boundary disappearance in medical images. Furthermore, the suggested conditional cross-attention module utilizes the conditional encoder and cross-attention weights to enhance important features of the segmentation encoder output, improving the network’s capacity to focus on target regions. The proposed method is validated across three medical image segmentation datasets with different modalities. Experimental results demonstrate that Cold SegDiffusion outperforms mainstream segmentation methods. The code is available at https://github.com/TimesXY/Cold-SegDiffusion.

3D MedDiffusion: A 3D Medical Diffusion Model for Controllable and High-quality Medical Image Generation

MedDiff-FM: A Diffusion-based Foundation Model for Versatile Medical Image Applications

Denoising diffusion probabilistic models for 3D medical image generation

Diffusion Models for Medical Image Computing: A Survey

Med-cDiff: Conditional Medical Image Generation with Diffusion Models

GH-DDM: the generalized hybrid denoising diffusion model for medical image generation

A New Chapter for Medical Image Generation: The Stable Diffusion Method

Diffusion Models for Memory-efficient Processing of 3D Medical Images

Generative Enhancement for 3D Medical Images

Enabling Competitive Performance of Medical Imaging with Diffusion Model-generated Images without Privacy Leakage

Memory-Efficient 3D Denoising Diffusion Models for Medical Image Processing

DiffusionBlend: Learning 3D Image Prior through Position-aware Diffusion Score Blending for 3D Computed Tomography Reconstruction

MediSyn: Text-Guided Diffusion Models for Broad Medical 2D and 3D Image Synthesis

Conditional Diffusion Models for Semantic 3D Brain MRI Synthesis

Anatomically-Controllable Medical Image Generation with Segmentation-Guided Diffusion Models

Improving 3D Imaging with Pre-Trained Perpendicular 2D Diffusion Models

WDM: 3D Wavelet Diffusion Models for High-Resolution Medical Image Synthesis

RenderDiffusion: Image Diffusion for 3D Reconstruction, Inpainting and Generation

2.5D Multi-view Averaging Diffusion Model for 3D Medical Image Translation: Application to Low-count PET Reconstruction with CT-less Attenuation Correction

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation

Cold SegDiffusion: A Novel Diffusion Model for Medical Image Segmentation