Out-of-Candidate Rectification for Weakly Supervised Semantic Segmentation

Zesen Cheng,Pengchong Qiao,Kehan Li,Siheng Li,Pengxu Wei,Xiangyang Ji,Li Yuan,Chang Liu,Jie Chen
DOI: https://doi.org/10.1109/cvpr52729.2023.02267
2022-01-01
Computer Vision and Pattern Recognition
Abstract:Weakly supervised semantic segmentation is typically inspired by class activation maps, which serve as pseudo masks with class-discriminative regions highlighted. Although tremendous efforts have been made to recall precise and complete locations for each class, existing methods still commonly suffer from the unsolicited Out-of-Candidate (OC) error predictions that do not belong to the label candidates, which could be avoidable since the contradiction with image-level class tags is easy to be detected. In this paper, we develop a group ranking-based Out-of-f;Candidate Rectification (OCR) mechanism in a plug-and-play fashion. Firstly, we adaptively split the semantic categories into In-Candidate (IC) and OC groups for each OC pixel according to their prior annotation correlation and posterior prediction correlation. Then, we derive a differentiable rectification loss to force OC pixels to shift to the IC group. Incorporating OCR with seminal baselines (e.g., AffinityNet, SEAM, MCTformer), we can achieve remarkable performance gains on both Pascal VOC (+3.2%, +3.3%, +0.8% mIoU) and MS COCO (+1.0%, +1.3%, +0.5% mIoU) datasets with negligible extra training overhead, which jus-tifies the effectiveness and generality of OCR. † † Ŋ github.com/sennnnn/Out-of-Candidate-Rectification
What problem does this paper attempt to address?