I3L: Iterative Interactive Imitation Learning for Whole-Body Mobile Manipulation

I3L overview: operator takes over with JoyLo+, corrections are aggregated for retraining, and representative tasks and embodiments.
As the policy nears a failure, the operator grasps JoyLo+ and corrects the arms, torso, and base together.
Corrections are aggregated with prior demonstrations, the policy is retrained, and the new policy is redeployed.
Three robots across simulated and real-world tasks.

I3L couples interactive correction with iterative policy refinement.

Abstract

Policies trained offline from teleoperated demonstrations can degrade outside their training distribution. This gap is particularly acute in long-horizon whole-body mobile manipulation, where long horizons amplify compounding errors and coordinated arm, torso, and base motion makes exhaustive demonstration coverage impractical. We introduce I3L, an integrated hardware and learning system that couples targeted whole-body correction with iterative policy refinement to address this last-mile reliability gap. Its low-cost, open-source JoyLo+ interface uses policy-tracking leader arms and touch-triggered takeover to enable coordinated corrections across the arms, torso, and mobile base, while repeated deployment and retraining allow operators to target failures of the updated policy. We evaluate I3L across three robot embodiments and eight simulated and real-world tasks, including four long-horizon whole-body mobile manipulation tasks. Controlled simulation comparisons show that four-round HG-DAgger achieves the highest final success on both simulation tasks. Under this protocol, I3L improves autonomous success over the base policies by an average of 68.3 percentage points and achieves at least 90% success on every task, outperforming demonstration-only training by 35.6 percentage points on average under matched training-trajectory budgets. Across four rounds, task-averaged takeover duration per episode decreases by 72%. Together, these results show that I3L helps close the last-mile gap between offline training and reliable deployment and provides practical insights into collecting and using corrections for long-horizon whole-body mobile manipulation.

Walkthrough Video

JoyLo+: A Hardware Interface for Policy Intervention

Explore the device

Explore the leader arms in 3D. The hardware overview is available below.

R1 Pro JoyLo+

Explore both seven-joint leaders. Select an arm to move its joints independently.

Motorized links track the policy. Grasping the touch-sensitive grip lets the operator take over.

JoyLo+ correction interface in single-arm (Franka) and dual-arm (R1 Pro) configurations.

JoyLo+ correction interface in single-arm (Franka) and dual-arm (R1 Pro) configurations.

JoyLo+ builds on GELLO's kinematic leaders and JoyLo's actuated interface to let an operator step in during autonomous execution.

  • Leaders track the policy. Motorized leaders mirror the policy's arm targets during autonomy and become gravity-compensated for hand guidance during a correction.
  • Grasp to take over. Capacitive electrodes in the grips detect touch, so the operator takes control simply by holding the device.
  • Whole-body takeover. On R1 Pro, one takeover hands over both arms, torso, base, and grippers; releasing returns control to the policy, which replans from the corrected state.

Bill of Materials

Component Unit cost Single-arm quantity
Franka R1 Pro
XM430-W350-T servo$22034
XC330-M288-T servo$5043
U2D2 USB adapter$3022
Power supply (12 V / 5 V)$2522
3D-printed links (set)
STL downloads: Franka ZIP · R1 Pro ZIP
Assembly models: Franka · R1 Pro
$1511
Joy-Con pair (L+R)$8011
Adafruit MPR121 touch sensor$7.9511
LILYGO T-Display S3$9.5411
Conductive copper tape$511
Estimated total (USD)$1,087.49$1,257.49

Single-arm JoyLo+ cost estimates (USD). Each column covers one seven-joint leader and the listed accessories. Robot, tax, shipping, cabling, fasteners, bearings, and mounting hardware are excluded. Vendor prices may differ. Power-supply and copper-tape links are sourcing references.

Iterative Interactive Imitation Learning

Starting from a demonstration dataset D0, each round deploys the previous policy and collects a labeled rollout dataset through JoyLo+. Under HG-DAgger filtering, only the original demonstrations and the active intervention frames from all rounds are retained for training. Pre-intervention and on-policy frames are excluded to concentrate training on human demonstrations and corrective supervision.

Each round trains from the same starting weights rather than continuing from the previous round's policy: the same pretrained checkpoint for the π0.5-based VLA, and the same initialization for diffusion policies. As earlier failures become less frequent, corrections naturally shift toward downstream stages such as transport and placement, producing an emergent failure-mode curriculum.

Tasks and Embodiments

Eight evaluation tasks across A1, Franka, and R1 Pro in simulation and the real world.
Prepare Make-Ahead Breakfast Bowls
Boxing Books up for Storage
Hang Mug on Rack
Pick and Place Strawberry
Make Microwave Popcorn
Turning on Radio
Insert Peg
Open Fridge and Pick Up Radio

Evaluation tasks. Initial configurations and successful outcomes; yellow mats mark where the R1 Pro base and task objects are randomized.

