Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Textual Query-Driven Mask Transformer for Domain Generalized Segmentation

About

In this paper, we introduce a method to tackle Domain Generalized Semantic Segmentation (DGSS) by utilizing domain-invariant semantic knowledge from text embeddings of vision-language models. We employ the text embeddings as object queries within a transformer-based segmentation framework (textual object queries). These queries are regarded as a domain-invariant basis for pixel grouping in DGSS. To leverage the power of textual object queries, we introduce a novel framework named the textual query-driven mask transformer (tqdm). Our tqdm aims to (1) generate textual object queries that maximally encode domain-invariant semantics and (2) enhance the semantic clarity of dense visual features. Additionally, we suggest three regularization losses to improve the efficacy of tqdm by aligning between visual and textual features. By utilizing our method, the model can comprehend inherent semantic information for classes of interest, enabling it to generalize to extreme domains (e.g., sketch style). Our tqdm achieves 68.9 mIoU on GTA5$\rightarrow$Cityscapes, outperforming the prior state-of-the-art method by 2.5 mIoU. The project page is available at https://byeonghyunpak.github.io/tqdm.

Byeonghyun Pak, Byeongju Woo, Sunghwan Kim, Dae-hwan Kim, Hoseong Kim• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU48.1
2888
Semantic segmentationGTA5 → Cityscapes (val)
mIoU68.9
586
Semantic segmentationSYNTHIA to Cityscapes (val)--
480
Semantic segmentationGTA5 to {Cityscapes, Mapillary, BDD} (test)
mIoU (Cityscapes)68.88
94
Semantic segmentationCityscapes 1/4 (744 labels)
mIoU81.6
91
Semantic segmentationCityScapes, BDD, and Mapillary (val)
Mean mIoU66.05
85
Semantic segmentationCityscapes 1/16 (186 labeled samples)
mIoU75.5
78
Semantic segmentationMapillary
mIoU76.15
75
Semantic segmentationMapillary (test)
mIoU76.15
43
Semantic segmentationGTA5 → {Cityscapes, BDD100K, Mapillary} (Target Domains)
Score (Cityscapes)68.88
36
Showing 10 of 28 rows

Other info

Code

Follow for update