Timezone: »
We present the task of "Social Rearrangement", consisting of cooperative everyday tasks like setting up the dinner table, tidying a house or unpacking groceries in a simulated multi-agent environment. In Social Rearrangement, two robots coordinate to complete a long-horizon task, using onboard sensing and egocentric observations, and no privileged information about the environment. We study zero-shot coordination (ZSC) in this task, where an agent collaborates with a new partner, emulating a scenario where a robot collaborates with a new human partner. Prior ZSC approaches struggle to generalize in our complex and visually rich setting, and on further analysis, we find that they fail to generate diverse coordination behaviors at training time. To counter this, we propose Behavior Diversity Play (BDP), a novel ZSC approach that encourages diversity through a discriminability objective. Our results demonstrate that BDP learns adaptive agents that can tackle visual coordination, and zero-shot generalize to new partners in unseen environments, achieving 35% higher success and 32% higher efficiency compared to baselines.
Author Information
Andrew Szot (Georgia Institute of Technology)
Unnat Jain (UIUC)
Dhruv Batra (Georgia Institute of Technology / Facebook AI Research)
Zsolt Kira (Georgia Institute of Technology)
Ruta Desai (Meta Reality Labs Research)
Akshara Rai (Facebook AI Research)
More from the Same Authors
-
2021 Poster: Cooperative Exploration for Multi-Agent Deep Reinforcement Learning »
Iou-Jen Liu · Unnat Jain · Raymond Yeh · Alex Schwing -
2021 Oral: Cooperative Exploration for Multi-Agent Deep Reinforcement Learning »
Iou-Jen Liu · Unnat Jain · Raymond Yeh · Alex Schwing -
2019 Poster: Probabilistic Neural Symbolic Models for Interpretable Visual Question Answering »
Shanmukha Ramakrishna Vedantam · Karan Desai · Stefan Lee · Marcus Rohrbach · Dhruv Batra · Devi Parikh -
2019 Poster: TarMAC: Targeted Multi-Agent Communication »
Abhishek Das · Theophile Gervet · Joshua Romoff · Dhruv Batra · Devi Parikh · Michael Rabbat · Joelle Pineau -
2019 Poster: Trainable Decoding of Sets of Sequences for Neural Sequence Models »
Ashwin Kalyan · Peter Anderson · Stefan Lee · Dhruv Batra -
2019 Oral: TarMAC: Targeted Multi-Agent Communication »
Abhishek Das · Theophile Gervet · Joshua Romoff · Dhruv Batra · Devi Parikh · Michael Rabbat · Joelle Pineau -
2019 Oral: Probabilistic Neural Symbolic Models for Interpretable Visual Question Answering »
Shanmukha Ramakrishna Vedantam · Karan Desai · Stefan Lee · Marcus Rohrbach · Dhruv Batra · Devi Parikh -
2019 Oral: Trainable Decoding of Sets of Sequences for Neural Sequence Models »
Ashwin Kalyan · Peter Anderson · Stefan Lee · Dhruv Batra -
2019 Poster: Counterfactual Visual Explanations »
Yash Goyal · Ziyan Wu · Jan Ernst · Dhruv Batra · Devi Parikh · Stefan Lee -
2019 Oral: Counterfactual Visual Explanations »
Yash Goyal · Ziyan Wu · Jan Ernst · Dhruv Batra · Devi Parikh · Stefan Lee -
2018 Poster: Learn from Your Neighbor: Learning Multi-modal Mappings from Sparse Annotations »
Ashwin Kalyan · Stefan Lee · Anitha Kannan · Dhruv Batra -
2018 Oral: Learn from Your Neighbor: Learning Multi-modal Mappings from Sparse Annotations »
Ashwin Kalyan · Stefan Lee · Anitha Kannan · Dhruv Batra