arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2407.09033cs.CV

面向域泛化分割的文本查询驱动掩码Transformer

Textual Query-Driven Mask Transformer for Domain Generalized Segmentation

Agency for Defense Development (ADD)(国防开发局)

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

AI总结:

本文提出文本查询驱动掩码Transformer(tqdm),利用视觉-语言模型文本嵌入作为域不变目标查询,并通过三种正则化损失对齐视觉与文本特征,在GTA5→Cityscapes上以68.9 mIoU超越先前最优方法2.5 mIoU。

摘要:

在本文中,我们提出一种方法,通过利用视觉-语言模型文本嵌入中的域不变语义知识来应对域泛化语义分割(DGSS)。我们将文本嵌入用作基于Transformer的分割框架中的目标查询(文本目标查询)。这些查询被视为DGSS中像素分组的域不变基础。为了发挥文本目标查询的能力,我们引入了一个名为文本查询驱动掩码Transformer(tqdm)的新框架。我们的tqdm旨在(1)生成最大程度编码域不变语义的文本目标查询,以及(2)增强密集视觉特征的语义清晰度。此外,我们提出三种正则化损失,通过对齐视觉特征与文本特征来提高tqdm的有效性。利用我们的方法,模型能够理解感兴趣类别的固有语义信息,从而使其能够泛化到极端域(例如素描风格)。我们的tqdm在GTA5→Cityscapes上达到68.9 mIoU,比先前最先进的方法高出2.5 mIoU。项目页面可在 https://byeonghyunpak.github.io/tqdm 获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