Area Intervention for Enhancing Class Activation Maps in Weakly Supervised Semantic Segmentation

Xuewei Li,Yujie Diao,Mei Yu,Chenhan Wang,Jie Gao,Ruiguo Yu
DOI: https://doi.org/10.1109/icme57554.2024.10687454
2024-01-01
Abstract:Generating class activation maps (CAM) as seed regions is a crucial step in weakly supervised semantic segmentation (WSSS). During the training of classification models, instance area information is incorporated into feature vectors and learned by the network. However, during CAM generation, pixel-level prediction scores are solely dependent on features and classifier weights, neglecting the area factor. Such area information becomes a source of bias between training and CAM generation, leading to a significant under-activation (false negatives) of foreground regions. To address this issue, we propose a class area intervention strategy to fine-tune the classification model and rectify this bias. Additionally, we employ a partition activation strategy that can simultaneously consider both under activation and over-activation to generate more precise AICAMs. AICAM is applicable to various WSSS methods based on CAM or its variants. The evaluations on both the PASCAL VOC and MS COCO datasets demonstrate that our method enhances multiple WSSS techniques.
What problem does this paper attempt to address?