UViM: A Unified Modeling Approach for Vision with Learned Guiding Codes
Not on the map. Why?
Abstract (arXiv)
We introduce UViM, a unified approach capable of modeling a wide range of computer vision tasks. In contrast to previous models, UViM has the same functional form for all tasks; it requires no task-specific modifications which require extensive human expertise. The approach involves two components: (I) a base model (feed-forward) which is trained to directly predict raw vision outputs, guided by a learned discrete code and (II) a language model (autoregressive) that is trained to generate the guiding code. These components complement each other: the language model is well-suited to modeling structured interdependent data, while the base model is efficient at dealing with high-dimensional outputs. We demonstrate the effectiveness of UViM on three diverse and challenging vision tasks: panoptic segmentation, depth prediction and image colorization, where we achieve competitive and near state-of-the-art results. Our experimental results suggest that UViM is a promising candidate for a unified modeling approach in computer vision.
Key facts. Cited by 77 papers in the map (data through 7 Oct 2026). Builds on 47 of its 62 references that are in the map. First posted on arXiv 20 May 2022.
Builds on
47 of 62 references.
References · 47
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2020End-to-End Object Detection with Transformerscited by 7,384
35 more
- 2017Mask R-CNNcited by 7,259
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2017Neural Discrete Representation Learningcited by 4,650
- 2016Image-to-Image Translation with Conditional Adversarial Networkscited by 4,510
- 2021Zero-Shot Text-to-Image Generationcited by 4,276
- 2014Going Deeper with Convolutionscited by 4,231
- 2013Auto-Encoding Variational Bayescited by 3,980
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,239
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,338
- 2014On the Properties of Neural Machine Translation: Encoder-Decoder Approachescited by 1,303
- 2021Palette: Image-to-Image Diffusion Modelscited by 1,296
- 2014Depth Map Prediction from a Single Image using a Multi-Scale Deep Networkcited by 1,264
- 2021Per-Pixel Classification is Not All You Need for Semantic Segmentationcited by 1,148
- 2016Colorful Image Colorizationcited by 1,113
- 2021Scaling Vision Transformerscited by 993
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
- 2016Conditional Image Generation with PixelCNN Decoderscited by 859
- 2016Pixel Recurrent Neural Networkscited by 814
- 2018Panoptic Segmentationcited by 796
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 656
- 2021Perceiver IO: A General Architecture for Structured Inputs & Outputscited by 616
- 2021How to train your ViT? Data, Augmentation, and Regularization in Vision Transformerscited by 562
- 2017PixelCNN++: Improving the PixelCNN with Discretized Logistic Mixture Likelihood and Other Modificationscited by 415
- 2021Pix2seq: A Language Modeling Framework for Object Detectioncited by 345
- 2021N"UWA: Visual Synthesis Pre-training for Neural visUal World creAtioncited by 253
- 2022BinsFormer: Revisiting Adaptive Bins for Monocular Depth Estimationcited by 166
- 2018High Quality Monocular Depth Estimation via Transfer Learningcited by 133
- 2021Colorization Transformercited by 89
- 2022Transframer: Arbitrary Frame Prediction with Generative Modelscited by 37
- 2017PixColor: Pixel Recursive Colorizationcited by 32
- 2018Detecting Visual Relationships Using Box Attentioncited by 16
- 2016Chained Predictions Using Convolutional Neural Networkscited by 15
- 2017Probabilistic Image Colorizationcited by 13
Built on by
Citing papers per year: 2022: 12; 2023: 24; 2024: 21; 2025: 13; 2026: 7.
77 papers.
Sort:
By area:
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2023Segment Everything Everywhere All at Oncecited by 546
- 2023Finite Scalar Quantization: VQ-VAE Made Simplecited by 425
- 2022Unified-IO: A Unified Model for Vision, Language, and Multi-Modal Taskscited by 420
- 2022VIMA: General Robot Manipulation with Multimodal Promptscited by 403
- 2022Images Speak in Images: A Generalist Painter for In-Context Visual Learningcited by 317
- 2022Generalized Decoding for Pixel, Image, and Languagecited by 292
- 2023Vid2Seq: Large-Scale Pretraining of a Visual Language Model for Dense Video Captioningcited by 263
- 2023Multimodal Foundation Models: From Specialists to General-Purpose Assistantscited by 247
- 2023GPT4Tools: Teaching Large Language Model to Use Tools via Self-instructioncited by 245
- 2023SegGPT: Segmenting Everything In Contextcited by 193
- 2022A Unified Sequence Interface for Vision Taskscited by 152
38 more
- 2022FlexiViT: One Model for All Patch Sizescited by 151
- 20234M: Massively Multimodal Masked Modelingcited by 118
- 2022Vision-Language Pre-training: Basics, Recent Advances, and Future Trendscited by 117
- 2022A Generalist Framework for Panoptic Segmentation of Images and Videoscited by 99
- 2024Toward Generalist Anomaly Detection via In-context Residual Learning with Few-shot Sample Promptscited by 88
- 2023Getting ViT in Shape: Scaling Laws for Compute-Optimal Model Designcited by 87
- 2023GIVT: Generative Infinite-Vocabulary Transformerscited by 80
- 2023UniHCP: A Unified Model for Human-Centric Perceptionscited by 59
- 2022CLIPPO: Image-and-Language Understanding from Pixels Onlycited by 53
- 2023All in Tokens: Unifying Output Space of Visual Tasks via Soft Tokencited by 52
- 20244M-21: An Any-to-Any Vision Model for Tens of Tasks and Modalitiescited by 52
- 2024Restructuring Vector Quantization with the Rotation Trickcited by 49
- 2022Retrospectives on the Embodied AI Workshopcited by 46
- 2024ARC: A Generalist Graph Anomaly Detector with In-Context Learningcited by 46
- 2024JetFormer: An Autoregressive Generative Model of Raw Images and Textcited by 44
- 2023Exploring Effective Factors for Improving Visual In-Context Learningcited by 43
- 2022RUST: Latent Neural Scene Representations from Unposed Imagerycited by 41
- 2023Tuning computer vision models with task rewardscited by 36
- 2023Neural Graph Reasoning: Complex Logical Query Answering Meets Graph Databasescited by 35
- 2023TarViS: A Unified Approach for Target-based Video Segmentationcited by 32
- 2023Generative Semantic Segmentationcited by 32
- 2023InstructCV: Instruction-Tuned Text-to-Image Diffusion Models as Vision Generalistscited by 30
- 2023UniRef++: Segment Every Reference Object in Spatial and Temporal Spacescited by 22
- 2024LaVin-DiT: Large Vision Diffusion Transformercited by 21
- 2022Egocentric Video Task Translationcited by 19
- 2023Selective Visual Representations Improve Convergence and Generalization for Embodied AIcited by 19
- 2024A Simple Image Segmentation Framework via In-Context Examplescited by 19
- 2024A Simple Latent Diffusion Approach for Panoptic Segmentation and Mask Inpaintingcited by 18
- 2025ARIG: Autoregressive Interactive Head Generation for Real-time Conversationscited by 16
- 2023PolyMaX: General Dense Prediction with Mask Transformercited by 14
- 2024Data-efficient Large Vision Models through Sequential Autoregressioncited by 14
- 2026OAT: Ordered Action Tokenizationcited by 14
- 2024Denoising with a Joint-Embedding Predictive Architecturecited by 12
- 2025OmniCast: A Masked Latent Diffusion Model for Weather Forecasting Across Time Scalescited by 11
- 2023A Study of Autoregressive Decoders for Multi-Tasking in Computer Visioncited by 10
- 2025EgoM2P: Egocentric Multimodal Multitask Pretrainingcited by 9
- 2024Chameleon: A Data-Efficient Generalist for Dense Visual Prediction in the Wildcited by 6
- 2023Towards AGI in Computer Vision: Lessons Learned from GPT and Large Language Modelscited by 5
50 most cited of 77 All 77 (JSON)
- 2026AHMAD: Adaptive Hybrid Multi-task Vision Learning with Assisted Distillation for Keypoint Detection28 Sep 2026
- 2026Falcon Perception-HD: High Density Perception via Reinforcement Learning19 Aug 2026
- 2026UniEvo-RS: Omni-Prompt Unified Remote Sensing Segmentation with Representative Exemplar-Driven Prototype Evolution4 Aug 2026
- 2026Ordered Action Tokens for Visuomotor Policy Learning23 Jul 2026
- 2026Hidden-Shot: Towards One-Shot Task Generalization for Low-Level Vision Generalist Models1 Jul 2026
- 2026InCTRLv2: Generalist Residual Models for Few-Shot Anomaly Detection and Segmentation6 Apr 2026
- 2026OAT: Ordered Action Tokenization4 Feb 2026
- 2025T2T-VICL: Cross-Task Visual In-Context Learning via Implicit Text-Driven VLMs20 Nov 2025
- 2025Seg-VAR: Image Segmentation with Visual Autoregressive Modeling16 Nov 2025
- 2025OmniCast: A Masked Latent Diffusion Model for Weather Forecasting Across Time Scales20 Oct 2025
- 2025Product-Quantised Image Representation for High-Quality Image Synthesis3 Oct 2025
- 2025DiVeQ: Differentiable Vector Quantization Using the Reparameterization Trick30 Sep 2025
38 more
- 2025ARIG: Autoregressive Interactive Head Generation for Real-time Conversations1 Jul 2025
- 2025Improving Token-based Object Detection with Video27 Jun 2025
- 2025Vision Generalist Model: A Survey11 Jun 2025
- 2025EgoM2P: Egocentric Multimodal Multitask Pretraining9 Jun 2025
- 2025ALTo: Adaptive-Length Tokenizer for Autoregressive Mask Generation22 May 2025
- 2025UniSTD: Towards Unified Spatio-Temporal Learning across Diverse Disciplines26 Mar 2025
- 2025Quantization-Free Autoregressive Action Transformer18 Mar 2025
- 2025Enhancing Monocular Depth Estimation with Multi-Source Auxiliary Tasks22 Jan 2025
- 2024JetFormer: An Autoregressive Generative Model of Raw Images and Text29 Nov 2024
- 2024LaVin-DiT: Large Vision Diffusion Transformer18 Nov 2024
- 2024VQ-Map: Bird's-Eye-View Map Layout Estimation in Tokenized Discrete Space via Vector Quantization3 Nov 2024
- 2024Diffusion-nested Auto-Regressive Synthesis of Heterogeneous Tabular Data28 Oct 2024
- 2024Restructuring Vector Quantization with the Rotation Trick8 Oct 2024
- 2024A Simple Image Segmentation Framework via In-Context Examples7 Oct 2024
- 2024Denoising with a Joint-Embedding Predictive Architecture2 Oct 2024
- 2024Context is the Key: Backdoor Attacks for In-Context Learning with Vision Transformers6 Sep 2024
- 2024A Simple and Generalist Approach for Panoptic Segmentation29 Aug 2024
- 2024Toward a Diffusion-Based Generalist for Dense Vision Tasks29 Jun 2024
- 2024Autoregressive Image Generation without Vector Quantization17 Jun 2024
- 20244M-21: An Any-to-Any Vision Model for Tens of Tasks and Modalities13 Jun 2024
- 2024Trajectory Forecasting through Low-Rank Adaptation of Discrete Latent Codes31 May 2024
- 2024ARC: A Generalist Graph Anomaly Detector with In-Context Learning27 May 2024
- 2024Chameleon: A Data-Efficient Generalist for Dense Visual Prediction in the Wild29 Apr 2024
- 2024In-Context Translation: Towards Unifying Image Recognition, Processing, and Generation15 Apr 2024
- 2024GLID: Pre-training a Generalist Encoder-Decoder Vision Model11 Apr 2024
- 2024Masked AutoDecoder is Effective Multi-Task Vision Generalist12 Mar 2024
- 2024Toward Generalist Anomaly Detection via In-context Residual Learning with Few-shot Sample Prompts11 Mar 2024
- 2024Data-efficient Large Vision Models through Sequential Autoregression7 Feb 2024
- 2024A Simple Latent Diffusion Approach for Panoptic Segmentation and Mask Inpainting18 Jan 2024
- 2023UniRef++: Segment Every Reference Object in Spatial and Temporal Spaces25 Dec 2023
- 20234M: Massively Multimodal Masked Modeling11 Dec 2023
- 2023GIVT: Generative Infinite-Vocabulary Transformers4 Dec 2023
- 2023PolyMaX: General Dense Prediction with Mask Transformer9 Nov 2023
- 2023Selective Visual Representations Improve Convergence and Generalization for Embodied AI7 Nov 2023
- 2023InstructCV: Instruction-Tuned Text-to-Image Diffusion Models as Vision Generalists30 Sep 2023
- 2023Finite Scalar Quantization: VQ-VAE Made Simple27 Sep 2023
- 2023Multimodal Foundation Models: From Specialists to General-Purpose Assistants18 Sep 2023
- 2023Understanding Deep Neural Networks via Linear Separability of Hidden Layers26 Jul 2023
50 newest of 77 All 77 (JSON)
- 2023InstructCV: Instruction-Tuned Text-to-Image Diffusion Models as Vision Generalistscited by 30
- 2024LaVin-DiT: Large Vision Diffusion Transformercited by 21
- 2024A Simple Latent Diffusion Approach for Panoptic Segmentation and Mask Inpaintingcited by 18
- 2024Denoising with a Joint-Embedding Predictive Architecturecited by 12
- 2025OmniCast: A Masked Latent Diffusion Model for Weather Forecasting Across Time Scalescited by 11
- 2024Toward a Diffusion-Based Generalist for Dense Vision Taskscited by 3
- 2025OmniCast: A Masked Latent Diffusion Model for Weather Forecasting Across Time Scales20 Oct 2025
- 2024LaVin-DiT: Large Vision Diffusion Transformer18 Nov 2024
- 2024Denoising with a Joint-Embedding Predictive Architecture2 Oct 2024
- 2024Toward a Diffusion-Based Generalist for Dense Vision Tasks29 Jun 2024
- 2024A Simple Latent Diffusion Approach for Panoptic Segmentation and Mask Inpainting18 Jan 2024
- 2023InstructCV: Instruction-Tuned Text-to-Image Diffusion Models as Vision Generalists30 Sep 2023
- 2023Vid2Seq: Large-Scale Pretraining of a Visual Language Model for Dense Video Captioningcited by 263
- 2022Vision-Language Pre-training: Basics, Recent Advances, and Future Trendscited by 117
- 2024JetFormer: An Autoregressive Generative Model of Raw Images and Textcited by 44
- 2025ALTo: Adaptive-Length Tokenizer for Autoregressive Mask Generationcited by 5
- 2026UniEvo-RS: Omni-Prompt Unified Remote Sensing Segmentation with Representative Exemplar-Driven Prototype Evolutioncited by 0
- 2026UniEvo-RS: Omni-Prompt Unified Remote Sensing Segmentation with Representative Exemplar-Driven Prototype Evolution4 Aug 2026
- 2025ALTo: Adaptive-Length Tokenizer for Autoregressive Mask Generation22 May 2025
- 2024JetFormer: An Autoregressive Generative Model of Raw Images and Text29 Nov 2024
- 2023Vid2Seq: Large-Scale Pretraining of a Visual Language Model for Dense Video Captioning27 Feb 2023
- 2022Vision-Language Pre-training: Basics, Recent Advances, and Future Trends17 Oct 2022
- 2025Quantization-Free Autoregressive Action Transformercited by 5
- 2025Quantization-Free Autoregressive Action Transformer18 Mar 2025
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2023Segment Everything Everywhere All at Oncecited by 546
- 2023Finite Scalar Quantization: VQ-VAE Made Simplecited by 425
- 2022Unified-IO: A Unified Model for Vision, Language, and Multi-Modal Taskscited by 420
- 2022VIMA: General Robot Manipulation with Multimodal Promptscited by 403
- 2022Images Speak in Images: A Generalist Painter for In-Context Visual Learningcited by 317
- 2022Generalized Decoding for Pixel, Image, and Languagecited by 292
- 2023Multimodal Foundation Models: From Specialists to General-Purpose Assistantscited by 247
- 2023SegGPT: Segmenting Everything In Contextcited by 193
- 2022A Unified Sequence Interface for Vision Taskscited by 152
- 2022FlexiViT: One Model for All Patch Sizescited by 151
- 20234M: Massively Multimodal Masked Modelingcited by 118
12 most cited of 62 All 62 (JSON)
- 2026AHMAD: Adaptive Hybrid Multi-task Vision Learning with Assisted Distillation for Keypoint Detection28 Sep 2026
- 2026Falcon Perception-HD: High Density Perception via Reinforcement Learning19 Aug 2026
- 2026Ordered Action Tokens for Visuomotor Policy Learning23 Jul 2026
- 2026Hidden-Shot: Towards One-Shot Task Generalization for Low-Level Vision Generalist Models1 Jul 2026
- 2026InCTRLv2: Generalist Residual Models for Few-Shot Anomaly Detection and Segmentation6 Apr 2026
- 2026OAT: Ordered Action Tokenization4 Feb 2026
- 2025T2T-VICL: Cross-Task Visual In-Context Learning via Implicit Text-Driven VLMs20 Nov 2025
- 2025Seg-VAR: Image Segmentation with Visual Autoregressive Modeling16 Nov 2025
- 2025Product-Quantised Image Representation for High-Quality Image Synthesis3 Oct 2025
- 2025DiVeQ: Differentiable Vector Quantization Using the Reparameterization Trick30 Sep 2025
- 2025ARIG: Autoregressive Interactive Head Generation for Real-time Conversations1 Jul 2025
- 2025Improving Token-based Object Detection with Video27 Jun 2025
12 newest of 62 All 62 (JSON)