MarineGym: A High-Performance Reinforcement Learning Platform for Underwater Robotics

Shuguang Chu, Zebin Huang, Yutong Li, Mingwei Lin, Dejun Li, Ignacio Carlucho, Yvan R. Petillot, Canjun Yang

Research output: Working paperPreprint

Abstract

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/.
Original languageEnglish
PublisherarXiv
DOIs
Publication statusPublished - 12 Mar 2025

Keywords

  • cs.RO
  • cs.LG

Fingerprint

Dive into the research topics of 'MarineGym: A High-Performance Reinforcement Learning Platform for Underwater Robotics'. Together they form a unique fingerprint.

Cite this