CoRL 2024 · 机器人操作 · Robotics

Universal Manipulation Interface

在野外用手持夹爪采集演示,无需真实机器人即可训练可部署机器人策略
Cheng Chi, Zhenjia Xu, Chuer Pan, Eric Cousineau, Benjamin Burchfiel, Siyuan Feng, Russ Tedrake, Shuran Song · Stanford / MIT / Toyota Research Institute

UMI(Universal Manipulation Interface)提出了一套"野外人类演示 → 可部署机器人策略"的完整框架:研究者手持轻量级夹爪在任意真实场景采集示范,通过鱼眼相机、视觉惯性 SLAM 和推理时延迟匹配,将人类动作精确转化为机器人可执行的相对轨迹,最终训练出可在多款机器人上零样本泛化的硬件无关策略(hardware-agnostic policy)。

arXiv · 2024-02 CoRL 2024 $371 / 套硬件成本 📄 arXiv:2402.10329 🌐 Project Page
机器人操作 imitation learning in-the-wild demonstration diffusion policy hand-held gripper visual-inertial SLAM latency matching relative trajectory bimanual manipulation hardware-agnostic policy

01 动机

机器人操作策略的训练数据严重不足:远程操控需要昂贵硬件和专家操作员,而直接利用人类在野外的视频又存在巨大的动作域差(embodiment gap)。现有手持夹爪方案往往只能处理简单的拾取任务,无法支持动态投掷、双臂折叠或长时序洗碗等复杂操作。

"We identify four core issues that prevent direct action transfer from human demonstration to robot execution: insufficient visual context, action imprecision, latency discrepancies, and insufficient policy representations for multimodal action distributions."
UMI teaser
图1:UMI 框架总览。研究者手持配备鱼眼相机和侧面反射镜的夹爪,在咖啡馆、户外等真实场景采集演示数据,最终训练出可在 UR5、Franka FR2 等多款机器人上直接部署的策略,涵盖杯子摆放(Cup Arrangement)、动态投掷(Dynamic Tossing)、双臂折衣(Bimanual Cloth Folding)和洗碗(Dish Washing)四项任务。
20/20杯子摆放满分成功率(Full UMI)
87.5%动态投掷成功率(105/120)
70%7步洗碗任务成功率(14/20)
$371单套夹爪总硬件成本

02 方法

UMI 分为两个设计层次:演示接口(Demonstration Interface, HD1–HD6)解决数据采集中的观测与动作精度问题;策略接口(Policy Interface, PD1–PD2)解决训练到推理的延迟和坐标系不一致问题。两者共同支撑以 Diffusion Policy 为骨干的硬件无关策略。

UMI hardware design
图2:手持夹爪硬件设计。左:155° 鱼眼 GoPro + 侧面反射镜 + 柔性手指 + fiducial markers(用于连续夹爪宽度跟踪)。右:夹爪可直接安装到 UR5、Franka 等末端执行器。总重 780g,3D 打印成本 $73,GoPro 套装 $298。

演示接口设计(HD1–HD6)

HD1–HD2 · 腕部鱼眼相机

单相机安装于夹爪腕部,消除外部相机的标定需求,并通过相机运动自然实现数据多样化。采用 155° 鱼眼镜头(raw fisheye,不做矫正),在保留中心分辨率的同时压缩边缘信息,策略学习效果优于等效pinhole模型(消融:去掉鱼眼后杯子任务从100%降至55%)。

HD3 · 侧面反射镜

在夹爪两侧放置物理反射镜,使单张图像中隐含立体视角(implicit stereo)。训练时对镜像内容做数字翻转(digital reflection),获得最优策略效果(消融:去掉镜面翻转后成功率从100%降至85%)。

HD4 · IMU 感知 SLAM

将 GoPro 内置 IMU 与 ORB-SLAM3 结合,实现视觉-惯性 SLAM。即使存在运动模糊也能保持追踪,并恢复具有度量尺度(metric scale)的动作——这对动态投掷等任务至关重要。SLAM 位置精度:6.1mm 平均轨迹误差,旋转精度:3.5°。

HD5–HD6 · 连续夹爪控制 + 运动学过滤

