arXiv:2505.21501cs.CV2025-05NeurIPS被引 15

给预训练ViT添加注册令牌,无需重训即可减少伪影、提升定位精度。

Vision Transformers with Self-Distilled Registers

  • 用自蒸馏方式在不重训下向现有ViT注入随机初始化的注册令牌。
  • 在零样本和线性探测下,显著减少伪影令牌数,提升分割与深度预测性能。
  • 适合想优化已有ViT模型但无资源重训的研究者或工程师。

视觉变换器(ViTs)已成为视觉处理任务的主流架构,展现出随训练数据和模型规模增长而优异的可扩展性。然而,近期研究发现ViTs中存在与局部语义不符的伪影令牌,这类异常令牌会损害需要精细定位或结构一致性的任务表现。有效缓解该问题的方法是向ViTs中引入注册令牌,以在训练过程中隐式“吸收”伪影。鉴于现有大规模预训练ViTs的存在,本文旨在不从头训练的前提下将注册令牌加入现有模型,因完全重训在如此大模型下不可行。为此,提出后处理注册(PH-Reg),一种高效自蒸馏方法,可在无需额外标注数据且不需完整重训的情况下将注册令牌融入现有ViT。PH-Reg将教师与学生网络均初始化为同一预训练ViT,教师保持冻结不变,学生则增加随机初始化的注册令牌。通过在测试时对教师输入施加增强,生成无伪影的密集嵌入,并仅优化学生中解锁的一小部分权重。实验表明,该方法能有效减少伪影令牌数量,提升学生ViT在零样本和线性探测下的分割与深度预测性能。

原文摘要 · Abstract (English)

Vision Transformers (ViTs) have emerged as the dominant architecture for visual processing tasks, demonstrating excellent scalability with increased training data and model size. However, recent work has identified the emergence of artifact tokens in ViTs that are incongruous with local semantics. These anomalous tokens degrade ViT performance in tasks that require fine-grained localization or structural coherence. An effective mitigation of this issue is the addition of register tokens to ViTs, which implicitly "absorb" the artifact term during training. Given the availability of existing large-scale pre-trained ViTs, in this paper we seek add register tokens to existing models without needing to re-train from scratch, which is infeasible considering their size. Specifically, we propose Post Hoc Registers (PH-Reg), an efficient self-distillation method that integrates registers into an existing ViT without requiring additional labeled data and full retraining. PH-Reg initializes both teacher and student networks from the same pre-trained ViT. The teacher remains frozen and unmodified, while the student is augmented with randomly initialized register tokens. By applying test-time augmentation to the teacher's inputs, we generate denoised dense embeddings free of artifacts, which are then used to optimize only a small subset of unlocked student weights. We show that our approach can effectively reduce the number of artifact tokens, improving the segmentation and depth prediction of the student ViT under zero-shot and linear probing.

视觉变换器注册令牌自蒸馏模型优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。