arXiv:2605.19622cs.CV2026-05

让预训练视觉模型学会自动清理干扰信息,提升空间感知能力。

UniRefiner: Teaching Pre-trained ViTs to Self-Dispose Dross via Contrastive Register

论文配图:UniRefiner: Teaching Pre-trained ViTs to Self-Dispose Dross via Contrastive Register
图 1 · 摘自论文原文
  • 用对比寄存器识别并重分配干扰性图像令牌
  • 微调仅需5000张图,使大模型在ADE20K上达51.9% mIoU
  • 适合需要高精度空间理解的视觉任务开发者

基于视觉变换器(ViTs)的表征学习发展迅速,但大规模模型在空间敏感任务中常受冗余令牌干扰。现有方法多将此类干扰定义为简单高范数异常值,范围过窄。本文提出更广义的定义:任何未能编码位置对齐语义的令牌均为冗余。据此系统分类出三类关键冗余令牌。基于此诊断,提出通用精炼框架UniRefiner,利用对比寄存器通过双目标机制实现自净化:一、将图像令牌与过滤后的正常令牌对齐以保留语义;二、将寄存器令牌与检测到的冗余令牌对齐以捕获干扰信号。仅需约5000张图像的几轮微调,即可优化多种ViT模型,包括EVA-CLIP-8B和InternViT-6B。实验显示显著提升:经精炼的EVA-CLIP-8B在ADE20K上达51.9% mIoU(+9.4%),超越DINOv2(49.1%),零样本分割准确率最高提升22%。UniRefiner释放了现有大模型的潜在空间能力,推动其广泛应用。

原文摘要 · Abstract (English)

Representation learning with Vision Transformers (ViTs) has advanced rapidly, yet the utility of large-scale models in spatially sensitive tasks is hindered by spurious tokens. Prior efforts to mitigate this have been limited, often defining these artifacts narrowly, for example, as simple high-norm outliers. We argue that this scope is insufficient. For dense prediction tasks, we posit that any token failing to encode location-aligned semantics should be treated as a spurious artifact. This broader definition reveals a more complex problem, leading us to systematically categorize and characterize three fundamental types of spurious tokens that corrupt spatial representations. Based on this comprehensive diagnosis, we propose UniRefiner, a universal refinement framework that teaches pre-trained ViTs to self-dispose of these artifacts. UniRefiner uses contrastive registers to explicitly isolate and redistribute spurious tokens via a dual objective: (i) it aligns image tokens with filtered regular tokens to preserve semantics, and (ii) it aligns register tokens with detected spurious tokens to capture the spurious signals. Our method requires only a few epochs of fine-tuning on ~5k images to refine diverse ViTs, including massive models like EVA-CLIP-8B and InternViT-6B. Experiments demonstrate consistent and significant improvements: notably, the refined EVA-CLIP-8B achieves 51.9\% mIoU on ADE20K (+9.4\%), surpassing specialized vision models like DINOv2 (49.1\%), while zero-shot segmentation accuracy improves by up to 22\%. UniRefiner unlocks the latent spatial potential of existing large-scale foundation models, paving the way for their broader application.

视觉变换器模型精炼空间感知自净化

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