Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ViT-Up: Faithful Feature Upsampling for Vision Transformers

About

Vision Transformers (ViTs) have become a dominant architecture for visual representation learning, providing exceptionally strong and broadly reusable backbone features. However, ViTs are commonly operated on relatively small patch-token grids due to the quadratic cost of global self-attention, which creates a persistent bottleneck for dense prediction tasks such as semantic segmentation and depth estimation. This has motivated the development of task-agnostic feature upsamplers. While recent state-of-the-art methods produce visually sharp dense representations, their reliance on shallow image encoders for guided upsampling can introduce feature leakage, fragmentation, and blur. We introduce ViT-Up, an implicit feature upsampling framework that replaces external image guidance with layer-wise query construction from intermediate ViT hidden states. This enables feature prediction at arbitrary continuous image coordinates while preserving alignment with the backbone feature space. Experiments demonstrate that ViT-Up consistently outperforms state-of-the-art image-guided upsamplers across dense prediction and semantic correspondence. On DINOv3-S+, ViT-Up improves over prior methods by up to +2.07 mIoU on Cityscapes and +4.17 PCK@0.10 on SPair-71k. With the larger DINOv3-B backbone, these gains increase to +3.36 mIoU and +8.09 PCK@0.10, demonstrating that ViT-Up scales favorably with backbone capacity.

Krispin Wandel, Jingchuan Wang, Hesheng Wang• 2026

Related benchmarks

TaskDatasetResultRank
Semantic segmentationCityscapes
mIoU69.81
140
Semantic segmentationADE20K
mIoU44.73
90
Semantic segmentationVOC
mIoU89.18
64
Semantic CorrespondenceSPair-71k
PCK @ 0.017.93
40
Semantic segmentationCOCO
mIoU64.09
25
Geometric CorrespondenceNAVI
Avg. Recall80.81
14
Depth EstimationCOCO
Delta 1 (δ1)62.72
6
Semantic segmentationCityscapes
mIoU65.41
6
Showing 8 of 8 rows

Other info

GitHub

Follow for update