Re-distributing Biased Pseudo Labels for Semi-supervised Semantic Segmentation: A Baseline Investigation

被引:91
作者
He, Ruifei [1 ,2 ]
Yang, Jihan [1 ]
Qi, Xiaojuan [1 ]
机构
[1] Univ Hong Kong, Hong Kong, Peoples R China
[2] Zhejiang Univ, Hangzhou, Peoples R China
来源
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021) | 2021年
关键词
D O I
10.1109/ICCV48922.2021.00685
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
While self-training has advanced semi-supervised semantic segmentation, it severely suffers from the long-tailed class distribution on real-world semantic segmentation datasets that make the pseudo-labeled data bias toward majority classes. In this paper, we present a simple and yet effective Distribution Alignment and Random Sampling (DARS) method to produce unbiased pseudo labels that match the true class distribution estimated from the labeled data. Besides, we also contribute a progressive data augmentation and labeling strategy to facilitate model training with pseudo-labeled data. Experiments on both Cityscapes and PASCAL VOC 2012 datasets demonstrate the effectiveness of our approach. Albeit simple, our method performs favorably in comparison with state-of-the-art approaches.
引用
收藏
页码:6910 / 6920
页数:11
相关论文
共 61 条
[1]  
[Anonymous], Vision and Pattern Recognition (CVPR)
[2]  
[Anonymous], 2017, PROC CVPR IEEE
[3]  
[Anonymous], 2015, ARXIV150604579
[4]  
[Anonymous], 2010, International journal of computer vision, DOI DOI 10.1007/s11263-009-0275-4
[5]  
[Anonymous], 2014, Overfeat: integrated recognition, localization and detection using convolutional networks
[6]  
[Anonymous], 2019, P EUR C COMP VIS ECC, DOI DOI 10.1007/S13143-018-0064-5
[7]   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
[8]  
Bengio Y., 2005, Advances in Neural Information Processing Systems (NeurIPS)
[9]  
Berthelot D., 2019, P INT C LEARN REPR
[10]  
Berthelot D, 2019, ADV NEUR IN, V32