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

被引:6
|
作者
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] LeViT-UNet: Make Faster Encoders with Transformer for Medical Image Segmentation
    Xu, Guoping
    Zhang, Xuan
    He, Xinwei
    Wu, Xinglong
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT VIII, 2024, 14432 : 42 - 53
  • [32] Hybrid transformer-CNN and LSTM model for lung disease segmentation and classification
    Shafi, Syed Mohammed
    Chinnappan, Sathiya Kumar
    PEERJ COMPUTER SCIENCE, 2024, 10
  • [33] CorFormer: a hybrid transformer-CNN architecture for corrosion segmentation on metallic surfaces
    Subedi, Abhishek
    Qian, Cheng
    Sadeghian, Reza
    Jahanshahi, Mohammad R.
    MACHINE VISION AND APPLICATIONS, 2025, 36 (02)
  • [34] Dual encoder network with transformer-CNN for multi-organ segmentation
    Hong, Zhifang
    Chen, Mingzhi
    Hu, Weijie
    Yan, Shiyu
    Qu, Aiping
    Chen, Lingna
    Chen, Junxi
    MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2023, 61 (03) : 661 - 671
  • [35] Lithium Metal Battery Quality Control via Transformer-CNN Segmentation
    Quenum, Jerome
    Zenyuk, Iryna V. V.
    Ushizima, Daniela
    JOURNAL OF IMAGING, 2023, 9 (06)
  • [36] 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
  • [37] 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):
  • [38] 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
  • [39] 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
  • [40] MixFormer: A Mixed CNN-Transformer Backbone for Medical Image Segmentation
    Liu, Jun
    Li, Kunqi
    Huang, Chun
    Dong, Hua
    Song, Yusheng
    Li, Rihui
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2025, 74