arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.04440cs.CV

现成的 Vision Transformer:面向小样本类别无关计数的令人惊讶的基线

Vision Transformer Off-the-Shelf: A Surprising Baseline for Few-Shot Class-Agnostic Counting

Huazhong University of Science and Technology(华中科技大学)

Zhicheng Wang, Liwen Xiao, Zhiguo Cao, Hao Lu

AI总结:

本文提出 CACViT,将类别无关计数简化为单个预训练普通 ViT 中的同步特征提取与匹配,并通过尺度和数量级嵌入补偿信息损失,在 FSC147 与 CARPK 上显著优于现有方法。

摘要:

类别无关计数(CAC)旨在给定少量示例的情况下,从查询图像中统计感兴趣对象的数量。该任务通常通过分别提取查询图像和示例的特征,然后匹配它们的特征相似性来解决,从而形成“先提取后匹配”的范式。在本工作中,我们表明 CAC 可以以“提取并匹配”的方式得到简化,尤其是使用 vision transformer(ViT)时,特征提取和相似性匹配可在自注意力内部同时执行。我们从自注意力的解耦视角揭示了这种简化的基本原理。由此得到的模型称为 CACViT,它将 CAC 流程简化为单个预训练的普通 ViT。此外,为弥补普通 ViT 中由于缩放和归一化造成的尺度与数量级信息损失,我们提出了两种有效的尺度嵌入和数量级嵌入策略。在 FSC147 和 CARPK 数据集上的大量实验表明,CACViT 在有效性(误差降低 23.60%)和泛化性方面均显著优于最先进的 CAC 方法,这表明 CACViT 为 CAC 提供了一个简洁而强大的基线。代码将会公开。

英文摘要:

Class-agnostic counting (CAC) aims to count objects of interest from a query image given few exemplars. This task is typically addressed by extracting the features of query image and exemplars respectively and then matching their feature similarity, leading to an extract-then-match paradigm. In this work, we show that CAC can be simplified in an extract-and-match manner, particularly using a vision transformer (ViT) where feature extraction and similarity matching are executed simultaneously within the self-attention. We reveal the rationale of such simplification from a decoupled view of the self-attention. The resulting model, termed CACViT, simplifies the CAC pipeline into a single pretrained plain ViT. Further, to compensate the loss of the scale and the order-of-magnitude information due to resizing and normalization in plain ViT, we present two effective strategies for scale and magnitude embedding. Extensive experiments on the FSC147 and the CARPK datasets show that CACViT significantly outperforms state-of-the art CAC approaches in both effectiveness (23.60% error reduction) and generalization, which suggests CACViT provides a concise and strong baseline for CAC. Code will be available.

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

↑