LGAIMLApr 20, 2023

Efficient Deep Reinforcement Learning Requires Regulating Overfitting

Berkeley
arXiv:2304.10466v147 citationsh-index: 166
Originality Incremental advance
AI Analysis

This addresses the problem of data inefficiency in deep RL for researchers and practitioners, offering a generalizable principle, though it is incremental as it builds on existing regularization ideas.

The paper identifies high temporal-difference error on validation transitions as the main bottleneck in sample-efficient deep reinforcement learning, showing that a simple online model selection method targeting this error improves performance across state-based DeepMind control and Gym tasks.

Deep reinforcement learning algorithms that learn policies by trial-and-error must learn from limited amounts of data collected by actively interacting with the environment. While many prior works have shown that proper regularization techniques are crucial for enabling data-efficient RL, a general understanding of the bottlenecks in data-efficient RL has remained unclear. Consequently, it has been difficult to devise a universal technique that works well across all domains. In this paper, we attempt to understand the primary bottleneck in sample-efficient deep RL by examining several potential hypotheses such as non-stationarity, excessive action distribution shift, and overfitting. We perform thorough empirical analysis on state-based DeepMind control suite (DMC) tasks in a controlled and systematic way to show that high temporal-difference (TD) error on the validation set of transitions is the main culprit that severely affects the performance of deep RL algorithms, and prior methods that lead to good performance do in fact, control the validation TD error to be low. This observation gives us a robust principle for making deep RL efficient: we can hill-climb on the validation TD error by utilizing any form of regularization techniques from supervised learning. We show that a simple online model selection method that targets the validation TD error is effective across state-based DMC and Gym tasks.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes