PAC-Bayesian Soft Actor-Critic Learning

Bahareh Tasdighi, Abdullah Akgül, Manuel Haussmann, Kenny Kazimirzak Brink, Melih Kandemir

Keywords: reinforcement learning, PAC Bayesian theory, actor-critic algorithms

TL;DR: We introduce the first PAC Bayesian bound trainable in a modern deep reinforcement learning pipeline to solve continuous control tasks.

Abstract: Actor-critic algorithms address the dual goals of reinforcement learning (RL), policy evaluation, and improvement via two separate function approximators. The practicality of this approach comes at the expense of training instability, caused mainly by the destructive effect of the approximation errors of the critic on the actor. We tackle this bottleneck by employing an existing Probably Approximately Correct (PAC) Bayesian bound for the first time as the critic training objective of the Soft Actor-Critic (SAC) algorithm. We further demonstrate that online learning performance improves significantly when a stochastic actor explores multiple futures by critic-guided random search. We observe our resulting algorithm to compare favorably against the state-of-the-art SAC implementation on multiple classical control and locomotion tasks in terms of both sample efficiency and regret.

Submission Number: 11