← 返回 2026-09-04

RoboTok:面向人类示范检索与灵巧操作学习的互联网规模数据引擎 RoboTok: An Internet-Scale Data Engine for Human Demonstration Retrieval and Dexterous Manipulation Learning

Howard Qian, Yiting Chen, Yunfei Xie, Kejia Ren, Podshara Chanrungmaneekul, Gaotian Wang, Bowen Wen, Chen Wei, Kaiyu Hang 📅 2026-09-02 👍 11 2026-11-03 18:30
DTW 度量学习 强化学习 数据引擎 机器人学习 灵巧操作 示范检索

用躯干坐标系下的3D手部轨迹嵌入,从网络视频中检索人类示范指导灵巧手训练。

前置知识

动态时间规整(Dynamic Time Warping, DTW)

一种序列对齐算法,用于比较长度不同、速度不均的两条时间序列。它通过动态规划寻找最优对齐路径 $\pi$,最小化累计距离 $\mathrm{DTW}(x_i,x_j)=\min_{\pi\in\Pi}\sum_{(t,u)\in\pi}\|x_t^i-x_u^j\|_2^2$,允许序列局部拉伸或压缩。本文对齐21关节的3D手部轨迹,并用长度归一化的负对齐代价 $s(i,j)=-\mathrm{DTW}(x_i,x_j)/(2(L_i+L_j))$ 作为动作相似度的伪真值。

RoboTok 的全部训练信号与检索评价指标(DTW cost、相关集合定义)都源自这个 DTW oracle。不懂 DTW 的对齐机制,就无法理解嵌入空间为何能近似动作相似度、以及正负样本是如何划定的。

度量学习与超球面嵌入

度量学习训练编码器 $\Gamma:\mathcal{X}\to\mathbb{S}^{d-1}$,把输入映射到单位超球面,使行为相近的样本内积(余弦相似度)更大。本文要求内积保持 DTW 诱导的排序,即 $s(i,j)>s(i,k)\Rightarrow\langle\Gamma(x_i),\Gamma(x_j)\rangle>\langle\Gamma(x_i),\Gamma(x_k)\rangle$,从而把昂贵的逐对 DTW 比较替换为一次前向编码加向量近邻检索。

理解集合损失与排序损失为何如此设计、为什么新视频只需一次前向即可入索引并支撑互联网规模持续扩展,都必须先掌握'内积保持排序'这一核心目标。

3D手部姿态估计工具链(WiLoR / MoGe-2 / HaWoR)

WiLoR 是端到端3D手部关键点估计模型(21关节、弱透视相机假设);MoGe-2 是单目几何估计模型,输出带度量尺度的深度,用于把 WiLoR 输出变换到度量相机坐标系;HaWoR 做世界系手部运动重建,用来补全遮挡导致的缺失帧。三者串联构成从普通网络视频到度量3D手部轨迹的完整提取管线。

数据引擎的每一步输入输出都建立在这三个现成模型的能力边界上(如弱透视与度量尺度的差异)。理解各自分工,才能看清管线中为什么需要'度量接地'和'补全'这两个环节。

检索评价指标(mAP / nDCG / Recall@k / CKNNA / Kendall τ)

mAP@K 衡量相关邻居是否整体排在前列;nDCG@K 引入位置折损的增益;Recall@K 是命中率;CKNNA 度量两个表征空间局部邻域结构的一致性;Kendall τ 是秩相关系数,检验检索结果内部排序与真值的一致性;MRR@K 是首个相关结果的倒数排名均值。本文以 DTW top-K 邻居为相关集合(k=20 或 5)。

表2/表3 的全部结论都用这些指标表达。理解它们才能正确解读 mAP@20=0.3531 对 STRAP 的 0.0071 究竟意味着什么量级的差距,以及 Recall@20=0.996 的实际含义。

PPO 与势函数奖励塑形(potential-based reward shaping)

