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 条
  • [21] MIXMODULE: MIXED CNN KERNEL MODULE FOR MEDICAL IMAGE SEGMENTATION
    Yu, Henry H.
    Feng, Xue
    Wang, Ziwen
    Sun, Hao
    2020 IEEE 17TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI 2020), 2020, : 1508 - 1512
  • [22] SmokeSeger: A Transformer-CNN coupled model for urban scene smoke segmentation
    Jing, Tao
    Meng, Qing-Hao
    Hou, Hui-Rang
    IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2024, 20 (02) : 1385 - 1396
  • [23] A transformer-CNN parallel network for image guided depth completion
    Li, Tao
    Dong, Xiucheng
    Lin, Jie
    Peng, Yonghong
    PATTERN RECOGNITION, 2024, 150
  • [24] Lithium Metal Battery Quality Control via Transformer-CNN Segmentation
    Quenum, Jerome
    Zenyuk, Iryna V. V.
    Ushizima, Daniela
    JOURNAL OF IMAGING, 2023, 9 (06)
  • [25] 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
  • [26] A novel full-convolution UNet-transformer for medical image segmentation
    Zhu, Tianyou
    Ding, Derui
    Wang, Feng
    Liang, Wei
    Wang, Bo
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 89
  • [27] CorFormer: a hybrid transformer-CNN architecture for corrosion segmentation on metallic surfaces
    Abhishek Subedi
    Cheng Qian
    Reza Sadeghian
    Mohammad R. Jahanshahi
    Machine Vision and Applications, 2025, 36 (2)
  • [28] DSTUNET: UNET WITH EFFICIENT DENSE SWIN TRANSFORMER PATHWAY FOR MEDICAL IMAGE SEGMENTATION
    Cai, Zhuotong
    Xin, Jingmin
    Shi, Peiwen
    Wu, Jiayi
    Zheng, Nanning
    2022 IEEE INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (IEEE ISBI 2022), 2022,
  • [29] RM-UNet: UNet-like Mamba with rotational SSM module for medical image segmentation
    Tang, Hao
    Huang, Guoheng
    Cheng, Lianglun
    Yuan, Xiaochen
    Tao, Qi
    Chen, Xuhang
    Zhong, Guo
    Yang, Xiaohui
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, : 8427 - 8443
  • [30] 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