SemPLeS: Semantic Prompt Learning for Weakly-Supervised Semantic Segmentation

Ci-Siang Lin,Chien-Yi Wang,Yu-Chiang Frank Wang,Min-Hung Chen

2024-03-11

Abstract:Weakly-Supervised Semantic Segmentation (WSSS) aims to train segmentation models using image data with only image-level supervision. Since precise pixel-level annotations are not accessible, existing methods typically focus on producing pseudo masks for training segmentation models by refining CAM-like heatmaps. However, the produced heatmaps may capture only the discriminative image regions of object categories or the associated co-occurring backgrounds. To address the issues, we propose a Semantic Prompt Learning for WSSS (SemPLeS) framework, which learns to effectively prompt the CLIP latent space to enhance the semantic alignment between the segmented regions and the target object categories. More specifically, we propose Contrastive Prompt Learning and Prompt-guided Semantic Refinement to learn the prompts that adequately describe and suppress the co-occurring backgrounds associated with each target object category. In this way, SemPLeS can perform better semantic alignment between object regions and the associated class labels, resulting in desired pseudo masks for training the segmentation model. The proposed SemPLeS framework achieves SOTA performance on the standard WSSS benchmarks, PASCAL VOC and MS COCO, and shows compatibility with other WSSS methods. The source codes are provided in the supplementary.

Computer Vision and Pattern Recognition,Computation and Language,Machine Learning

What problem does this paper attempt to address?

The paper focuses on the problem of Weakly-Supervised Semantic Segmentation (WSSS). In WSSS, the goal is to train a segmentation model using only image-level labels, without the need for accurate pixel-level annotations. Existing methods typically generate pseudo ground-truth masks by refining Class Activation Maps (CAM). However, these methods may only capture discriminative image regions within object categories or relevant common backgrounds.

The paper proposes a new framework called Semantic Prompt Learning for Weakly-Supervised Semantic Segmentation (SemPLeS), which leverages the latent space of the CLIP (Contrastive Language-Image Pretraining) model to enhance the semantic alignment between segmentation regions and target object categories. Specifically, the paper introduces Contrastive Prompt Learning and Prompt-guided Semantic Refinement to learn prompts that can adequately describe and suppress common backgrounds associated with each target category. This approach improves the semantic matching between object regions and their corresponding categories, resulting in the desired activation maps for training the segmentation model.

The SemPLeS framework achieves state-of-the-art performance on the standard WSSS benchmarks, PASCAL VOC and MS COCO, and is compatible with other WSSS methods. Compared to previous methods that required manual prompt engineering, SemPLeS automatically learns prompts embedded with class-associated semantic knowledge from the latent space of CLIP without any manual effort.

In summary, this paper aims to address the problem of weakly-supervised semantic segmentation by effectively utilizing the latent space of the CLIP model and learning semantic prompts, thereby improving the performance and accuracy of the segmentation model in the absence of accurate pixel-level annotations.