arXiv:2607.01987cs.CV2026-07中稿 · ECCV

通过子空间干预揭示自监督ViT如何编码几何信息

Understanding Geometric Representations in Self-Supervised Vision Transformers via Subspace Intervention

论文配图:Understanding Geometric Representations in Self-Supervised Vision Transformers via Subspace Intervention
图 1 · 摘自论文原文
  • 用SVD分解线性探测器权重,定位包含几何信号的低秩子空间
  • 发现DINOv2的特征更集中,MAE需更大上下文,中间层几何精度最高
  • 为轻量级解码器设计和特征筛选提供可解释依据

我们提出一种受控子空间干预框架,研究自监督视觉Transformer(ViTs)如何编码密集几何信息。传统线性探测将特征视为黑箱,无法解耦底层拓扑结构。为此,我们通过奇异值分解(SVD)分解收敛线性探测器权重,分离出包含显式几何信号的低秩子空间。研究得出三个关键发现:(1) 预训练目标决定特征编码方式:DINOv2使空间特征对齐,利于高效线性提取;而掩码自编码器(MAE)倾向于分散这些信号,需更广的空间上下文。(2) 显式几何表示高度可压缩,表明密集预测头可约束在低秩子空间,性能损失极小。(3) 层级任务亲和性显示,几何精度在中间层达到峰值,随后在最终层转向语义抽象。这些发现连接内部编码机制与下游性能,为有效特征选择和轻量级解码器设计提供基础。代码已公开于https://github.com/Zhou-Weichen/Geosubprobe。

原文摘要 · Abstract (English)

We introduce a controlled subspace intervention framework to investigate how self-supervised Vision Transformers (ViTs) encode dense geometric information. While linear probing is widely used to assess geometric representations, it treats features as a black box, failing to disentangle the underlying topology. To address this issue, we decompose the weights of converged linear probes to isolate the low-rank subspaces containing explicit geometric signals using Singular Value Decomposition (SVD). Our perspective yields three key insights: (1) Pre-training objectives determine how features are encoded. DINOv2 aligns spatial features for efficient linear extraction, while Masked Autoencoders (MAE) tend to disperse these signals, requiring a broader spatial context. (2) Explicit geometric representations are highly compressible, suggesting dense predictive heads could potentially be constrained to low-rank subspaces with minimal performance loss. (3) The layer-wise task affinity suggests that geometric precision peaks at intermediate layers before yielding to semantic abstraction in the final layers. By connecting internal encoding mechanics with downstream performance, these findings provide a basis for effective feature selection and lightweight decoder design. The source code is available at https://github.com/Zhou-Weichen/Geosubprobe.

自监督学习视觉变换器几何表示子空间分析

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