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.

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

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

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

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

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation

Cold SegDiffusion: A Novel Diffusion Model for Medical Image Segmentation

DIM-UNet: Boosting medical image segmentation via diffusion models and information bottleneck theory mixed with MLP

Denoising Diffusions in Latent Space for Medical Image Segmentation

MedSegDiff: Medical Image Segmentation with Diffusion Probabilistic Model

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

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

DFormer: Diffusion-guided Transformer for Universal Image Segmentation

Denoising Diffusion Medical Models

High-Precision Dichotomous Image Segmentation via Probing Diffusion Capacity

Analysing Diffusion Segmentation for Medical Images

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

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

MFLUnet: multi-scale fusion lightweight Unet for medical image segmentation

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

Introducing Shape Prior Module in Diffusion Model for Medical Image Segmentation

HiDiff: Hybrid Diffusion Framework for Medical Image Segmentation