通过构建视觉-语义图关系,提升零样本学习的泛化能力
Visual-Semantic Graph Matching Net for Zero-Shot Learning
- 引入图结构建模类别间语义关系,实现跨模态对齐
- 在三个基准数据集上超越现有方法,尤其在广义零样本场景下表现突出
- 适合研究零样本学习、图神经网络与跨模态对齐的学者参考
零样本学习(ZSL)旨在利用额外的语义信息识别未见类别。现有方法通常通过简单对齐视觉特征与语义原型来学习共享嵌入空间,但这种孤立的模态对齐方式忽略了类别间的关联性,导致嵌入空间鲁棒性不足。为此,本文提出视觉-语义图匹配网络(VSGMN),利用类别间的语义关系辅助视觉-语义嵌入对齐。VSGMN采用两阶段设计:首先,图构建网络(GBN)基于嵌入在语义空间中构建视觉与语义图,并对齐嵌入与其原型,同时根据语义关系补充未见类节点;其次,图匹配网络(GMN)在类关系约束下,持续融合邻域与跨图信息,对齐两图节点关系。在三个基准数据集上的大量实验表明,VSGMN在传统与广义零样本学习场景下均取得优异性能。代码与实验结果已公开于GitHub:https://github.com/dbwfd/VSGMN。
原文摘要 · Abstract (English)
Zero-shot learning (ZSL) aims to leverage additional semantic information to recognize unseen classes. To transfer knowledge from seen to unseen classes, most ZSL methods often learn a shared embedding space by simply aligning visual embeddings with semantic prototypes. However, methods trained under this paradigm often struggle to learn robust embedding space because they align the two modalities in an isolated manner among classes, which ignore the crucial class relationship during the alignment process. To address the aforementioned challenges, this paper proposes a Visual-Semantic Graph Matching Net, termed as VSGMN, which leverages semantic relationships among classes to aid in visual-semantic embedding. VSGMN employs a Graph Build Network (GBN) and a Graph Matching Network (GMN) to achieve two-stage visual-semantic alignment. Specifically, GBN first utilizes an embedding-based approach to build visual and semantic graphs in the semantic space and align the embedding with its prototype for first-stage alignment. Additionally, to supplement unseen class relations in these graphs, GBN also build the unseen class nodes based on semantic relationships. In the second stage, GMN continuously integrates neighbor and cross-graph information into the constructed graph nodes, and aligns the node relationships between the two graphs under the class relationship constraint. Extensive experiments on three benchmark datasets demonstrate that VSGMN achieves superior performance in both conventional and generalized ZSL scenarios. The implementation of our VSGMN and experimental results are available at github: https://github.com/dbwfd/VSGMN
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。