Source-linked AI summary

Multiple Interactions Made Easy (MIME): Large Scale Demonstrations Data for Imitation

Pratyusha Sharma, Lekha Mohan, Lerrel Pinto, Abhinav Gupta

arXiv:1810.07121v1cs.ROcs.CVcs.LG

TL;DR

Robotics datasets often isolate single tasks, while complex manipulation calls for diverse demonstration data. MIME collects paired human videos and robot trajectories across 20 tasks and demonstrates visual imitation by mapping third-person video features to robot trajectories. The dataset contains 8,260 human-object interactions and corresponding robot trajectories.

  • Problem

    Existing robotics datasets often focus on isolated tasks, while diverse manipulation data is needed to study tasks ranging from grasping and pushing to complex stacking.

    Source: §1 · p. 1
  • Method

    MIME collects paired human videos and kinesthetic robot trajectories across diverse tasks, then studies mapping human-video features to robot joint-angle trajectories.

    Source: §1 · p. 2
  • Results

    MIME contains 8,260 human-object interactions and 8,260 robot trajectories across 20 diverse tasks, and the paper demonstrates its use for visual imitation.

    Source: §5 · p. 8
  • Takeaways & Limitations

    MIME provides a dataset for visual imitation research that maps third-person video features to robot trajectories.

    Source: §5 · p. 8

    Each demonstration may admit multiple valid imitation trajectories, making direct regression to joint angles problematic and motivating a classification-based output.

    Source: §4.2 · p. 6

Abstract

from arXiv · show

