Kamino用GPU并行模拟复杂机械系统,支持闭链结构的高保真物理仿真。
Kamino: GPU-based Massively Parallel Simulation of Multi-Body Systems with Challenging Topologies
- 基于约束优化算法,原生支持闭链与强耦合机械系统仿真
- 单卡并行4096个环境,实现六重闭环机器人行走策略训练
- 适合需要高精度多体仿真的机器人强化学习研究者
我们提出Kamino,一种基于GPU的物理求解器,用于大规模并行仿真异构高度耦合的机械系统。使用Python结合NVIDIA Warp实现,并集成至Newton框架,支持数据驱动方法(如大规模强化学习)在具有强运动学与动力学约束的复杂机器人系统中的应用,例如包含运动学环的系统。传统方法常将系统简化为树状结构或引入显式环闭合约束,而Kamino直接原生支持此类耦合结构。该能力使高吞吐并行仿真成为可能,真实捕捉利用闭链结构获取机械优势的系统特性。同时,Kamino支持异构世界,可在单张GPU上批量仿真结构各异的机器人。其核心是先进的约束优化算法,通过将受约束刚体多体正向动力学转化为非线性互补问题求解约束力,实现无需近似模型即可解析接触动力学的高保真仿真。我们在一个具有六重嵌套运动学环的双足机器人DR Legs上展示了强化学习策略训练,成功生成可行步行策略,且在单卡上并行运行4096个仿真环境。
原文摘要 · Abstract (English)
We present Kamino, a GPU-based physics solver for massively parallel simulations of heterogeneous highly-coupled mechanical systems. Implemented in Python using NVIDIA Warp and integrated into the Newton framework, it enables the application of data-driven methods, such as large-scale reinforcement learning, to complex robotic systems that exhibit strongly coupled kinematic and dynamic constraints such as kinematic loops. The latter are often circumvented by practitioners; approximating the system topology as a kinematic tree and incorporating explicit loop-closure constraints or so-called mimic joints. Kamino aims at alleviating this burden by natively supporting these types of coupling. This capability facilitates high-throughput parallelized simulations that capture the true nature of mechanical systems that exploit closed kinematic chains for mechanical advantage. Moreover, Kamino supports heterogeneous worlds, allowing for batched simulation of structurally diverse robots on a single GPU. At its core lies a state-of-the-art constrained optimization algorithm that computes constraint forces by solving the constrained rigid multi-body forward dynamics transcribed as a nonlinear complementarity problem. This leads to high-fidelity simulations that can resolve contact dynamics without resorting to approximate models that simplify and/or convexify the problem. We demonstrate RL policy training on DR Legs, a biped with six nested kinematic loops, generating a feasible walking policy while simulating 4096 parallel environments on a single GPU.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。