Reliable validation of Reinforcement Learning Benchmarks

TL;DR

This paper introduces a method for validating reinforcement learning benchmarks by providing minimal traces that enable re-simulation and verification of experimental results, enhancing reproducibility and trustworthiness.

Contribution

It proposes a minimal trace approach for RL benchmarks, allowing easy validation, re-use, and inspection of results without extensive computation, integrated with existing RL tools.

Findings

Minimal traces enable re-simulation of action sequences.

The approach significantly reduces data size for storage and sharing.

Proof-of-concept results demonstrate effectiveness across various games.

Abstract

Reinforcement Learning (RL) is one of the most dynamic research areas in Game AI and AI as a whole, and a wide variety of games are used as its prominent test problems. However, it is subject to the replicability crisis that currently affects most algorithmic AI research. Benchmarking in Reinforcement Learning could be improved through verifiable results. There are numerous benchmark environments whose scores are used to compare different algorithms, such as Atari. Nevertheless, reviewers must trust that figures represent truthful values, as it is difficult to reproduce an exact training curve. We propose improving this situation by providing access to the original experimental data to validate study results. To that end, we rely on the concept of minimal traces. These allow re-simulation of action sequences in deterministic RL environments and, in turn, enable reviewers to verify, re-use, and manually inspect experimental results without needing large compute clusters. It also permits validation of presented reward graphs, an inspection of individual episodes, and re-use of result data (baselines) for proper comparison in follow-up papers. We offer plug-and-play code that works with Gym so that our measures fit well in the existing RL and reproducibility eco-system. Our approach is freely available, easy to use, and adds minimal overhead, as minimal traces allow a data compression ratio of up to $\approx 10^4:1$ (94GB to 8MB for Atari Pong) compared to a regular MDP trace used in offline RL datasets. The paper presents proof-of-concept results for a variety of games.