Generative adversarial imitation learning for robot swarms: Learning from human demonstrations and trained policies
Authors: Mattes Kraus, Jonas Kuckling
Organizations: University of Konstanz, Department of Computer and Information Science, Konstanz, Germany · Cluster of Excellence “Centre for the Advanced Study of Collective Behavior”, University of Konstanz · Zukunftskolleg, University of Konstanz
In imitation learning, robots are supposed to learn from demonstrations of the desired behavior. Most of the work in imitation learning for swarm robotics provides the demonstrations as rollouts of an existing policy. In this work, we provide a framework based on generative adversarial imitation learning that aims to learn collective behaviors from human demonstrations. Our framework is evaluated across six different missions, learning both from manual demonstrations and demonstrations derived from a PPO-trained policy. Results show that the imitation learning process is able to learn qualitatively meaningful behaviors that perform similarly well as the provided demonstrations. Additionally, we deploy the learned policies on a swarm of TurtleBot 4 robots in real-robot experiments. The exhibited behaviors preserved their visually recognizable character and their performance is comparable to the one achieved in simulation.
Figures & tables
Sensor
Range
Dim.
Speed ∣v∣
[0,0.31m]m
1
LiDAR
[0,200]cm
5
Color
{0;1}
3
Bumper
{0;1}
1
Actuator
Range
Dim.
Linear vel. v
[0,0.31]m
1
TABLE I : The simplified model for the TurtleBot 4 robot.
Fig. 1 : Two screen captures from our demonstration tool. The user can build an experimental arena and provide demonstrations of swarm behaviors.
Fig. 2 : Information flow in our SwarmGAIL implementation. The discriminator receives swarm-level features as observation. The policy is used in a round-robin fashion to provide actions to all robots based on their local observations.
Fig. 3 : Representations of the demonstrated behavior for all considered missions. In the case of Full Speed and Controlled Speed , the trajectories taken by the robots look visually similar, but the velocity of the robots differs.
Fig. 4 : Violin plots of the return (cumulative reward) of all evaluations across all considered missions. Colors indicate the source of the demonstrations (blue from human-operated, brown from PPO-trained ones). Dark colors represent the initial policies of each imitation learning experiment, light colors the final policy. Black circles correspond to one evaluation of one policy/demonstration. The higher the return, the better the performance.
Fig. 5 : Violin plots of the returns (cumulative reward) of imitated policies in simulation (light colored) and reality (dark colored). Colors indicate the source of the demonstrations (blue from human-operated , brown from PPO-trained ones). Black circles correspond to one evaluation of one policy/demonstration. The higher the return the better the performance.
With the increasing use of robot-free demonstration interfaces that provide state trajectories without action labels, imitation from observation has become a promising approach for learning robot behaviors from human demonstrations. However, due to differences in embodiment and dynamics between humans and robots, demonstrated human motions may not be feasible for the robot, potentially degrading policy performance. In this study, we propose Experience-Based Feasibility-Aware Generative Adversarial Imitation from Observation (EF-GAIfO), which estimates the feasibility of state-only demonstrations from the robot's own experience rather than relying on explicit dynamics models or large prior exploration datasets. A key feature of EF-GAIfO is that the notion of feasibility evolves with policy learning: as the policy improves and the robot experiences a broader range of state transitions, the feasible region is progressively expanded, allowing additional demonstrations to be incorporated into learning. This enables feasibility-aware imitation that adapts to the current stage of policy learning, rather than relying on a pre-designed feasibility criterion. We validate the effectiveness of EF-GAIfO on a locomotion task in simulation and on a real quadruped robot performing a object-reaching-and-grasping task.
Yoshiki Takebayashi, Giovanni Perantoni, Hikaru Sasaki +2
Division of Information Science, Graduate School of Information Science, Nara Institute of Science and Technology (NAIST), Nara, Japan · Department of Industrial Engineering of the University of Trento, Trento, Italy
Imitation learning frameworks that learn robot control policies from demonstrators' motions via hand-mounted demonstration interfaces have attracted increasing attention. However, due to differences in physical characteristics between demonstrators and robots, this approach faces two limitations: i) the demonstration data do not include robot actions, and ii) the demonstrated motions may be infeasible for robots. These limitations make policy learning difficult. To address them, we propose Feasibility-Aware Behavior Cloning from Observation (FABCO). FABCO integrates behavior cloning from observation, which complements robot actions using robot dynamics models, with feasibility estimation. In feasibility estimation, the demonstrated motions are evaluated using a robot-dynamics model, learned from the robot's execution data, to assess reproducibility under the robot's dynamics. The estimated feasibility is used for multimodal feedback and feasibility-aware policy learning to improve the demonstrator's motions and learn robust policies. Multimodal feedback provides feasibility through the demonstrator's visual and haptic senses to promote feasible demonstrated motions. Feasibility-aware policy learning reduces the influence of demonstrated motions that are infeasible for robots, enabling the learning of policies that robots can execute stably. We conducted experiments with 15 participants on two tasks and confirmed that FABCO improves imitation learning performance by more than 3.2 times compared to the case without feasibility feedback.
Kei Takahashi, Hikaru Sasaki, Takamitsu Matsubara
Division of Information Science, Graduate School of Science and Technology, Nara Institute of Science and Technology (NAIST), Ikoma-shi, Nara 630-0192, Japan
Learning robot policies from demonstrations is often limited by the cost of collecting diverse, high-quality expert data. While an appealing alternative is to expand a single successful demonstration into a diverse set of successful trajectories, existing augmentation methods typically rely on manually designed transformation rules or complex trajectory planning, requiring substantial task-specific engineering and human effort. We introduce OneDemo, a reinforcement learning framework that learns to edit a single successful demonstration for new task configurations. We formulate demonstration editing as a structured reinforcement learning problem, in which an editor learns, from sparse task-success rewards, both when to transition between object-centric trajectory segments and how to geometrically adapt each segment. The learned editor serves both as a task policy and as a reusable demonstration generator. Across 15 Meta-World tasks, OneDemo reaches a 92.7% success rate within 53.7k environment interactions on average. Across 9 DexMimicGen tasks, it generates successful demonstrations with a 98.4% average success rate, and BC-RNN policies trained on the generated data achieve 99.6% average success at matched dataset scale. These results show that learning to edit a single successful demonstration provides a structured and effective basis for both policy learning and scalable demonstration generation.
Ying Qian, Jianshu Hu, Han Fang +2
Stanford University · Shanghai Jiao Tong University · Tsinghua University