In recent years, we have seen an emergence of data-driven approaches in robotics. However, most existing efforts and datasets are either in simulation or focus on a single task in isolation such as grasping, pushing or poking. In order to make progress and capture the space of manipulation, we would need to collect a large-scale dataset of diverse tasks such as pouring, opening bottles, stacking objects etc. But how does one collect such a dataset? In this paper, we present the largest available robotic-demonstration dataset (MIME) that contains 8260 human-robot demonstrations over 20 different robotic tasks (https://sites.google.com/view/mimedataset). These tasks range from the simple task of pushing objects to the difficult task of stacking household objects. Our dataset consists of videos of human demonstrations and kinesthetic trajectories of robot demonstrations. We also propose to use this dataset for the task of mapping 3rd person video features to robot trajectories. Furthermore, we present two different approaches using this dataset and evaluate the predicted robot trajectories against ground-truth trajectories. We hope our dataset inspires research in multiple areas including visual imitation, trajectory prediction, and multi-task robotic learning.

1 Introduction

The introduction argues that robotics needs diverse, large-scale demonstration data beyond datasets centered on isolated tasks, particularly for complex manipulation. MIME addresses this need with paired human videos and robot trajectories, and the paper uses it to study mapping visual demonstrations to robot motion.

1 Introduction

  • Existing robotics datasets largely isolate tasks such as grasping, pushing, poking, or knot-tying, motivating broader task and object diversity.The authors argue diverse data could support visual representations transferable across tasks.Source: §1 · p. 1
  • Complex manipulation such as stacking is presented as requiring supervised demonstrations, since random-exploration self-supervision is unlikely to succeed.The passage ends mid-sentence after contrasting demonstrations with self-supervision.Source: §1 · p. 1
  • MIME contains 8,260 human-robot demonstrations across more than 20 tasks, spanning simple pushing to complex household-object stacking.The paper frames this as a first step toward scaling demonstration data for complex manipulation.Source: §1 · p. 2
  • For each kinesthetic robot demonstration, MIME also records a video of the same human demonstrator performing the task by hand.Kinesthetic demonstrations let people express desired motion while staying within the robot's constraints.Source: §1 · p. 2
  • Multiple demonstrators reduce dependence on one person's peculiarities and permit parallel collection, while training and cross-checking address the difficulty newcomers face in kinesthetic operation.The paper also proposes mapping human-video features to robot joint angles, evaluating predictions against the nearest of multiple test trajectories.Source: §1 · p. 2

2 Related Work

The related work contrasts single-task robotics datasets and simulator data with the need for real-world, multi-task demonstrations. It positions MIME alongside work scaling Learning from Demonstrations while emphasizing broader task diversity and physical-action data.

2.1 Data Driven Robotics

  • Prior real-world robotics datasets span grasping, pushing, rope manipulation, surgery, and door opening, but typically focus on single, often simple tasks.The authors contrast this with MIME's roughly 20 tasks and its intended use for imitation learning across tasks.Source: §2.1 · p. 3
  • Simulators can supply data faster than real-world collection, but the reality gap makes transfer challenging; MIME instead provides real-world data.The passage cites simulator examples across grasping, driving, and indoor navigation.Source: §2.1 · p. 3

2.2 Learning From Demonstrations (LfD)

  • Learning from Demonstrations can support complex policies, but prior methods often use only a handful of expert demonstrations for one task.The passage connects demonstrations with reduced exploration and guidance during policy learning.Source: §2.2 · p. 3
  • A prior effort collected hundreds of demonstrations across 15 tasks through virtual-reality teleoperation, but its data was not public and was used task-specifically.The authors note that kinesthetic robot movement keeps demonstrations in suitable configuration spaces for whole-arm manipulation.Source: §2.2 · p. 3
  • Large visual datasets lack physical actions, limiting transfer to new tasks; MIME pairs rich visual information with demonstrations across 20 tasks.The authors present this combination as relevant to multi-task learning from demonstrations.Source: §2.2 · p. 3

3 The MIME Dataset

MIME's dataset design addresses task selection, robot sensing, and the scaling and verification of kinesthetic collection. The resulting dataset pairs human and robot demonstrations across 20 tasks and includes variation among objects and task setups.

3 The MIME Dataset

  • The dataset section frames task vocabulary, scaling kinesthetic collection, and correcting human errors as its three collection challenges.These challenges organize the account of dataset construction.Source: §3 · p. 3

3.1 Vocabulary of Tasks

  • Tasks were selected to be feasible for Baxter, not require haptic feedback, and provide enough diversity for learning task embeddings.Demonstrations also vary objects; the pouring examples vary object shape and opacity, arrangement, and the number of pour targets.Source: §3.1 · p. 4, Figure 3 · p. 5

3.2 Robot Setup

  • A Baxter robot in gravity-compensated kinesthetic mode records synchronized RGBD images, joint angles, and gripper positions from a head camera and two wrist cameras.The head camera observes the table externally, while wrist cameras move with the arms.Source: §3.2 · p. 4

3.3 Data Collection Procedure

  • Collection uses multiple demonstrators, with training, step-by-step instructions, and participant review; rejected demonstrations are checked and redone.Participants give repeated demonstrations using varied objects, and each task is collected by multiple demonstrators.Source: §3.3 · p. 4
  • The collection produced 8,260 paired human and robot demonstrations across 20 discrete tasks, with incorrect demonstrations removed after participant review.The paper reports that newer participants made most errors, with fewer errors and faster trajectories after practice.Source: §3.3 · p. 5

4 Experiments

The experiments evaluate task recognition from joint angles and behavioural cloning from human video to robot trajectories. They use an 80/10/10 train-validation-test split and multiple test trajectories to reflect multimodal outcomes.

4 Experiments

  • The dataset is split 80%/10%/10% across training, validation, and test sets, with multiple test trajectories collected for each test demonstration.The same task-category distribution is maintained across splits, and multiple ground truths support trajectory-distance evaluation.Source: §4 · p. 5

4.1 Task Recognition

  • DTW task-recognition accuracy rises from 58.9% with 10% of the data to 79.7% with the full dataset.DTW classifies each test trajectory using the nearest training trajectory in time-series distance.Source: §4.1 · p. 5
  • The task-recognition LSTM processes joint angles sequentially and predicts a 20-class task distribution at each timestep.A cross-entropy loss is accumulated across timesteps and minimized with Adam.Source: §4.1 · p. 6
  • The joint-angle LSTM achieves 61.2% held-out accuracy, with errors often between tasks having similar or overlapping trajectories.Examples include single-hand picking confused with stacking, and placing in a box confused with picking.Source: §4.1 · p. 6, Figure 4 · p. 6

4.2 Behavioural Cloning

  • The output space uses 600 joint-angle clusters, with denser clustering during the manipulation portion of each trajectory.The first 25 timesteps have 150 clusters and the second half has 450, to capture finer motion near objects.Source: §4.2 · p. 7
  • Behavioural cloning maps human-video features and the robot’s current object-state image to a predicted robot joint-angle trajectory.The LSTM is initialized from VGG features of the robot image and processes human-video features sequentially, predicting joint clusters.Source: §4.2 · p. 7, Figure 5 · p. 7
  • The method addresses trajectory multimodality with classification-based outputs and evaluation against multiple ground-truth trajectories.Using two test ground truths reduced MSE from 0.1296 to 0.1076; the reported metric uses the lower error against either trajectory.Source: §4.2 · p. 8

4.3 Comparing baselines

  • Behaviour cloning improves as training data increases, outperforming the average-trajectory and k-NN baselines with a steeper improvement.Figure 6 reports MSE variation with data amount; its error bars show ±1 standard deviation across random splits.Source: Figure 6 · p. 8, §4.3 · p. 8

5 Conclusion

The paper presents MIME, a large human and robot demonstration dataset spanning diverse tasks, and demonstrates its use for visual imitation.

  • MIME contains 8,260 human-object interactions and 8,260 corresponding robot trajectories across 20 diverse tasks, and supports mapping third-person video features to robot trajectories.Source: §5 · p. 8
Loading 1810.07121v1…