Learning Sparse Visual Representations via Spatial-Semantic Factorization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhao, Theodore Zhengde, Kiblawi, Sid, Yang, Jianwei, Usuyama, Naoto, Tan, Reuben, Codella, Noel C, Naumann, Tristan, Poon, Hoifung, Wei, Mu
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914301253517312
author Zhao, Theodore Zhengde
Kiblawi, Sid
Yang, Jianwei
Usuyama, Naoto
Tan, Reuben
Codella, Noel C
Naumann, Tristan
Poon, Hoifung
Wei, Mu
author_facet Zhao, Theodore Zhengde
Kiblawi, Sid
Yang, Jianwei
Usuyama, Naoto
Tan, Reuben
Codella, Noel C
Naumann, Tristan
Poon, Hoifung
Wei, Mu
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2602_01905
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learning Sparse Visual Representations via Spatial-Semantic Factorization
Zhao, Theodore Zhengde
Kiblawi, Sid
Yang, Jianwei
Usuyama, Naoto
Tan, Reuben
Codella, Noel C
Naumann, Tristan
Poon, Hoifung
Wei, Mu
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
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.
title Learning Sparse Visual Representations via Spatial-Semantic Factorization
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2602.01905