Semi-Supervised Cross-Domain Imitation Learning
This addresses the problem of costly expert data collection in robotics and simulation domains, offering an incremental improvement over existing supervised and unsupervised methods.
The paper tackles cross-domain imitation learning by introducing a semi-supervised setting and algorithm that uses offline data with minimal expert demonstrations, achieving stable and data-efficient policy learning with consistent gains over baselines in experiments on MuJoCo and Robosuite.
Cross-domain imitation learning (CDIL) accelerates policy learning by transferring expert knowledge across domains, which is valuable in applications where the collection of expert data is costly. Existing methods are either supervised, relying on proxy tasks and explicit alignment, or unsupervised, aligning distributions without paired data, but often unstable. We introduce the Semi-Supervised CDIL (SS-CDIL) setting and propose the first algorithm for SS-CDIL with theoretical justification. Our method uses only offline data, including a small number of target expert demonstrations and some unlabeled imperfect trajectories. To handle domain discrepancy, we propose a novel cross-domain loss function for learning inter-domain state-action mappings and design an adaptive weight function to balance the source and target knowledge. Experiments on MuJoCo and Robosuite show consistent gains over the baselines, demonstrating that our approach achieves stable and data-efficient policy learning with minimal supervision. Our code is available at~ https://github.com/NYCU-RL-Bandits-Lab/CDIL.