Computational Vision

Donald Bren School of Information and Computer Sciences, UC Irvine

GViT: Representing Images as Gaussians for Visual Recognition

Jefferson Hernandez, Ruozhen He, Guha Balakrishnan, Alexander C. Berg, Vicente Ordonez

arXiv:2506.23532, 2025.

Abstract

We introduce GVIT, a classification framework that abandons conventional pixel or patch grid input representations in favor of a compact set of learnable 2D Gaussians. Each image is encoded as a few hundred Gaussians whose positions, scales, orientations, colors, and opacities are optimized jointly with a ViT classifier trained on top of these representations. We reuse the classifier gradients as constructive guidance, steering the Gaussians toward class-salient regions while a differentiable renderer optimizes an image reconstruction loss. We demonstrate that by 2D Gaussian input representations coupled with our GVIT guidance, using a relatively standard ViT architecture, closely matches the performance of a traditional patch-based ViT, reaching a 76.9% top-1 accuracy on Imagenet-1k using a ViT-B architecture.

Cite

@article{gvit-representing-images-as-gaussians-for-visual-recognition-2025, author = {Jefferson Hernandez and Ruozhen He and Guha Balakrishnan and Alexander C. Berg and Vicente Ordonez}, title = {GViT: Representing Images as Gaussians for Visual Recognition}, year = {2025}, eprint = {2506.23532}, archivePrefix = {arXiv}, url = {https://arxiv.org/abs/2506.23532}, }