ABC:开源大规模行为克隆全栈方案
ABC: Scalable Behavior Cloning with Open Data, Training, and Evaluation
摘要 Abstract
本文提出了 ABC(A, B, C of Behavior Cloning),一个完全开源的行为克隆全栈方案。其核心是 ABC-130K——目前最大的开源遥操作数据集,包含 3,500 小时的真实世界交互数据,涵盖 130,000+ 轨迹和 195 个多样化任务。此外,作者还开源了硬件设计、训练基础设施和仿真流水线,并释放了 400 小时的仿真遥操作数据和配套的 sim-to-real 评估方案。
研究团队系统性地探索了 Diffusion Transformer(DiT)和 Vision-Language-Action(VLA)两类策略架构的设计选择,包括视觉编码器、条件化方案和训练超参等,并在真实世界中进行了大规模消融实验。最终训练出的策略能够完成折叠纸盒、从钱包中抽取信用卡等灵巧操作任务。
ABC 是首个覆盖"数据采集→模型训练→仿真评估→真实部署"全链路的开源行为克隆框架,以 $8,000 的低成本硬件实现了 195 个任务的大规模数据收集和策略学习。
研究背景 Background
行为克隆(Behavior Cloning, BC)是机器人操作领域最有前景的学习范式之一,但其技术栈非常深——从硬件搭建、数据采集到模型训练和评估,每一步都需要大量工程投入。当前该领域面临几个关键瓶颈:
- 数据封闭性:最先进的系统通常在工业实验室的私有数据集上开发,训练配方和关键设计选择很少公开,社区难以复现和比较。
- 现有开源数据不足:DROID(350小时单臂)和 BridgeData-V2(100小时)规模有限且任务简单;Open X-Embodiment 质量参差不齐;AgiBot-World 需要 $30,000 的昂贵硬件。
- 评估困难:真实世界评估昂贵且耗时,缺乏可靠的仿真替代方案来快速迭代设计决策。
- 架构选择不明确:DiT 和 VLA 两种主流架构在不同场景下各有优劣,但缺乏系统性的对比研究。
方法论 Methodology
ABC 由四个相互关联的模块组成,构成一个完整的开源生态系统:
ABC-130K:大规模数据集
- 规模:134,806 个轨迹,195 个任务,3,553 小时双臂操作数据
- 硬件:基于 YAM 双臂平台,整套硬件成本仅 $8,000
- 任务多样性:涵盖 7 大操作原语类别——抓取放置、精细抓取、折叠、插入/弹出、分拣、工具使用、系/解绳
- 附加标注:1,552 小时的子目标标注子集,以及匿名化的遥操作者 ID 和时间戳元数据
ABC-Models:架构探索
论文系统比较了两类策略架构的多种设计选择:
- ABC-DiT(2B 参数):基于 Diffusion Transformer 的策略,研究了 CLIP-AdaLN、CLIP-Cross-Attention、DINO-Cross-Attention 三种视觉条件化方案。结论:DINOv3 + Pooled Cross-Attention 效果最佳。
- ABC-VLA(4B 参数):基于 Gemma 3 VLM + DiT 动作头,比较了 Cross-Attention、Cross-Attention + FAST、AdaLN 三种连接器。结论:AdaLN 在该数据上表现最好(61.4% 进度 vs 32.6% vs 11.7%)。
- 方差缩减技巧:将单次 VLM 前向传播复用于多个 (ε, τ) 采样,显著降低梯度方差,几乎不增加计算开销。
ABC-Sim:仿真评估
- 环境:10 个 MuJoCo 仿真任务,支持 VR 遥操作(Meta Quest / Apple Vision Pro)
- 数据:400 小时仿真遥操作数据
- 关键验证:仿真性能与真实世界性能高度相关(严格成功率 r=0.85,任务进度 r=0.91)
ABC-Eval:真实世界评估
提供 100+ 小时的真实世界策略评估数据,包含评估分数和评分标准(rubrics),为社区提供可复现的评估基准。
关键结果 Key Results
论文的核心实验结果涵盖架构消融、计算缩放、离线指标验证和真实世界能力展示四个方面:
| 架构 | 视觉编码器 | 条件化方案 | 严格成功率 | 任务进度 | 延迟 (ms) |
|---|---|---|---|---|---|
| DiT | CLIP | AdaLN | 3.6% | 32.6% | 19.24 |
| DiT | CLIP | Cross-Attn | 0.0% | 11.7% | 19.24 |
| DiT | DINOv3 | Cross-Attn | 24.5% | 58.8% | 37.5 |
| VLA | Gemma 3 | Cross-Attn | 0.0% | 11.7% | 19.24 |
| VLA | Gemma 3 | Cross-Attn + FAST | 13.4% | 47.3% | 27.5 |
| VLA | Gemma 3 | AdaLN | 32.9% | 67.5% | 37.4 |
- 视觉编码器选择至关重要:DINOv3 替代 CLIP 后,DiT 的任务进度从 32.6% 提升到 58.8%(+80% 相对提升),表明像素对齐的视觉表征对操作任务更有效。
- AdaLN 优于 Cross-Attention:对于 VLA 架构,AdaLN 条件化方案(67.5% 进度)显著优于 Cross-Attention(11.7%)和 FAST 共训练(47.3%),说明扩散梯度与 VLM 特征并非天然不兼容。
- Batch size 对 VLA 更重要:从 1.5K 到 9.2K batch size,VLA 性能大幅提升,而 DiT 改善有限。大 batch 下 VLA 可以超越 DiT。
- 离线指标可预测真实性能:训练损失和验证动作误差与真实世界性能显著相关(r=-0.84 和 r=-0.89),验证损失则无相关性。
- 预训练数据越多,下游微调越好:从零训练 → 3.5K 小时预训练 → 7K 小时预训练,下游精细任务(抽取信用卡、分拣乐高、拧瓶盖)的表现持续提升。
- DAgger 对灵巧任务不可或缺:纸盒折叠任务中,微调仅达 24% 成功率,经过两轮 DAgger(每次 1-1.5 小时恢复行为采集)后显著提升。
ABC 证明了:(1) 开源、低成本的硬件方案可以收集到高质量的大规模操作数据;(2) DINOv3 视觉编码器 + AdaLN 条件化是 DiT/VLA 架构的最优组合;(3) 仿真评估与真实世界高度相关(r>0.85),可以作为可靠的预评估代理;(4) 预训练数据规模对下游精细任务有显著正向影响。
个人见解 Insights
这篇论文的最大贡献不仅在于数据集本身,更在于它为机器人学习社区建立了一个可复现的基准生态系统。以下是我对这篇工作的一些思考:
- 优势——全栈开源的标杆意义:不同于以往只释放数据或模型权重的工作,ABC 同时开源了硬件设计、训练代码、仿真环境和评估流程。这种"全栈开源"模式大幅降低了社区参与的门槛,尤其是 $8,000 的硬件成本远低于 AgiBot-World 的 $30,000。
- 优势——系统性的架构消融:论文对 DiT 和 VLA 的多种设计选择进行了严格的控制变量实验,这是以往封闭源代码工作中从未见过的。特别是 AdaLN vs Cross-Attention 的对比,澄清了"扩散梯度是否伤害 VLM 特征"这一长期争议。
- 不足——任务覆盖仍有局限:虽然 195 个任务已经很多,但大部分仍是桌面操作场景。移动操作、接触丰富的力控任务、长时间序列表达式任务仍然缺乏。
- 不足——缺少与最新 VLA 的直接对比:论文没有与 π0.5、GR00T N1 等最新闭源 VLA 进行直接性能对比,使得难以判断 ABC 模型在绝对性能上的位置。
- 启发——仿真作为设计决策的代理:仿真相关系数 r>0.9 的发现非常有价值,意味着社区可以在没有硬件的情况下进行有效的策略设计迭代,这对资源有限的实验室尤其重要。
- 启发——DAgger 的实用化路径:论文展示的"每轮 1-1.5 小时恢复行为采集"方案非常实用,为灵巧操作任务提供了一条低成本的性能提升路径。
参考资料 References
查看原文 View Original Paper
arXiv: 2606.27375
下载 PDF Download PDF
arXiv PDF
项目主页 Project Page
abc.bot