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

利用 Vision Transformers 中的寄存器实现鲁棒适应

Leveraging Registers in Vision Transformers for Robust Adaptation

Stony Brook University(石溪大学);Lawrence Livermore National Laboratory(劳伦斯利弗莫尔国家实验室)

Srikar Yellapragada, Kowshik Thopalli, Vivek Narayanaswamy, Wesam Sakla, Yang Liu, Yamen Mubarka, Dimitris Samaras, Jayaraman J. Thiagarajan

AI总结:

本文提出将 ViT 的 CLS token 与平均池化寄存器 embedding 融合以训练下游分类器,在不增加计算开销的前提下提升 OOD 泛化与异常拒绝。

摘要:

Vision Transformers(ViTs)因其捕捉全局图像表征的能力,已在多种任务中取得成功。近期研究发现,ViT 中存在高范数 token,这些 token 会干扰无监督目标发现。为解决该问题,研究者提出使用“registers(寄存器)”,即额外 token,用于隔离高范数 patch token,同时捕捉全局图像级信息。尽管寄存器已在目标发现中得到广泛研究,但其泛化特性,尤其是在分布外(OOD)场景下的表现,仍未得到充分探索。本文考察寄存器 token embedding 在提供额外特征以提升泛化和异常拒绝能力方面的效用。为此,我们提出一种简单方法,将 ViT 中常用的特殊 CLS token embedding 与经平均池化的寄存器 embedding 相结合,构建特征表征,并随后用于训练下游分类器。我们发现,这增强了 OOD 泛化和异常拒绝能力,同时保持了分布内(ID)性能。在多种 ViT 骨干网络上的大量实验(包括使用和未使用寄存器训练的模型)表明,top-1 OOD 准确率持续提升 2–4%,异常检测的假阳性率降低 2–3%。重要的是,这些增益不会带来额外计算开销。

英文摘要:

Vision Transformers (ViTs) have shown success across a variety of tasks due to their ability to capture global image representations. Recent studies have identified the existence of high-norm tokens in ViTs, which can interfere with unsupervised object discovery. To address this, the use of "registers" which are additional tokens that isolate high norm patch tokens while capturing global image-level information has been proposed. While registers have been studied extensively for object discovery, their generalization properties particularly in out-of-distribution (OOD) scenarios, remains underexplored. In this paper, we examine the utility of register token embeddings in providing additional features for improving generalization and anomaly rejection. To that end, we propose a simple method that combines the special CLS token embedding commonly employed in ViTs with the average-pooled register embeddings to create feature representations which are subsequently used for training a downstream classifier. We find that this enhances OOD generalization and anomaly rejection, while maintaining in-distribution (ID) performance. Extensive experiments across multiple ViT backbones trained with and without registers reveal consistent improvements of 2-4\% in top-1 OOD accuracy and a 2-3\% reduction in false positive rates for anomaly detection. Importantly, these gains are achieved without additional computational overhead.

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

↑