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 条
[41]   Meta semi-supervised medical image segmentation with label hierarchy [J].
Xu, Hai ;
Xie, Hongtao ;
Tan, Qingfeng ;
Zhang, Yongdong .
HEALTH INFORMATION SCIENCE AND SYSTEMS, 2023, 11 (01)
[42]   Consistency and adversarial semi-supervised learning for medical image segmentation [J].
Tang, Yongqiang ;
Wang, Shilei ;
Qu, Yuxun ;
Cui, Zhihua ;
Zhang, Wensheng .
COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 161
[43]   Mutual consistency learning for semi-supervised medical image segmentation [J].
Wu, Yicheng ;
Ge, Zongyuan ;
Zhang, Donghao ;
Xu, Minfeng ;
Zhang, Lei ;
Xia, Yong ;
Cai, Jianfei .
MEDICAL IMAGE ANALYSIS, 2022, 81
[44]   Semantic knowledge transfer for semi-supervised medical image segmentation [J].
Zhou, Shiwei ;
Zhao, Haifeng ;
Ma, Leilei ;
Sun, Dengdi .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2025, 158
[45]   CROSS-IMAGE DISTILLATION FOR SEMI-SUPERVISED SEMANTIC SEGMENTATION [J].
Zhang, Nan ;
Xiao, Fan ;
Hou, Junlin ;
Zhao, Ruiwei ;
Zhang, Xiaobo ;
Feng, Rui .
2024 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING, ICASSP 2024, 2024, :6745-6749
[46]   Semi-supervised hierarchical Transformer for hyperspectral Image classification [J].
He, Ziping ;
Zhu, Qianglin ;
Xia, Kewen ;
Ghamisi, Pedram ;
Zu, Baokai .
INTERNATIONAL JOURNAL OF REMOTE SENSING, 2024, 45 (01) :21-50
[47]   Quality-driven deep cross-supervised learning network for semi-supervised medical image segmentation [J].
Zhang Z. ;
Zhou H. ;
Shi X. ;
Ran R. ;
Tian C. ;
Zhou F. .
Computers in Biology and Medicine, 2024, 176
[48]   CROSS-LEVEL CONTRASTIVE LEARNING AND CONSISTENCY CONSTRAINT FOR SEMI-SUPERVISED MEDICAL IMAGE SEGMENTATION [J].
Zhao, Xinkai ;
Fang, Chaowei ;
Fan, De-Jun ;
Lin, Xutao ;
Gao, Feng ;
Li, Guanbin .
2022 IEEE INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (IEEE ISBI 2022), 2022,
[49]   Reliable semi-supervised mutual learning framework for medical image segmentation [J].
Hang, Wenlong ;
Bai, Kui ;
Liang, Shuang ;
Zhang, Qingfeng ;
Wu, Qiang ;
Jin, Yukun ;
Wang, Qiong ;
Qin, Jing .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 99
[50]   Overlay Mantle-Free for Semi-supervised Medical Image Segmentation [J].
Liu, Jiacheng ;
Qian, Wenhua ;
Cao, Jinde ;
Liu, Peng .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT X, 2024, 15010 :589-598