TCI-UNet: transformer-CNN interactive module for medical image segmentation

被引:4
|
作者
Bian, Xuan [1 ]
Wang, Guanglei [1 ]
Li, Yan [1 ]
Wang, Hongrui [1 ]
机构
[1] Hebei Univ, Coll Elect & Informat Engn, Baoding 071002, Hebei, Peoples R China
基金
中国国家自然科学基金;
关键词
NETWORK;
D O I
10.1364/BOE.499640
中图分类号
Q5 [生物化学];
学科分类号
071010 ; 081704 ;
摘要
Medical image segmentation is a crucial step in developing medical systems, especially for assisting doctors in diagnosing and treating diseases. Currently, UNet has become the preferred network for most medical image segmentation tasks and has achieved tremendous success. However, due to the limitations of convolutional operation mechanisms, its ability to model long-range dependencies between features is limited. With the success of transformers in the computer vision (CV) field, many excellent models that combine transformers with UNet have emerged, but most of them have fixed receptive fields and a single feature extraction method. To address this issue, we propose a transformer-CNN interactive (TCI) feature extraction module and use it to construct TCI-UNet. Specifically, we improve the self-attention mechanism in transformers to enhance the guiding ability of attention maps for computational resource allocation. It can strengthen the network's ability to capture global contextual information from feature maps. Additionally, we introduce local multi-scale information to supplement feature information, allowing the network to focus on important local information while modeling global contextual information. This improves the network's capability to extract feature map information and facilitates effective interaction between global and local information within the transformer, enhancing the representational power of transformers. We conducted a large number of experiments on the LiTS-2017 and ISIC-2018 datasets to verify the effectiveness of our proposed method, with DCIE values of 93.81% and 88.22%, respectively. Through ablation experiments, we proved the effectiveness of the TCI module, and in comparison with other state-of-the-art (SOTA) networks, we demonstrated the superiority of TCI-UNet in accuracy and generalization.
引用
收藏
页码:5904 / 5920
页数:17
相关论文
共 50 条
  • [41] TGDAUNet: Transformer and GCNN based dual-branch attention UNet for medical image segmentation
    Song, Pengfei
    Li, Jinjiang
    Fan, Hui
    Fan, Linwei
    COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 167
  • [42] DTCNet: Transformer-CNN Distillation for Super-Resolution of Remote Sensing Image
    Lin, Cong
    Mao, Xin
    Qiu, Chenghao
    Zou, Lilan
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 11117 - 11133
  • [43] TransCNN-HAE: Transformer-CNN Hybrid AutoEncoder for Blind Image Inpainting
    Zhao, Haoru
    Gu, Zhaorui
    Zheng, Bing
    Zheng, Haiyong
    PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2022, 2022, : 6813 - 6821
  • [44] TCPCNet: a transformer-CNN parallel cooperative network for low-light image enhancement
    Wanjun Zhang
    Yujie Ding
    Miaohui Zhang
    Yonghua Zhang
    Lvchen Cao
    Ziqing Huang
    Jun Wang
    Multimedia Tools and Applications, 2024, 83 : 52957 - 52972
  • [45] Multiresolution Aggregation Transformer UNet Based on Multiscale Input and Coordinate Attention for Medical Image Segmentation
    Chen, Shaolong
    Qiu, Changzhen
    Yang, Weiping
    Zhang, Zhiyong
    SENSORS, 2022, 22 (10)
  • [46] RT-Unet: An advanced network based on residual network and transformer for medical image segmentation
    Li, Bo
    Liu, Sikai
    Wu, Fei
    Li, GuangHui
    Zhong, Meiling
    Guan, Xiaohui
    INTERNATIONAL JOURNAL OF INTELLIGENT SYSTEMS, 2022, 37 (11) : 8565 - 8582
  • [47] LightingFormer: Transformer-CNN hybrid network for low-light image enhancement
    Bi, Cong
    Qian, Wenhua
    Cao, Jinde
    Wang, Xue
    COMPUTERS & GRAPHICS-UK, 2024, 124
  • [48] MS-TCNet: An effective Transformer-CNN combined network using multi-scale feature learning for 3D medical image segmentation
    Ao, Yu
    Shi, Weili
    Ji, Bai
    Miao, Yu
    He, Wei
    Jiang, Zhengang
    COMPUTERS IN BIOLOGY AND MEDICINE, 2024, 170
  • [49] CvT-UNet: A weld pool segmentation method integrating a CNN and a transformer
    Yang, Longcheng
    Wang, Huajun
    Meng, Wenjie
    Pan, Hongyu
    HELIYON, 2024, 10 (15)
  • [50] H2MaT-Unet:Hierarchical hybrid multi-axis transformer based Unet for medical image segmentation
    Ju Z.
    Zhou Z.
    Qi Z.
    Yi C.
    Computers in Biology and Medicine, 2024, 174