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
来源
INTERNATIONAL CONFERENCE ON MEDICAL IMAGING WITH DEEP LEARNING, VOL 172 | 2022年 / 172卷
关键词
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]   Semi-supervised Medical Image Segmentation with Confidence Calibration [J].
Xu, Qisen ;
Wu, Qian ;
Hu, Yiqiu ;
Jin, Bo ;
Hu, Bin ;
Zhu, Fengping ;
Li, Yuxin ;
Wang, Xiangfeng .
2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
[22]   Semi-supervised Medical Image Segmentation via Learning Consistency Under Transformations [J].
Bortsova, Gerda ;
Dubost, Florian ;
Hogeweg, Laurens ;
Katramados, Ioannis ;
de Bruijne, Marleen .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2019, PT VI, 2019, 11769 :810-818
[23]   Semi-supervised medical image segmentation via uncertainty rectified pyramid consistency [J].
Luo, Xiangde ;
Wang, Guotai ;
Liao, Wenjun ;
Chen, Jieneng ;
Song, Tao ;
Chen, Yinan ;
Zhang, Shichuan ;
Metaxas, Dimitris N. ;
Zhang, Shaoting .
MEDICAL IMAGE ANALYSIS, 2022, 80
[24]   Semi-supervised Pathological Image Segmentation via Cross Distillation of Multiple Attentions [J].
Zhong, Lanfeng ;
Liao, Xin ;
Zhang, Shaoting ;
Wang, Guotai .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT VI, 2023, 14225 :570-579
[25]   Diversity matters: Cross-head mutual mean-teaching for semi-supervised medical image segmentation [J].
Li, Wei ;
Bian, Ruifeng ;
Zhao, Wenyi ;
Xu, Weijin ;
Yang, Huihua .
MEDICAL IMAGE ANALYSIS, 2024, 97
[26]   Semi-supervised Image Segmentation [J].
Lazarova, Gergana Angelova .
ARTIFICIAL INTELLIGENCE: METHODOLOGY, SYSTEMS, AND APPLICATIONS, 2014, 8722 :59-68
[27]   Semi-CLMT: A Semi-Supervised Framework for Medical Image Segmentation [J].
Kong, Xiangyu ;
Ren, Zeyu ;
Zhu, Hengde ;
Wang, Shuihua ;
Liu, Lu .
IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTATIONAL INTELLIGENCE, 2025,
[28]   UT-MT: A Semi-Supervised Model of Fusion Transformer for 3D Medical Image Segmentation [J].
Liu, Xianchang ;
Liu, Peishun ;
Wang, Jinyu ;
Wang, Qinshuo ;
Guo, Qing ;
Tang, Ruichun .
2023 8TH INTERNATIONAL CONFERENCE ON CLOUD COMPUTING AND BIG DATA ANALYTICS, ICCCBDA, 2023, :190-196
[29]   Semi-supervised medical image segmentation via hard positives oriented contrastive learning [J].
Tang, Cheng ;
Zeng, Xinyi ;
Zhou, Luping ;
Zhou, Qizheng ;
Wang, Peng ;
Wu, Xi ;
Ren, Hongping ;
Zhou, Jiliu ;
Wang, Yan .
PATTERN RECOGNITION, 2024, 146
[30]   Pair Shuffle Consistency for Semi-supervised Medical Image Segmentation [J].
He, Jianjun ;
Cai, Chenyu ;
Li, Qiong ;
Ma, Andy J. .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT VIII, 2024, 15008 :489-499