ToSA: Token Merging with Spatial Awareness
Keywords: 3D Understanding, Token Merging
TL;DR: We propose a spatial-aware token merging method named ToSA, which demosntrate better performance than pervious semantic-only token merging method ToMe.
Abstract: Token merging has emerged as an effective strategy to accelerate Vision Transformers~(ViT) by reducing computational costs. However, existing methods primarily rely on the visual token's feature similarity for token merging, overlooking the potential of integrating spatial information, which can serve as a reliable criterion for token merging in the early layers of ViT, where the visual tokens only possess weak visual information. In this paper, we propose ToSA, a novel token merging method that combines both semantic and spatial awareness to guide the token merging process. ToSA leverages the depth image as input to generate pseudo spatial tokens, which serve as auxiliary spatial information for the visual token merging process. With the introduced spatial awareness, ToSA achieves a more informed merging strategy that better preserves critical scene structure. Experimental results demonstrate that ToSA outperforms previous token merging methods across multiple benchmarks on image question answering and spatial understanding while largely reducing the runtime of the ViT, making it an efficient solution for ViT acceleration.
Submission Number: 1
Loading