Where are we in the search for an Artificial Visual Cortex for Embodied Intelligence?

By Arjun Majumdar, Karmesh Yadav, Sergio Arnaud, Yecheng Jason Ma, C. Chen, S. Silwal, Aryan Jain, Vincent-Pierre Berges, Pieter Abbeel, Jitendra Malik and 5 more. Neural Information Processing Systems, 2023

303 citations recorded by the source corpus.

We present the largest and most comprehensive empirical study of pre-trained visual representations (PVRs) or visual 'foundation models' for Embodied AI. First, we curate CortexBench, consisting of 17 different tasks spanning locomotion, navigation, dexterous, and mobile manipulation. Next, we systematically evaluate existing PVRs and find that none are universally dominant. To study the effect of pre-training data size and diversity, we combine over 4,000 hours of egocentric videos from 7 different sources (over 4.3M images) and ImageNet to train different-sized vision transformers using Masked Auto-Encoding (MAE) on slices of this data. Contrary to inferences from prior work, we find that scaling dataset size and diversity does not improve performance universally (but does so on average). Our largest model, named VC-1, outperforms all prior PVRs on average but does not universally dominate either. Next, we show that task- or domain-specific adaptation of VC-1 leads to substantial gains, with VC-1 (adapted) achieving competitive or superior performance than the best known results on all of the benchmarks in CortexBench. Finally, we present real-world hardware experiments, in which VC-1 and VC-1 (adapted) outperform the strongest pre-existing PVR. Overall, this paper presents no new techniques but a rigorous systematic evaluation, a broad set of findings about PVRs (that in some cases, refute those made in narrow domains in prior work), and open-sourced code and models (that required over 10,000 GPU-hours to train) for the benefit of the research community.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

View Paper: https://www.semanticscholar.org/paper/326f6a8011e43322c433751b9cc31fd56564621c

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Sergey Levine
  2. Chelsea Finn
  3. Pieter Abbeel
  4. Tianhe Yu
  5. Li Fei-Fei
  6. Quoc V. Le
  7. Igor Mordatch
  8. Trevor Darrell

In Computer Vision and Pattern Recognition

  1. Sergey Levine
  2. Luc Van Gool
  3. Chelsea Finn
  4. Li Fei-Fei
  5. Dacheng Tao
  6. Trevor Darrell
  7. Andrew Zisserman
  8. Jitendra Malik

Citations in the corpus