通过双向因果注意力提升零样本学习的语义迁移能力
Mutually Causal Semantic Distillation Network for Zero-Shot Learning
- 构建双向因果注意力机制,分别学习视觉到属性和属性到视觉的深层关联
- 在CUB、SUN、AWA2和FLO数据集上达到新最好性能,显著超越现有方法
- 适合关注零样本识别中语义建模与跨模态对齐的研究者
零样本学习(ZSL)旨在利用属性等辅助信息识别开放世界中的未见类别。其核心挑战在于如何从已见类别中推断视觉与属性特征间的潜在语义知识,并实现有效的语义迁移。以往方法仅采用单向注意力机制,在弱监督下学习到的是表面且有限的潜在语义表示,难以捕捉视觉与属性之间的内在语义关系。为此,本文提出互因果语义蒸馏网络(MSDN++),用于提取内在且充分的语义表示。MSDN++包含两个子网络:属性→视觉因果注意力模块,用于学习基于属性的视觉特征;视觉→属性因果注意力模块,用于学习基于视觉的属性特征。因果注意力促使两个子网学习到可靠的因果视觉-属性关联,实现因果化的视觉/属性表征。通过语义蒸馏损失的引导,两个相互注意子网在训练过程中协同优化、相互促进。在三个广泛使用的基准数据集(CUB、SUN、AWA2和FLO)上的大量实验表明,MSDN++显著优于强基线,达到新的最先进性能。
原文摘要 · Abstract (English)
Zero-shot learning (ZSL) aims to recognize the unseen classes in the open-world guided by the side-information (e.g., attributes). Its key task is how to infer the latent semantic knowledge between visual and attribute features on seen classes, and thus conducting a desirable semantic knowledge transfer from seen classes to unseen ones. Prior works simply utilize unidirectional attention within a weakly-supervised manner to learn the spurious and limited latent semantic representations, which fail to effectively discover the intrinsic semantic knowledge (e.g., attribute semantic) between visual and attribute features. To solve the above challenges, we propose a mutually causal semantic distillation network (termed MSDN++) to distill the intrinsic and sufficient semantic representations for ZSL. MSDN++ consists of an attribute$\rightarrow$visual causal attention sub-net that learns attribute-based visual features, and a visual$\rightarrow$attribute causal attention sub-net that learns visual-based attribute features. The causal attentions encourages the two sub-nets to learn causal vision-attribute associations for representing reliable features with causal visual/attribute learning. With the guidance of semantic distillation loss, the two mutual attention sub-nets learn collaboratively and teach each other throughout the training process. Extensive experiments on three widely-used benchmark datasets (e.g., CUB, SUN, AWA2, and FLO) show that our MSDN++ yields significant improvements over the strong baselines, leading to new state-of-the-art performances.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。