Abstract:In recent years, the denoising diffusion model has achieved remarkable success in image segmentation modeling. With its powerful nonlinear modeling capabilities and superior generalization performance, denoising diffusion models have gradually been applied to medical image segmentation tasks, bringing new perspectives and methods to this field. However, existing methods overlook the uncertainty of segmentation boundaries and the fuzziness of regions, resulting in the instability and inaccuracy of the segmentation results. To solve this problem, a denoising diffusion fusion network based on fuzzy learning for 3D medical image segmentation (FDiff-Fusion) is proposed in this paper. By integrating the denoising diffusion model into the classical U-Net network, this model can effectively extract rich semantic information from input medical images, thus providing excellent pixel-level representation for medical image segmentation. In this paper, a fuzzy learning module is designed on the skip path of U-Net network because of the widespread boundary uncertainty and region blurring of medical image segmentation. The module sets several fuzzy membership functions for the input encoded features to describe the similarity degree between the feature points, and applies fuzzy rules to the fuzzy membership functions, thus enhancing the modeling ability of the model for uncertain boundaries and fuzzy regions. In addition, in order to improve the accuracy and robustness of the model segmentation results, we introduced an iterative attention feature fusion method in the test phase, which added local context information to the global context information in the attention module to fuse the prediction results of each denoising time step. Finally, to validate the effectiveness of FDiff-Fusion, we compare it with existing advanced segmentation networks on the BRATS 2020 brain tumor dataset and the BTCV abdominal multi-organ dataset. The results show that FDiff-Fusion significantly improves the Dice scores and HD95 distance on these two datasets, demonstrating its superiority in medical image segmentation tasks.

Introducing Shape Prior Module in Diffusion Model for Medical Image Segmentation

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation

Verdiff-Net: A Conditional Diffusion Framework for Spinal Medical Image Segmentation

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

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

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

Cold SegDiffusion: A Novel Diffusion Model for Medical Image Segmentation

BerDiff: Conditional Bernoulli Diffusion Model for Medical Image Segmentation

MedSegDiff: Medical Image Segmentation with Diffusion Probabilistic Model

Denoising Diffusions in Latent Space for Medical Image Segmentation

FDiff-Fusion: Denoising Diffusion Fusion Network Based on Fuzzy Learning for 3D Medical Image Segmentation

Denoising Diffusion Semantic Segmentation with Mask Prior Modeling

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

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

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

Analysing Diffusion Segmentation for Medical Images

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

DiffBoost: Enhancing Medical Image Segmentation via Text-Guided Diffusion Model

Diffuse-UDA: Addressing Unsupervised Domain Adaptation in Medical Image Segmentation with Appearance and Structure Aligned Diffusion Models

DTAN: Diffusion-based Text Attention Network for medical image segmentation

Accelerating Diffusion Models Via Pre-segmentation Diffusion Sampling for Medical Image Segmentation