Eight tasks on three robots: two controlled BEHAVIOR simulation tasks for comparing collection and learning strategies, two contact-rich precision tasks on Franka, and four long-horizon whole-body mobile manipulation tasks on a real R1 Pro.

Playback speed
96% success · 24/25
Prepare Make-Ahead Breakfast Bowls
92% success · 23/25
Boxing Books up for Storage
100% success · 25/25
Make Microwave Popcorn
96% success · 24/25
Turning on Radio
92% success · 23/25
Mug Hanging
92% success · 23/25
Peg Insertion
92.8% success
Pick-and-Place Strawberries
90.0% success
Open Fridge and Pick Up Radio

Targeted Corrections Beat Additional Demonstrations

Task Base 100 demos 100 corrections (ours)
Simulation
Pick-and-Place Strawberries53.2%55.6%92.8%
Open Fridge and Pick Up Radio39.6%30.4%90.0%
Real world (Franka)
Mug Hanging6/25 (24%)17/25 (68%)23/25 (92%)
Peg Insertion7/25 (28%)12/25 (48%)23/25 (92%)
Real world (R1 Pro)
Prepare Make-Ahead Breakfast Bowls1/25 (4%)19/25 (76%)24/25 (96%)
Boxing Books up for Storage3/25 (12%)13/25 (52%)23/25 (92%)
Make Microwave Popcorn5/25 (20%)20/25 (80%)25/25 (100%)
Turning on Radio6/25 (24%)14/25 (56%)24/25 (96%)

Autonomous full-task success. Each base policy receives either 100 additional demonstrations or 100 corrective trajectories collected over four HG-DAgger rounds.

With the same trajectory budget, targeted corrections outperform additional demonstrations on every task and reach at least 90% autonomous success on all eight.

Iteration Reduces Intervention Burden

Intervention burden and task completion across correction rounds.

Intervention burden across correction rounds. Dots are tasks and the dark line is the task mean. In (d), rounds show takeover-free collection episodes and Final shows autonomous evaluation success.

As the policy is retrained, operators step in less often, for less of each episode, and for less time, with all three measures declining every round.

Remaining bottlenecks are task dependent

Task-stage intervention patterns across correction rounds.
Prepare Make-Ahead Breakfast Bowls
Boxing Books up for Storage
Hang Mug on Rack
Insert Peg
Make Microwave Popcorn
Turning on Radio
Pick and Place Strawberry
Open Fridge and Pick Up Radio

Intervention rate (%) over episode progress (%)

  • Round 1
  • Round 2
  • Round 3
  • Round 4

Red arrows: corrected failures. Green dashed arrows: successful behavior.

Where corrections happen. Intervention rate over episode progress for Rounds 1–4 (lighter is earlier). Red arrows mark corrected failures; green arrows mark successful behavior.

As early-stage failures are fixed, the remaining corrections concentrate on each task's later stages, forming an emergent curriculum over failure modes.

Example Takeover Episodes

Example episodes with human takeover and correction. The policy drives the robot until the operator takes over through JoyLo+. The top-left badge, the robot's floor disc, and the timeline show what is controlling the robot at every frame. The JoyLo+ inset mirrors the recorded robot-arm poses, with an orange border during takeover. Click or drag the timeline to seek.

Head camera Left wrist Right wrist
3D pose · drag to orbit
0:00 / 0:00
Subtask
Control

Corrections Span the Whole Body

The leaders drive both arms while the JoyCon drives the torso and base, so a single takeover can coordinate the whole robot.

Body-part activation shares during JoyLo+ corrections.

Body-part activation shares during JoyLo+ corrections, pooled over four rounds. Each part gets one vote per takeover segment once active for three consecutive frames.

Operators use this: on every task, a large share of corrections move the base rather than only the arms.

Operator Study

Three operators with different teleoperation experience each run three correction rounds of 25 episodes on Franka Mug Hanging, each refining their own policy.

Operator study on Franka mug hanging across three teleoperators.

Operator study on Franka mug hanging. (a) Cumulative fraction of timesteps under human control. (b) Cumulative takeover-free episode counts.

Intervention burden falls and takeover-free episodes accumulate for every operator, so the gains from I3L do not depend on a single expert.

Which Correction-Learning Strategy?

Setting Residual BC (all data) SIRIUS HG-DAgger
Pick-and-Place Strawberries
Single round (100)45.214.876.089.2
Round 1 (25)49.20.470.875.6
Round 2 (50)45.61.278.888.4
Round 3 (75)40.04.437.688.4
Round 4 (100)46.810.878.892.8
Open Fridge and Pick Up Radio
Single round (100)21.679.273.280.0
Round 1 (25)16.070.854.851.6
Round 2 (50)24.086.468.472.4
Round 3 (75)30.087.270.482.0
Round 4 (100)19.688.071.690.0

Correction-learning comparison in simulation. Full-task success (%) over five evaluation seeds; parentheses give cumulative corrective trajectories.

Four-round HG-DAgger finishes best on both tasks, but more rounds alone do not guarantee a better policy: how corrections are learned from matters as much as collecting them.