Table of Contents
Fetching ...

Robust Gymnasium: A Unified Modular Benchmark for Robust Reinforcement Learning

Shangding Gu, Laixi Shi, Muning Wen, Ming Jin, Eric Mazumdar, Yuejie Chi, Adam Wierman, Costas Spanos

TL;DR

Robust-Gymnasium introduces a unified, modular benchmark to systematically evaluate robust reinforcement learning across disruptions occurring at observations, actions, rewards, and environment. By formalizing disruptions within a disrupted-MDP framework and assembling over 60 task bases from robotics, safe RL, and multi-agent RL, the paper enables standardized assessment of SOTA standard, robust, safe, and MARL algorithms, including a novel LLM-driven adversarial disturbance. Experimental results show that existing methods often struggle under multi-source, adversarial, and non-stationary disruptions, underscoring the need for new robust RL approaches and validating the benchmark’s utility for development and comparison. The work aims to accelerate progress toward reliable, real-world capable RL systems by providing rich task diversity, flexible disruption design, and clear evaluation protocols.

Abstract

Driven by inherent uncertainty and the sim-to-real gap, robust reinforcement learning (RL) seeks to improve resilience against the complexity and variability in agent-environment sequential interactions. Despite the existence of a large number of RL benchmarks, there is a lack of standardized benchmarks for robust RL. Current robust RL policies often focus on a specific type of uncertainty and are evaluated in distinct, one-off environments. In this work, we introduce Robust-Gymnasium, a unified modular benchmark designed for robust RL that supports a wide variety of disruptions across all key RL components-agents' observed state and reward, agents' actions, and the environment. Offering over sixty diverse task environments spanning control and robotics, safe RL, and multi-agent RL, it provides an open-source and user-friendly tool for the community to assess current methods and foster the development of robust RL algorithms. In addition, we benchmark existing standard and robust RL algorithms within this framework, uncovering significant deficiencies in each and offering new insights.

Robust Gymnasium: A Unified Modular Benchmark for Robust Reinforcement Learning

TL;DR

Robust-Gymnasium introduces a unified, modular benchmark to systematically evaluate robust reinforcement learning across disruptions occurring at observations, actions, rewards, and environment. By formalizing disruptions within a disrupted-MDP framework and assembling over 60 task bases from robotics, safe RL, and multi-agent RL, the paper enables standardized assessment of SOTA standard, robust, safe, and MARL algorithms, including a novel LLM-driven adversarial disturbance. Experimental results show that existing methods often struggle under multi-source, adversarial, and non-stationary disruptions, underscoring the need for new robust RL approaches and validating the benchmark’s utility for development and comparison. The work aims to accelerate progress toward reliable, real-world capable RL systems by providing rich task diversity, flexible disruption design, and clear evaluation protocols.

Abstract

Driven by inherent uncertainty and the sim-to-real gap, robust reinforcement learning (RL) seeks to improve resilience against the complexity and variability in agent-environment sequential interactions. Despite the existence of a large number of RL benchmarks, there is a lack of standardized benchmarks for robust RL. Current robust RL policies often focus on a specific type of uncertainty and are evaluated in distinct, one-off environments. In this work, we introduce Robust-Gymnasium, a unified modular benchmark designed for robust RL that supports a wide variety of disruptions across all key RL components-agents' observed state and reward, agents' actions, and the environment. Offering over sixty diverse task environments spanning control and robotics, safe RL, and multi-agent RL, it provides an open-source and user-friendly tool for the community to assess current methods and foster the development of robust RL algorithms. In addition, we benchmark existing standard and robust RL algorithms within this framework, uncovering significant deficiencies in each and offering new insights.

Paper Structure

This paper contains 48 sections, 8 equations, 28 figures, 13 tables.

Figures (28)

  • Figure 1: The overview of Robust-Gymnasium. For more details, please visit the https://robust-rl.github.io/.
  • Figure 2: The overview of a finite-horizon MDP with disruptors.
  • Figure 3: Illustration of two disruption modes of the environment-disruptor: internal dynamic shift and external disturbance.
  • Figure 4: Adversary attack on robot environments, dynamics and shape with different distributions (We can also attack on robot state space, action space and reward signal, etc.).
  • Figure 5: Adversary attack on state and action space in robust HalfCheetah-v4 tasks. S denotes attack on state and A denotes attack on action.
  • ...and 23 more figures