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 条
  • [21] DNetUnet: a semi-supervised CNN of medical image segmentation for super-computing AI service
    Tseng, Kuo-Kun
    Zhang, Ran
    Chen, Chien-Ming
    Hassan, Mohammad Mehedi
    JOURNAL OF SUPERCOMPUTING, 2021, 77 (04): : 3594 - 3615
  • [22] Diversity matters: Cross-head mutual mean-teaching for semi-supervised medical image segmentation
    Li, Wei
    Bian, Ruifeng
    Zhao, Wenyi
    Xu, Weijin
    Yang, Huihua
    MEDICAL IMAGE ANALYSIS, 2024, 97
  • [23] Semi-supervised Image Segmentation
    Lazarova, Gergana Angelova
    ARTIFICIAL INTELLIGENCE: METHODOLOGY, SYSTEMS, AND APPLICATIONS, 2014, 8722 : 59 - 68
  • [24] DNetUnet: a semi-supervised CNN of medical image segmentation for super-computing AI service
    Kuo-Kun Tseng
    Ran Zhang
    Chien-Ming Chen
    Mohammad Mehedi Hassan
    The Journal of Supercomputing, 2021, 77 : 3594 - 3615
  • [25] Cross-Adversarial Local Distribution Regularization for Semi-supervised Medical Image Segmentation
    Thanh Nguyen-Duc
    Trung Le
    Bammer, Roland
    Zhao, He
    Cai, Jianfei
    Dinh Phung
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT I, 2023, 14220 : 183 - 194
  • [26] Semi-supervised Medical Image Segmentation with Multiscale Contrastive Learning and Cross-Supervision
    Wu, Wenxia
    Yan, Jing
    Liang, Dong
    Zhang, Zhenyu
    Li, Zhi-Cheng
    2023 45TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE & BIOLOGY SOCIETY, EMBC, 2023,
  • [27] Cross Prompting Consistency with Segment Anything Model for Semi-supervised Medical Image Segmentation
    Miao, Juzheng
    Chen, Cheng
    Zhang, Keli
    Chuai, Jie
    Li, Quanzheng
    Heng, Pheng-Ann
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT XI, 2024, 15011 : 167 - 177
  • [28] Semi-supervised medical image segmentation via hard positives oriented contrastive learning
    Tang, Cheng
    Zeng, Xinyi
    Zhou, Luping
    Zhou, Qizheng
    Wang, Peng
    Wu, Xi
    Ren, Hongping
    Zhou, Jiliu
    Wang, Yan
    PATTERN RECOGNITION, 2024, 146
  • [29] Adaptive Bidirectional Displacement for Semi-Supervised Medical Image Segmentation
    Chi, Hanyang
    Pang, Jian
    Zhang, Bingfeng
    Liu, Weifeng
    2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2024, 2024, : 4070 - 4080
  • [30] Pair Shuffle Consistency for Semi-supervised Medical Image Segmentation
    He, Jianjun
    Cai, Chenyu
    Li, Qiong
    Ma, Andy J.
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT VIII, 2024, 15008 : 489 - 499