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 条
  • [31] Dual encoder network with transformer-CNN for multi-organ segmentation
    Zhifang Hong
    Mingzhi Chen
    Weijie Hu
    Shiyu Yan
    Aiping Qu
    Lingna Chen
    Junxi Chen
    Medical & Biological Engineering & Computing, 2023, 61 : 661 - 671
  • [32] CoTrFuse: a novel framework by fusing CNN and transformer for medical image segmentation
    Chen, Yuanbin
    Wang, Tao
    Tang, Hui
    Zhao, Longxuan
    Zhang, Xinlin
    Tan, Tao
    Gao, Qinquan
    Du, Min
    Tong, Tong
    PHYSICS IN MEDICINE AND BIOLOGY, 2023, 68 (17):
  • [33] TFCNs: A CNN-Transformer Hybrid Network for Medical Image Segmentation
    Li, Zihan
    Li, Dihan
    Xu, Cangbai
    Wang, Weice
    Hong, Qingqi
    Li, Qingde
    Tian, Jie
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2022, PT IV, 2022, 13532 : 781 - 792
  • [34] ConvWin-UNet: UNet-like hierarchical vision Transformer combined with convolution for medical image segmentation
    Feng, Xiaomeng
    Wang, Taiping
    Yang, Xiaohang
    Zhang, Minfei
    Guo, Wanpeng
    Wang, Weina
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (01) : 128 - 144
  • [35] SwinE-UNet3+: swin transformer encoder network for medical image segmentation
    Zou, Ping
    Wu, Jian-Sheng
    PROGRESS IN ARTIFICIAL INTELLIGENCE, 2023, 12 (01) : 99 - 105
  • [36] SwinE-UNet3+: swin transformer encoder network for medical image segmentation
    Ping Zou
    Jian-Sheng Wu
    Progress in Artificial Intelligence, 2023, 12 : 99 - 105
  • [37] CTA-UNet: CNN-transformer architecture UNet for dental CBCT images segmentation
    Chen, Zeyu
    Chen, Senyang
    Hu, Fengjun
    PHYSICS IN MEDICINE AND BIOLOGY, 2023, 68 (17):
  • [38] Dual branch Transformer-CNN parametric filtering network for underwater image enhancement
    Chang, Baocai
    Li, Jinjiang
    Ren, Lu
    Chen, Zheng
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2024, 100
  • [39] A Novel Elastomeric UNet for Medical Image Segmentation
    Cai, Sijing
    Wu, Yi
    Chen, Guannan
    FRONTIERS IN AGING NEUROSCIENCE, 2022, 14
  • [40] Improved UNet with Attention for Medical Image Segmentation
    AL Qurri, Ahmed
    Almekkawy, Mohamed
    SENSORS, 2023, 23 (20)