SPAR: Single-Pass Any-Resolution ViT for Open-vocabulary Segmentation
AI 摘要
SPAR提出了一种单次Any-Resolution ViT,通过知识蒸馏实现高效的高分辨率开放词汇分割。
主要贡献
- 提出了SPAR,一种resolution-agnostic的ViT
- 使用知识蒸馏将sliding-window teacher的空间推理能力传递给single-pass student
- 在开放词汇分割任务中取得了显著的性能提升
方法论
使用feature regression loss进行知识蒸馏,将高分辨率sliding-window teacher模型的知识迁移到single-pass student模型。
原文摘要
Foundational Vision Transformers (ViTs) have limited effectiveness in tasks requiring fine-grained spatial understanding, due to their fixed pre-training resolution and inherently coarse patch-level representations. These challenges are especially pronounced in dense prediction scenarios, such as open-vocabulary segmentation with ViT-based vision-language models, where high-resolution inputs are essential for accurate pixel-level reasoning. Existing approaches typically process large-resolution images using a sliding-window strategy at the pre-training resolution. While this improves accuracy through finer strides, it comes at a significant computational cost. We introduce SPAR: Single-Pass Any-Resolution ViT, a resolution-agnostic dense feature extractor designed for efficient high-resolution inference. We distill the spatial reasoning capabilities of a finely-strided, sliding-window teacher into a single-pass student using a feature regression loss, without requiring architectural changes or pixel-level supervision. Applied to open-vocabulary segmentation, SPAR improves single-pass baselines by up to 10.5 mIoU and even surpasses the teacher, demonstrating effectiveness in efficient, high-resolution reasoning. Code: https://github.com/naomikombol/SPAR