Region-level Contrastive and Consistency Learning for Semi-Supervised Semantic Segmentation

Region-level Contrastive and Consistency Learning for Semi-Supervised Semantic Segmentation

Jianrong Zhang, Tianyi Wu, Chuanghao Ding, Hongwei Zhao, Guodong Guo

Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence
Main Track. Pages 1622-1628. https://doi.org/10.24963/ijcai.2022/226

Current semi-supervised semantic segmentation methods mainly focus on designing pixel-level consistency and contrastive regularization. However, pixel-level regularization is sensitive to noise from pixels with incorrect predictions, and pixel-level contrastive regularization has a large memory and computational cost. To address the issues, we propose a novel region-level contrastive and consistency learning framework (RC^2L) for semi-supervised semantic segmentation. Specifically, we first propose a Region Mask Contrastive (RMC) loss and a Region Feature Contrastive (RFC) loss to accomplish region-level contrastive property. Furthermore, Region Class Consistency (RCC) loss and Semantic Mask Consistency (SMC) loss are proposed for achieving region-level consistency. Based on the proposed region-level contrastive and consistency regularization, we develop a region-level contrastive and consistency learning framework (RC^2L) for semi-supervised semantic segmentation, and evaluate our RC^2L on two challenging benchmarks (PASCAL VOC 2012 and Cityscapes), outperforming the state-of-the-art.
Keywords:
Computer Vision: Segmentation
Computer Vision: Transfer, low-shot, semi- and un- supervised learning