MarineGym:水中ロボティクス向け高性能強化学習プラットフォーム
MarineGym: A High-Performance Reinforcement Learning Platform for Underwater Robotics
水中ロボット向けにIsaac SimベースのGPU加速流体力学プラグインを統合し、単一GPUで毎秒25万フレームの高速学習を実現する強化学習プラットフォームを提案。
著者: Shuguang Chu, Zebin Huang, Yutong Li, Mingwei Lin, Ignacio Carlucho, Yvan R. Petillot, Canjun Yang
分類: cs.RO, cs.LG
原文アブストラクト
This work presents the MarineGym, a high-performance reinforcement learning (RL) platform specifically designed for underwater robotics. It aims to address the limitations of existing underwater simulation environments in terms of RL compatibility, training efficiency, and standardized benchmarking. MarineGym integrates a proposed GPU-accelerated hydrodynamic plugin based on Isaac Sim, achieving a rollout speed of 250,000 frames per second on a single NVIDIA RTX 3060 GPU. It also provides five models of unmanned underwater vehicles (UUVs), multiple propulsion systems, and a set of predefined tasks covering core underwater control challenges. Additionally, the DR toolkit allows flexible adjustments of simulation and task parameters during training to improve Sim2Real transfer. Further benchmark experiments demonstrate that MarineGym improves training efficiency over existing platforms and supports robust policy adaptation under various perturbations. We expect this platform could drive further advancements in RL research for underwater robotics. For more details about MarineGym and its applications, please visit our project page: https://marine-gym.com/.
関連論文
- 水中音響エネルギー伝送と情報鮮度最小化のためのAUV軌道学習水中ロボティクス/強化学習
- MarineGym: 高忠実度RLシミュレーションによる水中ビークルの高速学習水中ロボティクス/強化学習
