Prompt Categories Cluster for Weakly Supervised Semantic Segmentation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wu, Wangyu, Qiu, Xianglin, Song, Siqi, Huang, Xiaowei, Ma, Fei, Xiao, Jimin
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910904851890176
author Wu, Wangyu
Qiu, Xianglin
Song, Siqi
Huang, Xiaowei
Ma, Fei
Xiao, Jimin
author_facet Wu, Wangyu
Qiu, Xianglin
Song, Siqi
Huang, Xiaowei
Ma, Fei
Xiao, Jimin
contents Weakly Supervised Semantic Segmentation (WSSS), which leverages image-level labels, has garnered significant attention due to its cost-effectiveness. The previous methods mainly strengthen the inter-class differences to avoid class semantic ambiguity which may lead to erroneous activation. However, they overlook the positive function of some shared information between similar classes. Categories within the same cluster share some similar features. Allowing the model to recognize these features can further relieve the semantic ambiguity between these classes. To effectively identify and utilize this shared information, in this paper, we introduce a novel WSSS framework called Prompt Categories Clustering (PCC). Specifically, we explore the ability of Large Language Models (LLMs) to derive category clusters through prompts. These clusters effectively represent the intrinsic relationships between categories. By integrating this relational information into the training network, our model is able to better learn the hidden connections between categories. Experimental results demonstrate the effectiveness of our approach, showing its ability to enhance performance on the PASCAL VOC 2012 dataset and surpass existing state-of-the-art methods in WSSS.
format Preprint
id arxiv_https___arxiv_org_abs_2412_13823
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Prompt Categories Cluster for Weakly Supervised Semantic Segmentation
Wu, Wangyu
Qiu, Xianglin
Song, Siqi
Huang, Xiaowei
Ma, Fei
Xiao, Jimin
Computer Vision and Pattern Recognition
Weakly Supervised Semantic Segmentation (WSSS), which leverages image-level labels, has garnered significant attention due to its cost-effectiveness. The previous methods mainly strengthen the inter-class differences to avoid class semantic ambiguity which may lead to erroneous activation. However, they overlook the positive function of some shared information between similar classes. Categories within the same cluster share some similar features. Allowing the model to recognize these features can further relieve the semantic ambiguity between these classes. To effectively identify and utilize this shared information, in this paper, we introduce a novel WSSS framework called Prompt Categories Clustering (PCC). Specifically, we explore the ability of Large Language Models (LLMs) to derive category clusters through prompts. These clusters effectively represent the intrinsic relationships between categories. By integrating this relational information into the training network, our model is able to better learn the hidden connections between categories. Experimental results demonstrate the effectiveness of our approach, showing its ability to enhance performance on the PASCAL VOC 2012 dataset and surpass existing state-of-the-art methods in WSSS.
title Prompt Categories Cluster for Weakly Supervised Semantic Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.13823