Semi-Supervised Medical Image Segmentation via Cross Teaching between CNN and Transformer

被引:0
|
作者
Luo, Xiangde [1 ,2 ]
Hu, Minhao [3 ]
Song, Tao [3 ]
Wang, Guotai [1 ,2 ]
Zhang, Shaoting [1 ,2 ,3 ]
机构
[1] Univ Elect Sci & Technol China, Chengdu, Peoples R China
[2] Shanghai AI Lab, Shanghai, Peoples R China
[3] SenseTime, Shanghai, Peoples R China
关键词
Semi-supervised learning; CNN; transformer; cross teaching;
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recently, deep learning with Convolutional Neural Networks (CNNs) and Transformers has shown encouraging results in fully supervised medical image segmentation. However, it is still challenging for them to achieve good performance with limited annotations for training. This work presents a very simple yet efficient framework for semi-supervised medical image segmentation by introducing the cross teaching between CNN and Transformer. Specifically, we simplify the classical deep co-training from consistency regularization to cross teaching, where the prediction of a network is used as the pseudo label to supervise the other network directly end-to-end. Considering the difference in learning paradigm between CNN and Transformer, we introduce the Cross Teaching between CNN and Transformer rather than just using CNNs. Experiments on a public benchmark show that our method outperforms eight existing semi-supervised learning methods just with a more straightforward framework. Notably, this work may be the first attempt to combine CNN and transformer for semi-supervised medical image segmentation and achieve promising results on a public benchmark. Code is available at: https://github.com/HiLab-git/SSL4MIS.
引用
收藏
页码:820 / 833
页数:14
相关论文
共 50 条
  • [31] Deep semi-supervised learning for medical image segmentation: A review
    Han, Kai
    Sheng, Victor S.
    Song, Yuqing
    Liu, Yi
    Qiu, Chengjian
    Ma, Siqi
    Liu, Zhe
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 245
  • [32] Consistency and adversarial semi-supervised learning for medical image segmentation
    Tang, Yongqiang
    Wang, Shilei
    Qu, Yuxun
    Cui, Zhihua
    Zhang, Wensheng
    COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 161
  • [33] Mutual consistency learning for semi-supervised medical image segmentation
    Wu, Yicheng
    Ge, Zongyuan
    Zhang, Donghao
    Xu, Minfeng
    Zhang, Lei
    Xia, Yong
    Cai, Jianfei
    MEDICAL IMAGE ANALYSIS, 2022, 81
  • [34] Mutual consistency learning for semi-supervised medical image segmentation
    Wu, Yicheng
    Ge, Zongyuan
    Zhang, Donghao
    Xu, Minfeng
    Zhang, Lei
    Xia, Yong
    Cai, Jianfei
    Medical Image Analysis, 2022, 81
  • [35] Meta semi-supervised medical image segmentation with label hierarchy
    Xu, Hai
    Xie, Hongtao
    Tan, Qingfeng
    Zhang, Yongdong
    HEALTH INFORMATION SCIENCE AND SYSTEMS, 2023, 11 (01)
  • [36] Semi-supervised contextual cognitive augmentation-based cross-teaching network for multiclass medical image segmentation
    Gai, Di
    Wu, Yuxuan
    Xiao, Yusong
    Geng, Yuhan
    Cao, Lei
    Xiong, Xin
    Zhong, An-qi
    IET Image Processing, 2024, 18 (13) : 3989 - 4004
  • [37] SEMI-SUPERVISED HYPERSPECTRAL IMAGE SEGMENTATION
    Li, Jun
    Bioucas-Dias, Jose M.
    Plaza, Antonio
    2009 FIRST WORKSHOP ON HYPERSPECTRAL IMAGE AND SIGNAL PROCESSING: EVOLUTION IN REMOTE SENSING, 2009, : 215 - +
  • [38] Semi-Supervised Convolutional Vision Transformer with Bi-Level Uncertainty Estimation for Medical Image Segmentation
    Huang, Huimin
    Huang, Yawen
    Xie, Shiao
    Lin, Lanfen
    Tong Ruofeng
    Chen, Yen-Wei
    Li, Yuexiang
    Zheng, Yefeng
    PROCEEDINGS OF THE 31ST ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2023, 2023, : 5214 - 5222
  • [39] Quality-driven deep cross-supervised learning network for semi-supervised medical image segmentation
    Zhang Z.
    Zhou H.
    Shi X.
    Ran R.
    Tian C.
    Zhou F.
    Computers in Biology and Medicine, 2024, 176
  • [40] CROSS-LEVEL CONTRASTIVE LEARNING AND CONSISTENCY CONSTRAINT FOR SEMI-SUPERVISED MEDICAL IMAGE SEGMENTATION
    Zhao, Xinkai
    Fang, Chaowei
    Fan, De-Jun
    Lin, Xutao
    Gao, Feng
    Li, Guanbin
    2022 IEEE INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (IEEE ISBI 2022), 2022,