arXiv:2411.02074cs.CV2024-11中稿 · ACM ICVGIP 2024被引 3

用图神经网络增强视觉语言模型,提升未知类别发现的准确性

GraphVL: Graph-Enhanced Semantic Modeling via Vision-Language Models for Generalized Class Discovery

  • 将图卷积网络与CLIP文本编码器结合,保持类别间邻近关系
  • 在7个数据集上优于现有方法,显著降低对已知类别的偏差
  • 适合做开放世界图像分类与半监督聚类的研究者

广义类别发现(GCD)旨在利用已知类别的标注图像,将未标注图像聚类到已知和未知类别中。为解决从已知类向未知类迁移特征时带来的模型偏见问题,我们提出GraphVL,一种基于CLIP的视觉-语言建模新方法。该方法将图卷积网络(GCN)与CLIP的文本编码器融合,以保留类别间的邻接结构。同时采用轻量级视觉投影器处理图像数据,通过基于边距的对比损失实现图像-文本映射的判别性特征提取。邻域保持机制有效规范了语义空间,使其对已知类别不敏感。此外,从已知类别学习文本提示,并通过上下文相似性损失进行对齐,构建更具语境意义的语义空间供GCN使用。最终,基于未标注样本与GCN生成的类别提示之间的语义距离进行表示,实现半监督聚类与类别发现,减少错误。在七个基准数据集上的实验一致表明,GraphVL在集成CLIP主干网络时表现更优。

原文摘要 · Abstract (English)

Generalized Category Discovery (GCD) aims to cluster unlabeled images into known and novel categories using labeled images from known classes. To address the challenge of transferring features from known to unknown classes while mitigating model bias, we introduce GraphVL, a novel approach for vision-language modeling in GCD, leveraging CLIP. Our method integrates a graph convolutional network (GCN) with CLIP's text encoder to preserve class neighborhood structure. We also employ a lightweight visual projector for image data, ensuring discriminative features through margin-based contrastive losses for image-text mapping. This neighborhood preservation criterion effectively regulates the semantic space, making it less sensitive to known classes. Additionally, we learn textual prompts from known classes and align them to create a more contextually meaningful semantic feature space for the GCN layer using a contextual similarity loss. Finally, we represent unlabeled samples based on their semantic distance to class prompts from the GCN, enabling semi-supervised clustering for class discovery and minimizing errors. Our experiments on seven benchmark datasets consistently demonstrate the superiority of GraphVL when integrated with the CLIP backbone.

类别发现视觉语言图神经网络半监督

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。