Geometry without Position? When Positional Embeddings Help and Hurt Spatial Reasoning
Jian Shi, Michael Birsak, Wenqing Cui, Zhenyu Li, Peter Wonka
TL;DR
The paper addresses how positional embeddings govern geometry in vision transformers, arguing that PEs act as a learned spatial kernel that can both help and hinder spatial reasoning across views. It introduces token-level diagnostics to quantify multi-view geometric consistency and demonstrates, across 14 foundation ViTs, that geometry is driven by PE consistency rather than content alone. The authors propose a training-free token re-indexing to restore PE alignment and analyze VGGT to show how multi-view aggregation can implicitly infer a canonical coordinate system. The findings reveal PEs as a causal mechanism shaping spatial structure in ViTs, with practical implications for designing robust multi-view vision models.
Abstract
This paper revisits the role of positional embeddings (PEs) within vision transformers (ViTs) from a geometric perspective. We show that PEs are not mere token indices but effectively function as geometric priors that shape the spatial structure of the representation. We introduce token-level diagnostics that measure how multi-view geometric consistency in ViT representation depends on consitent PEs. Through extensive experiments on 14 foundation ViT models, we reveal how PEs influence multi-view geometry and spatial reasoning. Our findings clarify the role of PEs as a causal mechanism that governs spatial structure in ViT representations. Our code is provided in https://github.com/shijianjian/vit-geometry-probes
