Co-GAIL: Learning Diverse Strategies for Human-Robot Collaboration
Not on the map. Why?
Abstract (arXiv)
We present a method for learning a human-robot collaboration policy from human-human collaboration demonstrations. An effective robot assistant must learn to handle diverse human behaviors shown in the demonstrations and be robust when the humans adjust their strategies during online task execution. Our method co-optimizes a human policy and a robot policy in an interactive learning process: the human policy learns to generate diverse and plausible collaborative behaviors from demonstrations while the robot policy learns to assist by estimating the unobserved latent strategy of its human collaborator. Across a 2D strategy game, a human-robot handover task, and a multi-step collaborative manipulation task, our method outperforms the alternatives in both simulated evaluations and when executing the tasks with a real human operator in-the-loop. Supplementary materials and videos at https://sites.google.com/view/co-gail-web/home
Key facts. Cited by 16 papers in the map (data through 7 Oct 2026). Builds on 34 of its 54 references that are in the map. First posted on arXiv 13 Aug 2021.
Builds on
34 of 54 references.
References · 34
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2016Least Squares Generative Adversarial Networkscited by 1,000
- 2016InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Netscited by 858
- 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 734
- 2018QT-Opt: Scalable Deep Reinforcement Learning for Vision-Based Robotic Manipulationcited by 730
- 2019Solving Rubik's Cube with a Robot Handcited by 699
- 2017Sim-to-Real Transfer of Robotic Control with Dynamics Randomizationcited by 616
- 2019MediaPipe: A Framework for Building Perception Pipelinescited by 572
- 2017Deep Imitation Learning for Complex Manipulation Tasks from Virtual Reality Teleoperationcited by 286
- 2017A Unified Game-Theoretic Approach to Multiagent Reinforcement Learningcited by 265
- 2016Variational Intrinsic Controlcited by 262
22 more
- 2017A Multimodal Anomaly Detector for Robot-Assisted Feeding Using an LSTM-based Variational Autoencodercited by 209
- 2017Multi-agent Reinforcement Learning in Sequential Social Dilemmascited by 206
- 2018RoboTurk: A Crowdsourcing Platform for Robotic Skill Learning through Imitationcited by 202
- 2015Multiagent Cooperation and Competition with Deep Reinforcement Learningcited by 180
- 2015Variational Information Maximisation for Intrinsically Motivated Reinforcement Learningcited by 142
- 2018Variational Option Discovery Algorithmscited by 121
- 2018Multi-Agent Generative Adversarial Imitation Learningcited by 99
- 2020Learning Latent Representations to Influence Multi-Agent Interactioncited by 92
- 2017Robust Imitation of Diverse Behaviorscited by 85
- 20196-PACK: Category-level 6D Pose Tracker with Anchor-Based Keypointscited by 78
- 2018Modeling Others using Oneself in Multi-Agent Reinforcement Learningcited by 64
- 2020A Cordial Sync: Going Beyond Marginal Policies for Multi-Agent Embodied Taskscited by 52
- 2019Efficient Bimanual Manipulation Using Learned Task Schemascited by 44
- 2017A survey of robot learning from demonstrations for Human-Robot Collaborationcited by 38
- 2014Efficient Model Learning for Human-Robot Collaborative Taskscited by 35
- 2020Learning Multi-Arm Manipulation Through Collaborative Teleoperationcited by 31
- 2020Reactive Human-to-Robot Handovers of Arbitrary Objectscited by 25
- 2014Multi-agent Inverse Reinforcement Learning for Two-person Zero-sum Gamescited by 17
- 2020Multi-Agent Collaboration via Reward Attribution Decompositioncited by 17
- 2018Towards Robotic Feeding: Role of Haptics in Fork-based Food Manipulationcited by 16
- 2019Learning to Collaborate from Simulation for Robot-Assisted Dressingcited by 16
- 2016VIME: Variational Information Maximizing Explorationcited by 15
Built on by
Citing papers per year: 2022: 1; 2023: 6; 2024: 4; 2025: 4; 2026: 1.
16 papers.
Sort:
By area:
- 2023Motion Planning for Autonomous Driving: The State of the Art and Future Perspectivescited by 112
- 2024COMBO: Compositional World Models for Embodied Multi-Agent Cooperationcited by 47
- 2023Human-in-the-Loop Task and Motion Planning for Imitation Learningcited by 28
- 2024Learning to Cooperate with Humans using Generative Agentscited by 23
- 2023Adaptive Coordination in Social Embodied Rearrangementcited by 14
- 2022Heterogeneous Multi-agent Zero-Shot Coordination by Coevolutioncited by 12
- 2023Automated Task-Time Interventions to Improve Teamwork using Imitation Learningcited by 7
- 2025RHINO: Learning Real-Time Humanoid-Human-Object Interaction from Human Demonstrationscited by 4
- 2024MoVEInt: Mixture of Variational Experts for Learning Human-Robot Interactions from Demonstrationscited by 3
- 2025Autonomous Human-Robot Interaction via Operator Imitationcited by 3
- 2025Adaptively Coordinating with Novel Partners via Learned Latent Strategiescited by 3
- 2023Multi-Agent Strategy Explanations for Human-Robot Collaborationcited by 2
4 more
- 2023Learning Multimodal Latent Dynamics for Human-Robot Interactioncited by 1
- 2025Casper: Inferring Diverse Intents for Assistive Teleoperation with Vision Language Modelscited by 1
- 2024Open Human-Robot Collaboration using Decentralized Inverse Reinforcement Learningcited by 0
- 2026OmniRobotHome: A Multi-Camera Home Platform for Real-Time Human-Robot Interactioncited by 0
- 2026OmniRobotHome: A Multi-Camera Home Platform for Real-Time Human-Robot Interaction30 Apr 2026
- 2025Adaptively Coordinating with Novel Partners via Learned Latent Strategies16 Nov 2025
- 2025Casper: Inferring Diverse Intents for Assistive Teleoperation with Vision Language Models17 Jun 2025
- 2025Autonomous Human-Robot Interaction via Operator Imitation3 Apr 2025
- 2025RHINO: Learning Real-Time Humanoid-Human-Object Interaction from Human Demonstrations18 Feb 2025
- 2024Learning to Cooperate with Humans using Generative Agents21 Nov 2024
- 2024Open Human-Robot Collaboration using Decentralized Inverse Reinforcement Learning2 Oct 2024
- 2024MoVEInt: Mixture of Variational Experts for Learning Human-Robot Interactions from Demonstrations10 Jul 2024
- 2024COMBO: Compositional World Models for Embodied Multi-Agent Cooperation16 Apr 2024
- 2023Learning Multimodal Latent Dynamics for Human-Robot Interaction27 Nov 2023
- 2023Multi-Agent Strategy Explanations for Human-Robot Collaboration20 Nov 2023
- 2023Human-in-the-Loop Task and Motion Planning for Imitation Learning24 Oct 2023
4 more
- 2023Adaptive Coordination in Social Embodied Rearrangement31 May 2023
- 2023Motion Planning for Autonomous Driving: The State of the Art and Future Perspectives17 Mar 2023
- 2023Automated Task-Time Interventions to Improve Teamwork using Imitation Learning1 Mar 2023
- 2022Heterogeneous Multi-agent Zero-Shot Coordination by Coevolution9 Aug 2022
- 2024Learning to Cooperate with Humans using Generative Agents21 Nov 2024
- 2023Multi-Agent Strategy Explanations for Human-Robot Collaboration20 Nov 2023
- 2023Adaptive Coordination in Social Embodied Rearrangementcited by 14
- 2022Heterogeneous Multi-agent Zero-Shot Coordination by Coevolutioncited by 12
- 2025RHINO: Learning Real-Time Humanoid-Human-Object Interaction from Human Demonstrationscited by 4
- 2024MoVEInt: Mixture of Variational Experts for Learning Human-Robot Interactions from Demonstrationscited by 3
- 2025Autonomous Human-Robot Interaction via Operator Imitationcited by 3
- 2025Adaptively Coordinating with Novel Partners via Learned Latent Strategiescited by 3
- 2023Learning Multimodal Latent Dynamics for Human-Robot Interactioncited by 1
- 2026OmniRobotHome: A Multi-Camera Home Platform for Real-Time Human-Robot Interactioncited by 0
- 2026OmniRobotHome: A Multi-Camera Home Platform for Real-Time Human-Robot Interaction30 Apr 2026
- 2025Adaptively Coordinating with Novel Partners via Learned Latent Strategies16 Nov 2025
- 2025Autonomous Human-Robot Interaction via Operator Imitation3 Apr 2025
- 2025RHINO: Learning Real-Time Humanoid-Human-Object Interaction from Human Demonstrations18 Feb 2025
- 2024MoVEInt: Mixture of Variational Experts for Learning Human-Robot Interactions from Demonstrations10 Jul 2024
- 2023Learning Multimodal Latent Dynamics for Human-Robot Interaction27 Nov 2023
- 2023Adaptive Coordination in Social Embodied Rearrangement31 May 2023
- 2022Heterogeneous Multi-agent Zero-Shot Coordination by Coevolution9 Aug 2022