AdaStop: sequential testing for efficient and reliable comparisons of Deep RL Agents

Timothée Mathieu (1, 2, 3, 4, 5) , Riccardo Della Vecchia (1, 2, 3, 4, 5) , Alena Shilova (1, 2, 3, 4, 5) , Matheus Medeiros Centa (1, 2, 3, 4, 5) , Hector Kohler (1, 2, 3, 4, 5) , Odalric-Ambrym Maillard (1, 2, 3, 4, 5) , Philippe Preux (1, 2, 3, 4, 5)

1 CRIStAL - Centre de Recherche en Informatique, Signal et Automatique de Lille - UMR 9189
2 Centrale Lille
3 CNRS - Centre National de la Recherche Scientifique
4 Université de Lille
5 Scool - Scool

The reproducibility of many experimental results in Deep Reinforcement Learning (RL) is under question. To solve this reproducibility crisis, we propose a theoretically sound methodology to compare multiple Deep RL algorithms. The performance of one execution of a Deep RL algorithm is random so that independent executions are needed to assess it precisely. When comparing several RL algorithms, a major question is how many executions must be made and how can we assure that the results of such a comparison is theoretically sound. Researchers in Deep RL often use less than 5 independent executions to compare algorithms: we claim that this is not enough in general. Moreover, when comparing several algorithms at once, the error of each comparison accumulates and must be taken into account with a multiple tests procedure to preserve low error guarantees. To address this problem in a statistically sound way, we introduce AdaStop, a new statistical test based on multiple group sequential tests. When comparing algorithms, AdaStop adapts the number of executions to stop as early as possible while ensuring that we have enough information to distinguish algorithms that perform better than the others in a statistical significant way. We prove both theoretically and empirically that AdaStop has a low probability of making an error (Family-Wise Error). Finally, we illustrate the effectiveness of AdaStop in multiple use-cases, including toy examples and difficult cases such as Mujoco environments.

Mots clés: Reproducibility; Reinforcement Learning; Sequential Tests

Domaines: Machine Learning [stat.ML]

Timothée Mathieu, Riccardo Della Vecchia, Alena Shilova, Matheus Medeiros Centa, Hector Kohler, et al.. AdaStop: sequential testing for efficient and reliable comparisons of Deep RL Agents. Transactions on Machine Learning Research Journal, 2024. ⟨hal-04132861v2⟩