Skip to main content

ORIGINAL RESEARCH article

Front. Neurosci.
Sec. Visual Neuroscience
Volume 18 - 2024 | doi: 10.3389/fnins.2024.1448294
This article is part of the Research Topic Advances in Computer Vision: From Deep Learning Models to Practical Applications View all 6 articles

DFA-UNet: Dual-stream Feature-fusion Attention U-Net for Lymph Node Segmentation in Lung Cancer Diagnosis

Provisionally accepted
Qi Zhou Qi Zhou 1Yingwen Zhou Yingwen Zhou 1*Nailong Hou Nailong Hou 1*Yaxuan Zhang Yaxuan Zhang 1*Guanyu Zhu Guanyu Zhu 1*liang Li liang Li 2*
  • 1 School of Medical Imaging, Xuzhou Medical University, Xuzhou, China
  • 2 Department of Radiotherapy, The Affiliated Hospital of Xuzhou Medical University, Xuzhou, China

The final, formatted version of the article will be published soon.

    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.

    Keywords: ultrasound elastography, Mediastinal lymph nodes, Semantic segmentation, attention mechanism, deep learning

    Received: 13 Jun 2024; Accepted: 03 Jul 2024.

    Copyright: © 2024 Zhou, Zhou, Hou, Zhang, Zhu and Li. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence:
    Yingwen Zhou, School of Medical Imaging, Xuzhou Medical University, Xuzhou, China
    Nailong Hou, School of Medical Imaging, Xuzhou Medical University, Xuzhou, China
    Yaxuan Zhang, School of Medical Imaging, Xuzhou Medical University, Xuzhou, China
    Guanyu Zhu, School of Medical Imaging, Xuzhou Medical University, Xuzhou, China
    liang Li, Department of Radiotherapy, The Affiliated Hospital of Xuzhou Medical University, Xuzhou, China

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.