DFA-UNet: dual-stream feature-fusion attention U-Net for lymph node segmentation in lung cancer diagnosis

被引:0
|
作者
Zhou, Qi [1 ,2 ]
Zhou, Yingwen [2 ]
Hou, Nailong [2 ]
Zhang, Yaxuan [2 ]
Zhu, Guanyu [2 ]
Li, Liang [1 ]
机构
[1] Xuzhou Med Univ, Affiliated Hosp, Dept Radiotherapy, Xuzhou, Peoples R China
[2] Xuzhou Med Univ, Sch Med Imaging, Xuzhou, Peoples R China
关键词
ultrasound elastography; mediastinal lymph nodes; semantic segmentation; attention mechanism; deep learning; ULTRASOUND ELASTOGRAPHY; IMAGE SEGMENTATION;
D O I
10.3389/fnins.2024.1448294
中图分类号
Q189 [神经科学];
学科分类号
071006 ;
摘要
In bronchial ultrasound elastography, accurately segmenting mediastinal lymph nodes is of great significance for diagnosing whether lung cancer has metastasized. However, due to the ill-defined margin of ultrasound images and the complexity of lymph node structure, accurate segmentation of fine contours is still challenging. Therefore, we propose a dual-stream feature-fusion attention U-Net (DFA-UNet). Firstly, a dual-stream encoder (DSE) is designed by combining ConvNext with a lightweight vision transformer (ViT) to extract the local information and global information of images; Secondly, we propose a hybrid attention module (HAM) at the bottleneck, which incorporates spatial and channel attention to optimize the features transmission process by optimizing high-dimensional features at the bottom of the network. Finally, the feature-enhanced residual decoder (FRD) is developed to improve the fusion of features obtained from the encoder and decoder, ensuring a more comprehensive integration. Extensive experiments on the ultrasound elasticity image dataset show the superiority of our DFA-UNet over 9 state-of-the-art image segmentation models. Additionally, visual analysis, ablation studies, and generalization assessments highlight the significant enhancement effects of DFA-UNet. Comprehensive experiments confirm the excellent segmentation effectiveness of the DFA-UNet combined attention mechanism for ultrasound images, underscoring its important significance for future research on medical images.
引用
收藏
页数:9
相关论文
共 31 条
  • [31] FCSU-Net: A novel full-scale Cross-dimension Self-attention U-Net with collaborative fusion of multi-scale feature for medical image segmentation
    Xu, Shijie
    Chen, Yufeng
    Yang, Shukai
    Zhang, Xiaoqian
    Sun, Feng
    Computers in Biology and Medicine, 2024, 180