DBEF-Net: Diffusion-Based Boundary-Enhanced Fusion Network for medical image segmentation

被引:0
作者
Huang, Zhenyang [1 ]
Li, Jianjun [2 ]
Mao, Ning [3 ]
Yuan, Genji [1 ]
Li, Jinjiang [1 ]
机构
[1] Shandong Technol & Business Univ, Sch Comp Sci & Technol, Yantai, Peoples R China
[2] Capital Med Univ, Beijing Youan Hosp, Beijing, Peoples R China
[3] Qingdao Univ, Yantai Yuhuangding Hosp, Dept Radiol, Yantai, Peoples R China
关键词
Medical image segmentation; Diffusion model; Boundary-attentive; Group attention fusion; RETHINKING; REGION;
D O I
10.1016/j.eswa.2024.124467
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Medical image segmentation aims to locate lesions within a given image to assist doctors in diagnosis and treatment, playing a crucial role in improving patient outcomes. Recently, the Denoising Diffusion Probabilistic Model (DDPM) has demonstrated its advantages in various generative tasks. Building upon these advantages, we propose DBEF-Net: Diffusion -Based Boundary -Enhanced Fusion Network for Medical Image Segmentation. When applying diffusion models to image segmentation, the challenge of inconsistent semantic features and noise embedding needs to be addressed. To overcome this issue, we introduce the Group Attention Fusion Module (GAFM), which merges image features and noise features in groups to better utilize semantic information and noise characteristics in the diffusion model. Additionally, we design the Boundary -Attentive Fusion Module (BAFM) to incorporate boundary priors into the diffusion model to enhance sensitivity to fuzzy boundaries. We also introduce the Spatial Context Fusion Module (SCFM) to fully exploit multi -scale information in the encoder and decoder. We conduct experiments on four datasets with DBEF-Net, comparing it with other methods. Encouragingly, our approach shows significant performance improvement while maintaining novelty.
引用
收藏
页数:13
相关论文
共 70 条
  • [1] Azad R., 2022, arXiv, DOI DOI 10.48550/ARXIV.2211.14830
  • [2] Berseth M, 2017, Arxiv, DOI arXiv:1703.00523
  • [3] DermoSegDiff: A Boundary-Aware Segmentation Diffusion Model for Skin Lesion Delineation
    Bozorgpour, Afshin
    Sadegheih, Yousef
    Kazerouni, Amirhossein
    Azad, Reza
    Merhof, Dorit
    [J]. PREDICTIVE INTELLIGENCE IN MEDICINE, PRIME 2023, 2023, 14277 : 146 - 158
  • [4] Asymmetric U-shaped network with hybrid attention mechanism for kidney ultrasound images segmentation
    Chen, Gong -Ping
    Zhao, Yu
    Dai, Yu
    Zhang, Jian-Xun
    Yin, Xiao-Tao
    Cui, Liang
    Qian, Jiang
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2023, 212
  • [5] DCAN: Deep Contour-Aware Networks for Accurate Gland Segmentation
    Chen, Hao
    Qi, Xiaojuan
    Yu, Lequan
    Heng, Pheng-Ann
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 2487 - 2496
  • [6] BerDiff: Conditional Bernoulli Diffusion Model for Medical Image Segmentation
    Chen, Tao
    Wang, Chenhui
    Shan, Hongming
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT IV, 2023, 14223 : 491 - 501
  • [7] Visformer: The Vision-friendly Transformer
    Chen, Zhengsu
    Xie, Lingxi
    Niu, Jianwei
    Liu, Xuefeng
    Wei, Longhui
    Tian, Qi
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 569 - 578
  • [8] Diffusion Transformer U-Net for Medical Image Segmentation
    Chowdary, G. Jignesh
    Yin, Zhaozheng
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT IV, 2023, 14223 : 622 - 631
  • [9] Conoci S, 2017, 2017 EUROPEAN CONFERENCE ON CIRCUIT THEORY AND DESIGN (ECCTD)
  • [10] Ms RED: A novel multi-scale residual encoding and decoding network for skin lesion segmentation
    Dai, Duwei
    Dong, Caixia
    Xu, Songhua
    Yan, Qingsen
    Li, Zongfang
    Zhang, Chunyan
    Luo, Nana
    [J]. MEDICAL IMAGE ANALYSIS, 2022, 75