KlaskTron-IL: A Contact-Rich Adversarial Imitation Learning Benchmark with a Validated Sim-to-Real Bridge
Organizations: ETH Zürich, IDSC
Abstract
KlaskTron is an open-hardware physical platform for two-player adversarial reinforcement learning, with a learned actuator model that supports zero-shot sim-to-real transfer of trained policies. Building on this platform, we contribute KlaskTron-IL: a multi-tier imitation learning dataset (80k human demonstration timesteps collected directly on the physical robot, 320k symmetry-augmented timesteps, and synthetic expert trajectories up to 5M steps generated by the zero-shot-validated PPO policy), and diagnostic baselines that expose three characteristic failure modes of standard IL methods under contact-rich adversarial dynamics: BC's data scaling plateau, GAIL's interpretable but weak-gradient discriminator (Spearman peg-ball proximity recovery, but a logit span under 0.1), and an asymmetric BC-prior x reward-density interaction. Because the policy interface used throughout this paper is the same one the actuator model bridges to hardware, a policy trained on this benchmark can run unchanged on the physical robot, as we demonstrate zero-shot for the strongest BC baseline, so long as its commands respect the actuator model's m/s identification range.
Figures & tables
| Tier | Source | Timesteps | Episodes | Avg. Ep. Len. | Origin |
|---|---|---|---|---|---|
| 1 | Human (on robot) | 80,591 | 111 | 727 | Lab tournament |
| 2 | Augmented | 322,364 | 444 | 727 | Symmetry 4 |
| 3 | Synthetic (HW-validated policy) | 5,000,000 | 34,649 | 144 | PPO self-play |
| Method | Goals/Ep. | Player-in-Goal | HW-compat. | Diagnostic Failure Mode |
|---|---|---|---|---|
| BC (human, 80k) | 0.102 | 0.157 | ✓ ∗ | Data-scaling plateau: covariate-shift floor |
| GAIL | 0.116 | 0.201 | Interpretable but flat discriminator | |
| BC+RL (sparse) | 0.138 | 0.130 | Asymmetric prior: stabilizes sparse RL | |
| BC+RL (dense) | 0.128 | 0.067 | Asymmetric prior: no detectable effect under dense shaping |
| Metric | BC BC | GAIL GAIL | GAIL BC |
|---|---|---|---|
| Goals ratio | 1.053 0.278 | 1.137 0.043 | 1.357 0.154 |
| Goals scored/ep | 0.102 0.013 | 0.116 0.008 | 0.144 0.007 |
| Timeout rate | 0.472 0.024 | 0.309 0.005 | 0.379 0.023 |
| Player-in-Goal | 0.157 0.006 | 0.201 0.007 | 0.197 0.016 |
| Ep. length | 146.8 4.6 | 103.6 1.7 | 128.0 6.8 |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Reward term | Weight |
|---|---|
| goal_scored | |
| goal_conceded | |
| player_in_goal | |
| opponent_in_goal | |
| distance_ball_opponent_goal | |
| collision_player_ball |
Robot Imitation Learning
Robot imitation learning trains robots to perform tasks by reproducing and adapting behaviors shown in demonstrations.
See all 703 papers