Semi-supervised semantic segmentation with cross teacher training

被引:22
作者
Xiao, Hui [1 ]
Li, Dong [1 ]
Xu, Hao [1 ]
Fu, Shuibo [2 ]
Yan, Diqun [1 ]
Song, Kangkang [3 ]
Peng, Chengbin [1 ]
机构
[1] Ningbo Univ, Fac Elect Engn & Comp Sci, Ningbo 315211, Peoples R China
[2] Jiaochuan Acad, Ningbo 315299, Peoples R China
[3] Chinese Acad Sci, Ningbo Inst Mat Technol & Engn, Inst Adv Mfg Technol, Ningbo 315201, Peoples R China
基金
中国国家自然科学基金;
关键词
Semantic segmentation; Semi-supervised learning; Student-teacher networks;
D O I
10.1016/j.neucom.2022.08.052
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Convolutional neural networks can achieve remarkable performance in semantic segmentation tasks. However, such neural network approaches heavily rely on costly pixel-level annotation. Semi-supervised learning is a promising resolution to tackle this issue, but its performance still far falls behind the fully supervised counterpart. This work proposes a cross-teacher training framework with three modules that significantly improves traditional semi-supervised learning approaches. The core is a cross-teacher module, which could simultaneously reduce the coupling among peer networks and the error accumulation between teacher and student networks. In addition, we propose two complementary contrastive learning modules. The high-level module can transfer high-quality knowledge from labeled data to unlabeled ones and promote separation between classes in feature space. The low-level module can encourage low-quality features learning from the high-quality features among peer networks. In experiments, the cross-teacher module significantly improves the performance of traditional student-teacher approaches, and our framework outperforms state-of-the-art methods on benchmark datasets. (C) 2022 Elsevier B.V. All rights reserved.
引用
收藏
页码:36 / 46
页数:11
相关论文
共 58 条
[1]  
Alonso Inigo, 2021, ICCV, P8219
[2]  
[Anonymous], 2020, IEEE IJCNN
[3]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[4]  
Bae S, 2022, Arxiv, DOI arXiv:2106.15499
[5]  
Chen J, 2021, arXiv
[6]  
Chen LC, 2016, Arxiv, DOI arXiv:1412.7062
[7]  
Chen LC, 2017, Arxiv, DOI [arXiv:1706.05587, DOI 10.48550/ARXIV.1706.05587]
[8]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851
[9]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[10]   Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision [J].
Chen, Xiaokang ;
Yuan, Yuhui ;
Zeng, Gang ;
Wang, Jingdong .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :2613-2622