arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.08801cs.CV

CoBra:互补分支融合类别与语义知识以实现鲁棒的弱监督语义分割

CoBra: Complementary Branch Fusing Class and Semantic Knowledge for Robust Weakly Supervised Semantic Segmentation

Yonsei University(延世大学)

Woojung Han, Seil Kang, Kyobin Choo, Seong Jae Hwang

AI总结:

提出CoBra双分支框架,融合CNN的类别知识与ViT的语义知识,通过类别感知投影和语义感知投影生成鲁棒伪掩码,在PASCAL VOC 2012上实现最先进的弱监督语义分割。

摘要:

利用从图像级类别知识中导出的语义精确的伪掩码进行分割,即图像级弱监督语义分割(WSSS),仍然具有挑战性。虽然使用CNN的类激活图(CAMs)持续为WSSS的成功做出贡献,但生成的激活图通常狭窄地聚焦于类别特定的部分(例如,仅人的面部)。另一方面,基于视觉变换器(ViT)的近期工作凭借其自注意力机制在捕获语义部分方面显示出有希望的结果,但未能捕获完整的类别特定细节(例如,人的整个身体部分,但同时也包含附近的一条狗)。在这项工作中,我们提出了互补分支(CoBra),一种新颖的双分支框架,由两种不同的架构组成,为每个分支提供有价值的类别(来自CNN)和语义(来自ViT)互补知识。特别是,我们为CNN分支学习类别感知投影(CAP),为ViT分支学习语义感知投影(SAP),以显式融合它们的互补知识,并促进一种新型的额外补丁级监督。通过CoBra,我们的模型融合CNN和ViT的互补输出,以创建整合类别和语义信息的鲁棒伪掩码。在PASCAL VOC 2012数据集上的大量实验定性和定量地研究了CNN和ViT如何相互补充,展示了最先进的WSSS结果。这不仅包括我们模型生成的掩码,还包括利用这些掩码作为伪标签获得的分割结果。

英文摘要:

Leveraging semantically precise pseudo masks derived from image-level class knowledge for segmentation, namely image-level Weakly Supervised Semantic Segmentation (WSSS), still remains challenging. While Class Activation Maps (CAMs) using CNNs have steadily been contributing to the success of WSSS, the resulting activation maps often narrowly focus on class-specific parts (e.g., only face of human). On the other hand, recent works based on vision transformers (ViT) have shown promising results based on their self-attention mechanism to capture the semantic parts but fail in capturing complete class-specific details (e.g., entire body parts of human but also with a dog nearby). In this work, we propose Complementary Branch (CoBra), a novel dual branch framework consisting of two distinct architectures which provide valuable complementary knowledge of class (from CNN) and semantic (from ViT) to each branch. In particular, we learn Class-Aware Projection (CAP) for the CNN branch and Semantic-Aware Projection (SAP) for the ViT branch to explicitly fuse their complementary knowledge and facilitate a new type of extra patch-level supervision. Our model, through CoBra, fuses CNN and ViT's complementary outputs to create robust pseudo masks that integrate both class and semantic information effectively. Extensive experiments qualitatively and quantitatively investigate how CNN and ViT complement each other on the PASCAL VOC 2012 dataset, showing a state-of-the-art WSSS result. This includes not only the masks generated by our model, but also the segmentation results derived from utilizing these masks as pseudo labels.

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

↑