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)
In-context imitation learning (ICIL) enables robots to learn new tasks from a small number of demonstrations by conditioning a pre-trained policy on task-specific examples, without retraining at test time. Despite this promise, training generalizable and scalable in-context imitation policies remains an open challenge. We present SynthICL, a scalable framework that trains ICIL policies entirely from RGB-only synthetic data. Specifically, we build a data generation pipeline to produce high-fidelity ICIL data and train a flow-matching transformer policy on the resulting dataset. SynthICL avoids the need for depth sensing, precise camera calibration, and real-world training data in prior approaches, offering a simpler and more scalable alternative. We further incorporate subgoal prediction by training the model to predict the next subgoal images, enabling more precise and visually grounded control. Evaluated on 16 unseen real-world manipulation tasks, SynthICL achieves an average success rate of 79% with only one demonstration provided at test time and outperforms prior methods. Project page: https://synth-icl.github.io
Enabling robots to learn long-horizon manipulation tasks from a handful of demonstrations remains a central challenge in robotics. Existing neuro-symbolic approaches often rely on hand-crafted symbolic abstractions, semantically labeled trajectories or large demonstration datasets, limiting their scalability and real-world applicability. We present a scalable neuro-symbolic framework that autonomously constructs symbolic planning domains and data-efficient control policies from as few as one to thirty unannotated skill demonstrations, without requiring manual domain engineering. Our method segments demonstrations into skills and employs a Vision-Language Model (VLM) to classify skills and identify equivalent high-level states, enabling automatic construction of a state-transition graph. This graph is processed by an Answer Set Programming solver to synthesize a PDDL planning domain, which an oracle function exploits to isolate the minimal, task-relevant and target relative observation and action spaces for each skill policy. Policies are learned at the control reference level rather than at the raw actuator signal level, yielding a smoother and less noisy learning target. Known controllers can be leveraged for real-world data augmentation by projecting a single demonstration onto other objects in the scene, simultaneously enriching the graph construction process and the dataset for imitation learning. We validate our framework primarily on a real industrial forklift across statistically rigorous manipulation trials, and demonstrate cross-platform generality on a Kinova Gen3 robotic arm across two standard benchmarks. Our results show that grounding control learning, VLM-driven abstraction, and automated planning synthesis into a unified pipeline constitutes a practical path toward scalable, data-efficient, expert-free and interpretable neuro-symbolic robotics.
Pierrick Lorang, Johannes Huemer, Timothy Duggan +3
Department of Computer Science, Tufts University, Medford, MA, USA · Complex Dynamical Systems, Austrian Institute of Technology GmbH (AIT), Vienna, Austria
Imitation learning offers a promising framework for enabling robots to acquire diverse skills from human users. However, most imitation learning algorithms assume access to high-quality demonstrations an unrealistic expectation when collecting data from non-expert users, whose demonstrations often contain inadvertent errors. Naively learning from such demonstrations can result in unsafe policy behavior, while discarding entire demonstrations due to occasional mistakes wastes valuable data, especially in low-data settings. In this work, we introduce GiB (Good-in-Bad), an algorithm that automatically identifies and discards erroneous subtasks within demonstrations while preserving high-quality subtasks. The filtered data can then be used by any policy learning algorithm to train more robust policies. GiB first trains a self-supervised model to learn latent features and assigns binary weights to label each demonstration as good or bad. It then models the latent feature distribution of high-quality segments and uses the Mahalanobis distance to detect and evaluate poor-quality subtasks. We validate GiB on the Franka robot in both simulated and real-world multi-step tasks, demonstrating improved policy performance when learning from mixed-quality human demonstrations.
Noushad Sojib, Ola Ghattas, Momotaz Begum
Department of Computer Science, University of New Hampshire, USA