日本フィジカルAI新聞

世界のフィジカルAIを、日本語で。

週刊ニュースレター購読
sim2realarXiv:2405.11512

軌道投入:エピソード強化学習の大規模並列化

Going into Orbit: Massively Parallelizing Episodic Reinforcement Learning

シェア:XThreadsFacebookLINEはてブBluesky

NVIDIAのOrbitフレームワークを用いて箱押しタスクの強化学習をGPUで並列化し、CPU実装と比較して性能を評価した。

著者: Jan Oberst, Johann Bonneau

分類: cs.RO

原文アブストラクト

The possibilities of robot control have multiplied across various domains through the application of deep reinforcement learning. To overcome safety and sampling efficiency issues, deep reinforcement learning models can be trained in a simulation environment, allowing for faster iteration cycles. This can be enhanced further by parallelizing the training process using GPUs. NVIDIA's open-source robot learning framework Orbit leverages this potential by wrapping tensor-based reinforcement learning libraries for high parallelism and building upon Isaac Sim for its simulations. We contribute a detailed description of the implementation of a benchmark reinforcement learning task, namely box pushing, using Orbit. Additionally, we benchmark the performance of our implementation in comparison to a CPU-based implementation and report the performance metrics. Finally, we tune the hyper parameters of our implementation and show that we can generate significantly more samples in the same amount of time by using Orbit.

関連論文

PR本紙発行元 EmplifAI