LGJun 28, 2024

Contextualized Hybrid Ensemble Q-learning: Learning Fast with Control Priors

arXiv:2406.19768v28 citations
Originality Incremental advance
AI Analysis

This addresses the need for safer and more efficient RL training in control tasks, though it is incremental as it builds on prior hybrid RL methods.

The paper tackled the problem of adaptive weighting in hybrid reinforcement learning by proposing CHEQ, which dynamically adjusts the weight between RL and a control prior based on the agent's capabilities, resulting in substantially stronger data efficiency, exploration safety, and transferability in a car racing task.

Combining Reinforcement Learning (RL) with a prior controller can yield the best out of two worlds: RL can solve complex nonlinear problems, while the control prior ensures safer exploration and speeds up training. Prior work largely blends both components with a fixed weight, neglecting that the RL agent's performance varies with the training progress and across regions in the state space. Therefore, we advocate for an adaptive strategy that dynamically adjusts the weighting based on the RL agent's current capabilities. We propose a new adaptive hybrid RL algorithm, Contextualized Hybrid Ensemble Q-learning (CHEQ). CHEQ combines three key ingredients: (i) a time-invariant formulation of the adaptive hybrid RL problem treating the adaptive weight as a context variable, (ii) a weight adaption mechanism based on the parametric uncertainty of a critic ensemble, and (iii) ensemble-based acceleration for data-efficient RL. Evaluating CHEQ on a car racing task reveals substantially stronger data efficiency, exploration safety, and transferability to unknown scenarios than state-of-the-art adaptive hybrid RL methods.

Code Implementations1 repo
Foundations

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

Your Notes