SynIL: Leveraging Synergy for Offline Imitation Learning from Imperfect Demonstration Datasets
Authors: Yuto Tanaka, Kyo Kutsuzawa, Martina Doku, Dai Owaki, Mitsuhiro Hayashibe
Organizations: Department of Robotics, Graduate School of Engineering, Tohoku University, 6-6-01 Aoba, Aramaki, Aoba-ku, Sendai, 980-8579, Miyagi, Japan
Imitation learning enables robots to acquire complex skills directly from massive demonstration datasets, but its performance degrades severely when datasets are contaminated with suboptimal or noisy demonstrations. While prior quality-assessment methods attempt to filter or reweight data, they typically rely on manual pre-selection of expert reference data or task-specific heuristics, limiting scalability. To address this challenge, we introduce SynIL (Synergy-based Imitation Learning), a novel framework for automated, label-free demonstration quality assessment in offline reinforcement learning. Grounded in neuroscientific evidence that motor synergy, a low-dimensional coordinated structure in movement, correlates directly with motor proficiency, SynIL algorithmically quantifies synergy manifestation to generate dense, transition-level reward signals via self-supervised reward regression. Comprehensive evaluations on D4RL locomotion benchmarks and multi-human Robomimic manipulation datasets demonstrate that synergy-derived rewards correlate strongly with ground-truth rewards. Furthermore, SynIL substantially outperforms Behavior Cloning (BC) and achieves performance comparable to, and in sparse-reward human teleoperation scenarios, superior to, offline reinforcement learning trained on true environment rewards.
Figures & tables
Figure 1: Overview of SynIL. Synergy scores ξ are calculated from trajectories of a dataset D (top-right), and a reward model is trained based on the synergy scores (bottom-right). Offline reinforcement learning is finally performed with rewards computed by the trained reward model (bottom-left).
Hyperparameter
Value in D4RL
Value in Robomimic
RewardNet
Hidden layers
2
2
Hidden Size
512
1024
Activation
ReLU
ReLU
Last Activation
Linear
Linear
Epochs
500
10000
Batch Size
64
64
Table 1: Hyperparameters for SSRR
Hyperparameter
SynIL and IQL
BC
Hidden size
256
256
Hidden layers
2
2
Activation
ReLU
ReLU
Optimizer
Adam
Adam
Epochs
106
106
Evaluate period
5000
5000
Table 2: Hyperparameters for imitation learning
Figure 2: Comparison of synergy-based/true rewards for different datasets in D4RL. The vertical axes correspond to the cumulative rewards over 200 steps estimated by SSRR, and the horizontal axes correspond to the true cumulative rewards over 200 steps.
Dataset
Correlation coefficient
halfcheetah-medium-expert-v2
0.9959
walker2d-medium-expert-v2
0.8655
hopper-medium-expert-v2
0.8718
halfcheetah-medium-v2
0.9482
walker2d-medium-v2
0.5809
hopper-medium-v2
0.9526
Table 3: Correlation coefficients between the synergy-based and true rewards in D4RL
Dataset
BC
SynIL (Ours)
IQL (ground-truth reward)
halfcheetah-medium-v2
42.7±0.3
45.4±0.2
47.4±0.2
hopper-medium-v2
54.4±1.7
46.6±3.8
54.4±3.5
walker2d-medium-v2
71.7±6.3
72.0±5.4
80.0±4.9
halfcheetah-medium-replay-v2
37.1±2.4
40.6±1.4
44.3±0.5
hopper-medium-replay-v2
14.4±2.4
99.9±0.5
66.5±8.1
walker2d-medium-replay-v2
15.5±6.4
74.5±3.0
69.5±7.9
Table 4: Performance of each algorithm in D4RL (Average sum of rewards over 10 episodes)
Department of Computer Science, Tufts University, Medford, MA, USA · Complex Dynamical Systems, Austrian Institute of Technology GmbH (AIT), Vienna, Austria