Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Learning Sparse Visual Representations via Spatial-Semantic Factorization

About

Self-supervised learning (SSL) faces a fundamental conflict between semantic understanding and image reconstruction. High-level semantic SSL (e.g., DINO) relies on global tokens that are forced to be location-invariant for augmentation alignment, a process that inherently discards the spatial coordinates required for reconstruction. Conversely, generative SSL (e.g., MAE) preserves dense feature grids for reconstruction but fails to produce high-level abstractions. We introduce STELLAR, a framework that resolves this tension by factorizing visual features into a low-rank product of semantic concepts and their spatial distributions. This disentanglement allows us to perform DINO-style augmentation alignment on the semantic tokens while maintaining the precise spatial mapping in the localization matrix necessary for pixel-level reconstruction. We demonstrate that as few as 16 sparse tokens under this factorized form are sufficient to simultaneously support high-quality reconstruction (2.60 FID) and match the semantic performance of dense backbones (79.10% ImageNet accuracy). Our results highlight STELLAR as a versatile sparse representation that bridges the gap between discriminative and generative vision by strategically separating semantic identity from spatial geometry. Code available at https://aka.ms/stellar.

Theodore Zhengde Zhao, Sid Kiblawi, Jianwei Yang, Naoto Usuyama, Reuben Tan, Noel C Codella, Tristan Naumann, Hoifung Poon, Mu Wei• 2026

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU41.98
2731
Semantic segmentationADE20K
mIoU36.66
936
Semantic segmentationCityscapes
mIoU33.3
578
Image ClassificationFood-101
Accuracy77.43
494
Image ClassificationImageNet-1k (val)
Accuracy80.05
189
Semantic segmentationPascal VOC
mIoU0.859
172
Image ClassificationOxford-IIIT Pet
Accuracy92.53
161
Image ReconstructionImageNet1K (val)
FID2.6
83
Text-to-Image RetrievalMS-COCO--
79
Image ClassificationImageNet-1K
Top-1 Acc51.53
75
Showing 10 of 15 rows

Other info

Follow for update