ProcTHOR: Large-Scale Embodied AI Using Procedural Generation

By Matt Deitke, Eli VanderBilt, Alvaro Herrasti, Luca Weihs, Jordi Salvador, Kiana Ehsani, Winson Han, Eric Kolve, Ali Farhadi, Aniruddha Kembhavi and 1 more. Advances in Neural Information Processing Systems 35, 2022

516 citations recorded by the source corpus.

Massive datasets and high-capacity models have driven many recent advancements in computer vision and natural language understanding. This work presents a platform to enable similar success stories in Embodied AI. We propose ProcTHOR, a framework for procedural generation of Embodied AI environments. ProcTHOR enables us to sample arbitrarily large datasets of diverse, interactive, customizable, and performant virtual environments to train and evaluate embodied agents across navigation, interaction, and manipulation tasks. We demonstrate the power and potential of ProcTHOR via a sample of 10,000 generated houses and a simple neural model. Models trained using only RGB images on ProcTHOR, with no explicit mapping and no human task supervision produce state-of-the-art results across 6 embodied AI benchmarks for navigation, rearrangement, and arm manipulation, including the presently running Habitat 2022, AI2-THOR Rearrangement 2022, and RoboTHOR challenges. We also demonstrate strong 0-shot results on these benchmarks, via pre-training on ProcTHOR with no fine-tuning on the downstream benchmark, often beating previous state-of-the-art systems that access the downstream training data.

Subfields: Computer Vision and Pattern Recognition, Control and Systems Engineering

View Paper: https://www.semanticscholar.org/paper/d7bca917b753b996074203cc77ba45935b7c62d7

How researchers see this paper

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

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Li Fei-Fei
  3. Trevor Darrell
  4. Jitendra Malik
  5. Abhinav Kumar Gupta
  6. Dacheng Tao
  7. Kristen Grauman
  8. Xiaogang Wang

In Control and Systems Engineering

  1. Li Fei-Fei
  2. Yuke Zhu
  3. Sergey Levine
  4. Jitendra Malik
  5. Josh Tenenbaum
  6. Pieter Abbeel
  7. Ali Farhadi
  8. Aravind Rajeswaran

Citations in the corpus