PPO 是常用的策略梯度强化学习算法。势函数塑形在奖励中加入 $w(\gamma\Phi(s')-\Phi(s))$,理论上不改变最优策略但加速探索。本文定义 $\Phi(s)$ 为当前手部状态到检索示范库(重定向后的状态集合)的负加权 k-NN 距离,另加一个 $\beta\Phi(s)$ 的示范追踪奖励;策略没有任何动作标签,检索视频只通过该机制进入训练。

第5.2节下游实验完全靠这个机制把'检索质量'转化为'任务成功率'。理解 $\Phi$ 的定义才能明白为什么检索结果的好坏会直接决定 PPO 策略的成败,以及为什么方法间差异可完全归因于检索。

VTDexManip 灵巧操作仿真基准

一个基于仿真的灵巧手操作基准,含瓶盖拧转、水龙头旋拧、杠杆滑动、桌面重定向、手内重定向、双手传递六类任务,提供 seen/unseen 物体划分。原版使用稠密奖励并限制手腕位姿;本文将其改造为更难的版本:恢复完整3D手部控制并移除人工设计的稠密奖励,仅保留任务奖励加示范引导。

表4/表5 的成功率全部来自该基准。理解原版与困难版的区别,才能看懂'原版已饱和、困难版拉开数量级差距'这一实验设计的意图与结论。

研究动机

机器人学习严重依赖大规模多样化示范,但机器人数据采集昂贵且难以覆盖真实任务、物体、环境的长尾。Open X-Embodiment、BridgeData V2、DROID 等大型数据集即使投入巨大,也只能覆盖真实操作行为的一小部分。互联网视频(如 HowTo100M、Action100M)包含海量且持续增长的人类操作行为,是天然的数据源,但存在两个根本障碍:其一,网络视频的相机视角、场景外观、演员遮挡完全不受控;其二,更本质的是视觉或语义相似并不等于操作行为相似——画面相似的视频可能在执行完全不同的手部动作,而画面迥异的两段视频可能在执行同一种操作。现有检索方法各有局限:FlowRetrieval(CoRL 2024)用光流、STRAP(ICLR 2025)对齐视觉基础模型特征、HAND(ICRA 2026)匹配2D手部/末端路径,但它们要么受限于固定机器人语料库,要么使用图像空间运动表征,要么依赖语义相似度,无法在异构网络视频中按真实操作行为进行检索。

本文的目标是本文要构建一个互联网规模的数据引擎 RoboTok:给定一段人类操作查询视频,从持续索引的网络视频集合中自动检索出手部运动行为最相似的示范片段,为下游灵巧手(人形机器人/拟人灵巧手)策略训练提供基于轨迹的监督数据。具体目标有三:(1) 学到一个对相机视角、场景外观、演员遮挡不变、仅由3D手部轨迹决定的行为嵌入空间;(2) 用离线编码加内积索引的方式,把需要与全库逐对比较的 DTW 计算从查询期挪到训练期,使新片段只需一次前向传播即可入索引,支持持续扩展;(3) 用检索质量指标和下游强化学习任务成功率双重验证:检索到的人类视频确实能提升灵巧操作学习,从而把网络视频变成可扩展、持续增长的机器人监督来源。

与已有工作不同的是,本文的独特切入点是把'操作行为'锚定在以演员躯干为原点的自我中心参考系下的3D手部轨迹上,而非相机坐标或语义标签。核心洞察是:灵巧手动作只有相对演员身体表达才有意义,而躯干坐标系可以仅凭腕部轨迹本身估计出来——即使人体完全不可见也成立,这使比较天然免疫相机视角与遮挡,恰好解决演示视频中最常见的手部特写问题。与最近邻工作相比:HAND 限于自采机器人数据并用2D路径;STRAP 对齐视觉特征而非手部几何;FlowRetrieval 依赖光流这一图像空间信号;SiMDex 混合语义与运动相似度;RfV 用语言查询。RoboTok 则是首个专门面向灵巧操作、以示范视频本身为查询、直接在互联网人类视频上做非参数运动检索的框架,并创新性地用经典 DTW 作为免动作标注的运动相似性监督来源,把'定义什么算相似'从语义判断变成几何计算。

核心方法

直觉上,判断两段视频是否在做同一种操作,应该看手相对身体怎么动,而不是看画面像不像。RoboTok 据此设计两阶段系统:离线持续索引阶段与在线查询检索阶段。离线阶段处理 Action100M 网络视频:自动筛选出4-8秒、近静止相机、手部轻度可见的片段,用 WiLoR 以5fps提取21关节3D手部关键点,因 WiLoR 基于弱透视模型,再用 MoGe-2 估计度量深度把手部姿态变换到度量相机坐标系,缺失帧由 HaWoR 补全;随后轻量躯干估计器仅凭腕部轨迹预测静态躯干坐标系,把轨迹正则化到自我中心参考系,并以长度归一化负 DTW 代价 $s(i,j)=-\mathrm{DTW}(x_i,x_j)/(2(L_i+L_j))$ 计算相似度作监督。在线阶段,轻量交叉注意力编码器把每条轨迹一次性编码为 $\ell_2$ 归一化的超球面嵌入 $\Gamma:\mathcal{X}\to\mathbb{S}^{d-1}$ 存入内积索引;查询视频经同一编码器映射后按余弦相似度取 top-K 近邻,全程无需对全库跑 DTW。

核心创新有三层。第一,行为表征层:用估计的躯干中心参考系表达3D手部轨迹,把'什么算同一种操作'从外观问题转化为几何问题——由于输入轨迹已被正则化并显式编码3D运动,编码器无需再学习外观/视角不变性,只需忠实保持 DTW 诱导的邻域结构。第二,监督层:DTW 虽然在互联网规模上不可行(查询需与全库逐对比较),但可以离线算好并充当相似性 oracle,把检索质量直接变成训练目标,完全不需要动作标签或语义标注——这是与语义检索(RfV、R+X)的本质区别。第三,训练策略层:在 N=100,000 的语料上随机批次(b=196)几乎不可能含 DTW 相似样本,因此采用锚点中心分组采样:每个锚点取 DTW top-K=20 为相关集 $R(a)$,采2个正样本加1个恰在相关集边界外的硬负样本组成 $G(a)$,每批49组共196条,配合集合损失加排序损失 $\mathcal{L}=\mathcal{L}_{set}+\lambda\mathcal{L}_{rank}$,分别控制'谁进入邻域'与'邻域内如何排序'。新片段无需重训、单次前向即可入索引,实现数据引擎的持续扩展。

方法步骤详情

完整流程分四步。第一步数据筛选:从 Action100M 已预筛的人类动作片段中,自动保留4-8秒、近静止相机(Lucas-Kanade 光流判定)、每片段最多一左一右两只手且轻度可见的片段,重叠片段按更长者优先贪心去重。第二步轨迹提取与度量接地:以5fps用 WiLoR 估计3D手部关键点并跨帧链接以稳定左右手身份;用 MoGe-2 估计度量深度,把手部姿态从弱透视坐标变换到度量相机坐标系;缺失帧用 HaWoR 补全,得到变长的相机系21关节轨迹。第三步自我中心正则化与监督生成:按 Wang et al. (2026) 流程适配 SMPL-H 训练轻量躯干帧估计器(输入仅为腕部轨迹),预测静态躯干坐标系并变换轨迹;随后成对计算 DTW 相似度作为伪真值 oracle。第四步训练与检索:编码器对每帧输入做位置编码,经轻量交叉注意力网络池化为 $d$ 维 $\ell_2$ 归一化嵌入;按锚点分组构造批次(b=196),以 $\mathcal{L}=\mathcal{L}_{set}+\lambda\mathcal{L}_{rank}$ 训练;推理时全部轨迹离线编码进内积索引,查询时在线做余弦近邻检索,训练语料为10万条片段、其中1万条 held-out 作评测查询。

技术新颖性

技术新颖性体现在四方面。(1) 表征层面:首次把'演员相对躯干系的3D手部轨迹'作为网络视频操作检索的统一表征,且躯干帧只从腕部轨迹估计、不要求人体可见,这解决了手部特写视频的遮挡难题,是与 HAND(2D路径)和 STRAP(视觉特征对齐)的本质区别。(2) 监督层面:用 DTW 这一经典序列对齐算法充当免标注相似性 oracle,再蒸馏进嵌入空间,把 $O(N)$ 次序列对齐的推理开销转化为训练期开销,是'离线算力换在线可扩展性'的干净设计。(3) 优化层面:边界负样本加集合/排序双损失的分组批次针对 DTW 排序的局部性定制——只显式优化 k=20 局部邻域,实验(图5)却显示嵌入在 k=1 到 500 的全局尺度上自发组织出正确结构。(4) 系统层面:对比表1,此前没有方法同时满足'人类视频查询+互联网数据源+演员相对表征+3D手部姿态'四项;FlowRetrieval、HAND、STRAP 均依赖固定机器人语料或图像空间运动,RoboTok 是首个面向灵巧手的互联网规模示范检索数据引擎,并在原版 VTDexManip 上以检索示范引导的 PPO 平均超越基准最优预训练基线 7.45%(seen)。

RoboTok is an internet-scale data engine that indexes human demonstration clips from the web by their 3D egocentric hand trajectories...
Figure 1: RoboTok is an internet-scale data engine that indexes human demonstration clips from the web by their 3D egocentric hand trajectories...
Internet videos are filtered to retain clips with visible hands and near-static cameras. Hand trajectories are extracted from each clip and reconstructed in metric 3D...
Figure 2: Internet videos are filtered to retain clips with visible hands and near-static cameras. Hand trajectories are extracted from each clip and reconstructed in metric 3D...
Every egocentric trajectory is encoded once into the RoboTok learned motion embedding space...
Figure 3: Every egocentric trajectory is encoded once into the RoboTok learned motion embedding space...

实验结果

检索质量(表2,10万片段库、1万条 held-out 查询、k=20):RoboTok 达 mAP@20=0.3531、nDCG@20=0.5836、MRR@20=0.8576、Kendall τ=0.4867,Recall@20=0.996(几乎每个查询必命中真值 DTW 邻居);最强基线 STRAP 仅 0.0071,Flow(0.0004)与 HAND(0.0009)接近随机。top-20 平均 DTW 代价 1.333,仅比真值邻居的 1.145 高 16%(随机为 4.776)。跨域实验(表3,AssemblyHands 831 条双手装配片段、k=5):排序不变,RoboTok mAP@5=0.2614 对 STRAP 的 0.1330,DTW 代价 1.095 对真值 0.966(高 13%),证明表征可跨域迁移。下游实验:原版 VTDexManip(表4)上 RoboTok 引导的 PPO 五/六任务超过基准最强基线 VT-JointPretrain,平均高 7.45%(seen)/5.83%(unseen),如杠杆滑动 seen 95.5%±4.6 对 89.3%±3.6。困难版(表5,完整3D手控+移除稠密奖励,每方法 8 种子、每物体 100 次尝试)差距悬殊:瓶盖拧转 seen 77.3%±3.5 对次优 HAND 59.5%±9.0(+17.8),水龙头 44.8%±4.1 对 6.8%±4.8(+38.0),杠杆 79.3%±5.3 对 19.5%±4.8(+59.8);训练速度排序与最终成功率一致(图7)。

Comparison of demonstration retrieval methods for robot learning.
Table 1: Comparison of demonstration retrieval methods for robot learning.
Unseen retrieval quality on the RoboTok evaluation corpus against the pseudo-GT DTW metric at k = 20...
Table 2: Unseen retrieval quality on the RoboTok evaluation corpus against the pseudo-GT DTW metric at k = 20...
Retrieval quality measured on the AssemblyHands corpus of two-hand assembly clips against the GT DTW metric at k = 5...
Table 3: Retrieval quality measured on the AssemblyHands corpus of two-hand assembly clips against the GT DTW metric at k = 5...
Success rate (%) across all tasks of the ORIGINAL VTDexManip formulation...
Table 4: Success rate (%) across all tasks of the ORIGINAL VTDexManip formulation...
Success rate (%) for PPO policies guided by demonstrations retrieved from different retrieval methods across the modified HARDER VTDexManip task formulations...
Table 5: Success rate (%) for PPO policies guided by demonstrations retrieved from different retrieval methods across the modified HARDER VTDexManip task formulations...
Given a query demonstration video of knife cutting, we show the corresponding top-3 retrieved videos from each baseline method compared with RoboTok...
Figure 4: Given a query demonstration video of knife cutting, we show the corresponding top-3 retrieved videos from each baseline method compared with RoboTok...
Per-k plots of unseen retrieval quality on the RoboTok evaluation corpus: Recall@k (hit rate), mean DTW cost@k, and CKNNA@k (DTW alignment)...
Figure 5: Per-k plots of unseen retrieval quality on the RoboTok evaluation corpus: Recall@k (hit rate), mean DTW cost@k, and CKNNA@k (DTW alignment)...
The modified harder VTDexManip simulation-based robotic evaluation tasks. Visualized are successful, natural hand poses achieved by PPO guided by RoboTok retrieved demonstrations.
Figure 6: The modified harder VTDexManip simulation-based robotic evaluation tasks. Visualized are successful, natural hand poses achieved by PPO guided by RoboTok retrieved demonstrations.
Training curves of different retrieval-guided PPO policies on the modified HARDER tasks.
Figure 7: Training curves of different retrieval-guided PPO policies on the modified HARDER tasks.
查看结构化数据
任务指标本文基线提升
网络视频示范检索(RoboTok 评测库,10万片段、1万查询、k=20) mAP@20 / Recall@20 mAP@20=0.3531,Recall@20=0.996 STRAP 0.0071/0.12,HAND 0.0009,Flow 0.0004,Random 0.0000 mAP 约为最强基线 STRAP 的 50 倍,几乎每个查询必命中真值 DTW 邻居
跨域检索(AssemblyHands,831条双手装配片段,k=5) mAP@5 / DTW cost@5 mAP@5=0.2614,DTW 代价 1.095 STRAP 0.1330/1.304,Random 0.0025/1.911 mAP 约翻倍;DTW 代价仅比真值 0.966 高 13%
原版 VTDexManip 六任务(检索示范引导 PPO) 成功率(seen/unseen 平均) 平均超基准最优基线 7.45%(seen)/5.83%(unseen),5/6 任务第一(如杠杆滑动 seen 95.5%) VT-JointPretrain(Liu et al. 2025,如杠杆滑动 seen 89.3%) 六任务中五项超过基准最强预训练结果
困难版 VTDexManip:瓶盖拧转(3D手控+无稠密奖励) 成功率(seen/unseen) 77.3%±3.5 / 38.1%±5.5 HAND 59.5%±9.0 / 27.0%±6.0 seen +17.8 个百分点,unseen +11.1 个百分点
困难版 VTDexManip:水龙头旋拧 / 杠杆滑动(seen) 成功率 水龙头 44.8%±4.1,杠杆 79.3%±5.3 HAND 6.8%±4.8,19.5%±4.8 分别 +38.0 和 +59.8 个百分点

局限与改进

作者承认的局限:当前管线要求近静止相机(Lucas-Kanade 过滤)、片段限4-8秒、每片段最多一左一右两只手且轻度可见,排除了运动镜头、第三人称与自我中心视角视频(明确留作未来工作);DTW 伪真值建立在估计的手部姿态之上,WiLoR/MoGe-2 的误差会传导进监督信号。我的补充观察:(1) 评测存在循环论证风险——模型被训练去逼近 DTW 排序,又用 DTW 排名打分,AssemblyHands 跨域实验缓解但未消除该担忧,缺少人类判断或任务成功层面的外部行为等价验证;(2) 下游实验全部在仿真中完成,示范需经重定向进入奖励函数,真实机器人闭环效果未验证;(3) 检索只保证轨迹相似,不保证任务语义一致,查询'切菜'可能召回几何同构但语义不同的操作;(4) mAP@20=0.353 距上界 1.0 仍有很大空间,且 DTW 归一化因子 $2(L_i+L_j)$ 属启发式;(5) 并非全面占优:原版桌面重定向任务 RoboTok(seen 82.6%)低于 VT-JointPretrain(85.0%),双手传递 unseen(34.8%)甚至低于随机检索(42.1%),提示躯干系单手轨迹对双手协调与桌面平面运动的刻画能力不足。

独立分析的弱点

独立分析四个弱点。第一,数据管线假设过强:近静止相机加单手对过滤会丢弃互联网上大量有价值的第三人称、运动镜头与双手密集交互视频,而双手传递任务的相对弱势(unseen 34.8%,低于随机检索的 42.1%)可能正源于静态躯干系单手轨迹难以刻画双手协调;改进方向是引入物体接地的参考系或双手联合的相对坐标表征。第二,评测循环性:以 DTW 为真值训练又以 DTW 打分,真正的行为等价性缺乏外部验证;改进方向是构建小规模人工标注的行为等价金标集,或以多个任务的成功率作为主要评价。第三,检索语义不可控:嵌入空间不含任务意图,几何同构但语义无关的操作会混入示范库,用作 PPO 奖励引导时可能引入系统性偏置(例如把'拧瓶盖'引导向'拧水龙头'的动力学);改进方向是在检索时融合轻量语义通道做混合过滤,或按任务簇对索引分区。第四,躯干帧估计器只输出静态坐标系,对躯干大幅位移的操作(如搬移重物、蹲起)表征会失真;同时 DTW 对采样率(5fps)和归一化项敏感,超长或超短片段的相似度尺度不一致;改进方向是时变参考系与可微分软 DTW 端到端训练。

未来方向

作者明确提出的方向:扩展到运动相机视频,包括第三人称和自我中心(头戴式)示范——这需要把静态躯干帧估计推广为时变参考系,并处理自我中心视频中手部频繁出画、全局尺度漂移的问题;作者相信这能把网络视频变成机器人操作数据持续增长的来源,缓解数据采集瓶颈。基于本文成果可进一步延伸:(1) 把示范库接入 VLA 后训练流程(类比 SiMDex 的跨本体思路),让检索到的人类视频直接监督视觉-语言-动作模型,而非仅作 PPO 奖励;(2) 引入物体接地参考系,联合建模手-物体相对运动,提升双手与接触丰富任务的判别力;(3) 用检索库做课程学习——按 DTW 代价从近到远排序示范,渐进引导探索;(4) 在真实人形机器人上验证网络视频检索监督的 Sim2Real 闭环效果;(5) 用可微分软 DTW 端到端联合训练编码器与对齐,消除两阶段蒸馏的信息损失;(6) 将索引规模从10万扩展到真正的千万级 Action100M 全量,检验检索质量随规模的衰减规律。

复现评估

复现评估:论文的管线全部由公开组件搭建——Action100M 数据集、WiLoR、MoGe-2、HaWoR、SMPL-H、AssemblyHands、VTDexManip 均有开源实现,数据筛选规则(4-8秒、Lucas-Kanade 静止判定、手部可见过滤)在文中有明确描述,编码器是轻量交叉注意力网络,训练语料仅10万片段,训练本身单卡可承受。不利因素:正文只给出 Project Site 链接,未见明确的官方代码与索引发布承诺;躯干帧估计器的训练细节依赖转引 Wang et al. (2026) 的流程并适配 SMPL-H,需要自行实现;最大的隐性成本是离线数据预处理(5fps 关键点估计+度量深度推理+10万条轨迹的成对 DTW 计算),论文未披露硬件配置与耗时,成对 DTW 的计算量不可忽视。下游 PPO 实验规格清晰(每方法8种子、每物体100次尝试、每环境独立配对示范),仿真部分复现难度中等。总体判断:若官方放出预处理轨迹与索引则易于复现,否则需从原始视频重跑整个抽取管线,工程成本显著。

相似论文推荐

人类通用抓取 🎯 63.2%