Preventing Value Function Collapse in Ensemble Q-Learning by Maximizing Representation Diversity
The first deep RL algorithm, DQN, was limited by the overestimation bias of the learned Q-function. Subsequent algorithms proposed techniques to reduce this problem, without fully eliminating it. Recently, the Maxmin and Ensemble Q-learning algorithms used the different estimates provided by ensembles of learners to reduce the bias. Unfortunately, these learners can converge to the same point in the parametric or representation space, falling back to the classic single neural network DQN. In this paper, we describe a regularization technique to maximize diversity in the representation space in these algorithms. We propose and compare five regularization functions inspired from economics theory and consensus optimization. We show that the resulting approach significantly outperforms the Maxmin and Ensemble Q-learning algorithms as well as non-ensemble baselines.
Code (0)
등록된 구현이 없습니다.
Tasks
DiversityQ-LearningMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Preventing Value Function Collapse in Ensemble {Q}-Learning by Maximizing Representation Diversity
The classic DQN algorithm is limited by the overestimation bias of the learned Q-function. Subsequent algorithms have proposed techniques to reduce this problem, without fully eliminating it. Recently, the Maxmin and Ens…
DiversityQ-LearningTQL: Scaling Q-Functions with Transformers by Preventing Attention Collapse
Despite scale driving substantial recent advancements in machine learning, reinforcement learning (RL) methods still primarily use small value functions. Naively scaling value functions -- including with a transformer ar…
Reinforcement LearningModulate Your Spectrum in Self-Supervised Learning
Whitening loss offers a theoretical guarantee against feature collapse in self-supervised learning (SSL) with joint embedding architectures. Typically, it involves a hard whitening approach, transforming the embedding an…
object-detectionObject DetectionSelf-Supervised LearningBalancing policy constraint and ensemble size in uncertainty-based offline reinforcement learning
Offline reinforcement learning agents seek optimal policies from fixed data sets. With environmental interaction prohibited, agents face significant challenges in preventing errors in value estimates from compounding and…
Behavioural cloningBenchmarkingreinforcement-learningAnti-Collapse Loss for Deep Metric Learning Based on Coding Rate Metric
Deep metric learning (DML) aims to learn a discriminative high-dimensional embedding space for downstream tasks like classification, clustering, and retrieval. Prior literature predominantly focuses on pair-based and pro…
DiversityMetric Learning