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.

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation

LambdaUNet: 2.5D Stroke Lesion Segmentation of Diffusion-weighted MR Images

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

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

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

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

Introducing Shape Prior Module in Diffusion Model for Medical Image Segmentation

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

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

Denoising Diffusion Medical Models

MD-UNet: a medical image segmentation network based on mixed depthwise convolution

UNet 3+: A Full-Scale Connected UNet for Medical Image Segmentation

FD-DUNet: Frequency Domain Global Modeling Enhances Receptive Field Expansion UNet for Efficient Medical Image Segmentation

Re-DiffiNet: Modeling discrepancies in tumor segmentation using diffusion models

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

Cold SegDiffusion: A Novel Diffusion Model for Medical Image Segmentation

HiDiff: Hybrid Diffusion Framework for Medical Image Segmentation

Head and Neck Tumor Segmentation from [18F]F-FDG PET/CT Images Based on 3D Diffusion Model

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

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