arXiv:2505.15558cs.ROcs.AI2025-05ICRA被引 3

高效管理机器人大数据集,压缩比达70倍,训练更快更省资源。

Robo-DM: Data Management For Large Robot Datasets

  • 用EBML格式封装多模态机器人数据,支持高效存储与共享。
  • 相比RLDS,数据压缩最高节省70倍空间(有损),加载速度提升50倍。
  • 适合需要处理海量机器人轨迹数据的研究者和工业应用开发者。

近期研究表明,大规模遥操作机器人示范数据可训练基于Transformer的模型,具备泛化至新场景、新机器人和新任务的潜力。然而,收集、分发和加载包含视频、文本和数值流等多模态的机器人轨迹数据仍具挑战。我们提出Robo-DM,一个开源的云原生数据管理工具包,用于机器人数据的采集、共享与学习。Robo-DM将机器人数据以自包含的EBML格式存储,显著减少数据体积、传输成本和训练加载时间。相比OXE数据集使用的RLDS格式,Robo-DM在有损压缩下最多节省70倍空间,无损压缩下节省3.5倍。通过内存映射解码缓存实现负载均衡,数据检索速度大幅提升;相较于LeRobot框架,序列解码速度最快快50倍。我们物理验证了使用有损压缩训练的模型,在抓取放置任务中表现良好,数据压缩率达75倍,下游任务精度未下降。

原文摘要 · Abstract (English)

Recent results suggest that very large datasets of teleoperated robot demonstrations can be used to train transformer-based models that have the potential to generalize to new scenes, robots, and tasks. However, curating, distributing, and loading large datasets of robot trajectories, which typically consist of video, textual, and numerical modalities - including streams from multiple cameras - remains challenging. We propose Robo-DM, an efficient open-source cloud-based data management toolkit for collecting, sharing, and learning with robot data. With Robo-DM, robot datasets are stored in a self-contained format with Extensible Binary Meta Language (EBML). Robo-DM can significantly reduce the size of robot trajectory data, transfer costs, and data load time during training. Compared to the RLDS format used in OXE datasets, Robo-DM's compression saves space by up to 70x (lossy) and 3.5x (lossless). Robo-DM also accelerates data retrieval by load-balancing video decoding with memory-mapped decoding caches. Compared to LeRobot, a framework that also uses lossy video compression, Robo-DM is up to 50x faster when decoding sequentially. We physically evaluate a model trained by Robo-DM with lossy compression, a pick-and-place task, and In-Context Robot Transformer. Robo-DM uses 75x compression of the original dataset and does not suffer reduction in downstream task accuracy.

机器人数据数据管理压缩算法高效训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。