A Mixed Diet Makes DINO An Omnivorous Vision Encoder
About
Pre-trained vision encoders like DINOv2 have demonstrated exceptional performance on unimodal tasks. However, we observe that their features are poorly aligned across different visual modalities. For instance, the feature embedding for an RGB image and its corresponding depth map of the same scene exhibit a cosine similarity that is nearly identical to that of two random, unrelated images. To address this, we propose the Omnivorous Vision Encoder, a post-training framework that learns a modality-agnostic feature space. We fine-tune the encoder with a dual objective: first, to maximize the feature alignment between different modalities of the same scene; and second, a distillation objective that anchors the learned representations to a fully frozen teacher. The resulting student encoder becomes "omnivorous" by producing more consistent embeddings for a given scene, regardless of the input modality (RGB, Depth, Segmentation, etc.). This approach enables robust cross-modal understanding while retaining the discriminative semantics of the original foundation model. Omnivorous model weights are available at https://github.com/google-deepmind/representations4d.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | Cityscapes | mIoU73.2 | 526 | |
| Semantic segmentation | Pascal VOC | mIoU0.857 | 295 | |
| Monocular Depth Estimation | NYU V2 | Delta 1 Acc94.8 | 192 | |
| Surface Normal Estimation | NYU V2 | Mean Angular Error14.1 | 96 | |
| Classification | ImageNet-1k (val) | Top-1 Acc83.8 | 41 | |
| kNN search accuracy | iNaturalist | -- | 5 | |
| Cross-modal retrieval | MOVi N=128 (test) | R@186.2 | 4 | |
| Cross-modal retrieval | ScanNet N=3072 (test) | R@146.1 | 4 | |
| Cross-modal retrieval | TartanAir N=128 (test) | R@190.6 | 4 | |
| Monocular Depth Prediction | navi probe3d | Depth Delta178.1 | 4 |