通过 fiducial markers 连续跟踪夹爪宽度(而非二值开/闭),实现精确时序控制和隐式力控(soft-finger deformation)。训练时通过运动学可行性过滤,将机器人无关的演示数据适配到具体机器人。

策略接口设计(PD1–PD2)

Policy interface: latency matching and relative trajectory
图5:策略接口设计。左:推理时延迟匹配——分别测量 RGB、本体感知、夹爪各路信号延迟,推理时将所有流对齐至延迟最高的相机;动作延迟通过提前发送指令补偿。右:相对轨迹动作表示——动作为相对初始末端执行器位姿的 SE(3) 变换,本体感知表示为相对轨迹(等效速度),双臂相对位姿通过 map-then-localize SLAM 计算。

策略骨干网络采用 Diffusion Policy,专为建模多峰动作分布设计,适应野外数据中自然存在的多模态行为。

03 实验

实验分为两类:窄域能力验证(单场景,含完整消融)和野外泛化(30个多样场景,15种杯子)。机器人平台:UR5 + Franka FR2;演示采集:每任务250–305条;评估:每条件20次(投掷120次)。

窄域(Narrow-Domain)能力评估

任务演示数量Full UMI关键消融对比
Cup Arrangement(杯子摆放) 305 20/20 (100%) 无鱼眼: 55% · delta动作: 80% · 绝对动作: 25%
Dynamic Tossing(动态投掷) 280 105/120 (87.5%) 无延迟匹配: 57.5%
Bimanual Cloth Folding(双臂折衣) 250 14/20 (70%) 无双臂相对位姿: 30%
Dish Washing(7步洗碗) 258 14/20 (70%) ResNet-34(无CLIP): 0%
Narrow-domain evaluation results
图8:窄域评估结果。每列显示初始状态叠加图、典型失败案例和成功率条形图,同时对比各消融条件(无鱼眼、无延迟匹配、无双臂相对位姿等)。数字均来自论文原文,Full UMI 在四项任务上均显著优于所有消融变体。

野外泛化(In-the-Wild Generalization)

以 Cup Arrangement 为泛化测试任务,在 30 个多样场景(家居、办公室、餐厅、户外)共采集 1,400 条演示,涵盖 15 种不同杯子,采集时间约 12 人时。

测试环境已见杯子未见杯子合并
咖啡桌(Cafe table,未见场景) 28/40 (70%) 15/20 (75%) 43/60 (71.7%)
饮水机旁(Water fountain,分布外) 包含在合并结果中 71.7%
窄域对照基线(同预训练权重) 0%(在新环境完全失败)
In-the-wild generalization results
图9:野外泛化结果。左侧展示在咖啡馆、户外等多样场景中的成功执行;右侧对比在未见场景中 UMI 野外策略(~70–75%)与窄域基线(0%)的成功率。

数据采集效率 与 SLAM 精度

跨机器人迁移

UR5 上采集的杯子摆放演示直接迁移至 Franka FR2,成功率 18/20 (90%),验证了硬件无关策略的零样本跨平台迁移能力。

04 局限性

注:以下局限性均来自论文作者明确陈述(stated),最后一条为从系统设计推断(inferred)。
SLAM 依赖环境纹理

ORB-SLAM3 在纹理匮乏的场景(如纯白墙面、低光照环境)下追踪失败,导致无法恢复度量尺度动作。作者明确指出这是当前方案的主要瓶颈之一。

训练时需要运动学可行性过滤

采集的人类演示数据需要在训练阶段针对具体机器人执行运动学可行性过滤(kinematic feasibility filtering),系统尚未实现端到端的体态感知学习(embodiment-aware learning)。

手持夹爪的自由度和速度受限

780g 的夹爪重量和两指结构限制了采集效率(仅达人手速度 48%)及可表达的操作多样性,与人手相比在灵巧度上有明显差距。

[推断] 策略泛化依赖演示数据多样性

野外泛化实验表明,策略的跨场景迁移能力直接依赖于演示数据覆盖的场景和物体多样性;当测试场景与演示分布差异过大时,成功率会显著下降(inferred from 实验设计)。