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

用于少样本学习的具有金字塔提示的KNN Transformer

KNN Transformer with Pyramid Prompts for Few-Shot Learning

Shandong University(山东大学)

Wenhao Li, Qiangchang Wang, Peng Zhao, Yilong Yin

AI总结:

针对少样本学习中视觉与文本特征语义交互困难及无关token干扰问题,提出KTPP模型,利用KCA选择判别性token并引入金字塔跨模态提示自适应调制视觉特征,有效提取广义视觉表征。

摘要:

少样本学习(FSL)旨在利用有限的标注数据识别新类别。近期研究尝试利用文本提示来调制视觉特征以解决样本稀少的挑战。然而,它们通常难以捕捉文本与视觉特征之间复杂的语义关系。此外,普通自注意力机制受图像中无用信息影响严重,由于在交互过程中众多不相关token的干扰,严重限制了FSL中语义先验的潜力。为了解决上述问题,提出了一种带有金字塔提示的K-NN Transformer(KTPP),通过K-NN上下文注意力(KCA)选择判别性信息,并利用金字塔跨模态提示(PCP)自适应地调制视觉特征。首先,对于每个token,KCA仅选择K个最相关的token来计算自注意力矩阵,并将所有token的均值作为上下文提示,在三个级联阶段中提供全局上下文。从而,不相关的token可以被逐步抑制。其次,PCP中引入金字塔提示,通过基于文本的类感知提示与多尺度视觉特征之间的交互来强调视觉特征。这使得ViT能够根据不同尺度的丰富语义信息动态调整视觉特征的重要性权重,使模型对空间变化具有鲁棒性。最后,增强的视觉特征和类感知提示通过KCA进行交互,以提取特定类别的特征。因此,模型通过深度的跨模态交互进一步增强了无噪声的视觉表示,在少标注样本场景中提取通用的视觉表示。在四个基准数据集上的大量实验证明了本方法的有效性。

英文摘要:

Few-Shot Learning (FSL) aims to recognize new classes with limited labeled data. Recent studies have attempted to address the challenge of rare samples with textual prompts to modulate visual features. However, they usually struggle to capture complex semantic relationships between textual and visual features. Moreover, vanilla self-attention is heavily affected by useless information in images, severely constraining the potential of semantic priors in FSL due to the confusion of numerous irrelevant tokens during interaction. To address these aforementioned issues, a K-NN Transformer with Pyramid Prompts (KTPP) is proposed to select discriminative information with K-NN Context Attention (KCA) and adaptively modulate visual features with Pyramid Cross-modal Prompts (PCP). First, for each token, the KCA only selects the K most relevant tokens to compute the self-attention matrix and incorporates the mean of all tokens as the context prompt to provide the global context in three cascaded stages. As a result, irrelevant tokens can be progressively suppressed. Secondly, pyramid prompts are introduced in the PCP to emphasize visual features via interactions between text-based class-aware prompts and multi-scale visual features. This allows the ViT to dynamically adjust the importance weights of visual features based on rich semantic information at different scales, making models robust to spatial variations. Finally, augmented visual features and class-aware prompts are interacted via the KCA to extract class-specific features. Consequently, our model further enhances noise-free visual representations via deep cross-modal interactions, extracting generalized visual representation in scenarios with few labeled samples. Extensive experiments on four benchmark datasets demonstrate the effectiveness of our method.

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

↑