arXiv:2509.16434cs.ROcs.LG2025-09被引 7

通过拆分模拟器与强化学习,实现更高效的手部抓取视觉策略训练。

End-to-end RL Improves Dexterous Grasping Policies

  • 将模拟器与RL分离部署于不同GPU,提升训练效率。
  • 相同硬件下环境数量翻倍,批大小显著增加。
  • 基于深度信息的策略蒸馏在真实场景表现更优。

本文研究如何扩展基于图像的端到端强化学习在机械臂+手系统中的应用,以实现灵巧抓取。与基于状态的RL相比,视觉驱动的RL内存效率较低,导致批量较小,难以适配PPO等算法。然而,其优势在于可生成主动视觉行为。我们发现现有模拟器采用传统数据并行时存在性能瓶颈。提出将模拟器与强化学习(含训练和经验缓冲)拆分至独立GPU:4卡节点中3卡运行模拟器,1卡运行PPO。相比基准方案,相同设备下可使环境数翻倍,从而支持深度感知的端到端训练。我们同时训练并蒸馏深度与状态基策略至立体RGB网络,结果表明深度蒸馏效果更佳,原因在于深度策略避免了状态与视觉间的可观测性差距。更大的批量也提升了真实世界表现,所提方法在真实场景中超越此前最先进视觉基抓取结果。

原文摘要 · Abstract (English)

This work explores techniques to scale up image-based end-to-end learning for dexterous grasping with an arm + hand system. Unlike state-based RL, vision-based RL is much more memory inefficient, resulting in relatively low batch sizes, which is not amenable for algorithms like PPO. Nevertheless, it is still an attractive method as unlike the more commonly used techniques which distill state-based policies into vision networks, end-to-end RL can allow for emergent active vision behaviors. We identify a key bottleneck in training these policies is the way most existing simulators scale to multiple GPUs using traditional data parallelism techniques. We propose a new method where we disaggregate the simulator and RL (both training and experience buffers) onto separate GPUs. On a node with four GPUs, we have the simulator running on three of them, and PPO running on the fourth. We are able to show that with the same number of GPUs, we can double the number of existing environments compared to the previous baseline of standard data parallelism. This allows us to train vision-based environments, end-to-end with depth, which were previously performing far worse with the baseline. We train and distill both depth and state-based policies into stereo RGB networks and show that depth distillation leads to better results, both in simulation and reality. This improvement is likely due to the observability gap between state and vision policies which does not exist when distilling depth policies into stereo RGB. We further show that the increased batch size brought about by disaggregated simulation also improves real world performance. When deploying in the real world, we improve upon the previous state-of-the-art vision-based results using our end-to-end policies.

强化学习灵巧抓取视觉策略模拟器优化

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