Efficient Offline Reinforcement Learning: The Critic is Critical

Adam Jelley,Trevor McInroe,Sam Devlin,Amos Storkey
2024-06-19
Abstract:Recent work has demonstrated both benefits and limitations from using supervised approaches (without temporal-difference learning) for offline reinforcement learning. While off-policy reinforcement learning provides a promising approach for improving performance beyond supervised approaches, we observe that training is often inefficient and unstable due to temporal difference bootstrapping. In this paper we propose a best-of-both approach by first learning the behavior policy and critic with supervised learning, before improving with off-policy reinforcement learning. Specifically, we demonstrate improved efficiency by pre-training with a supervised Monte-Carlo value-error, making use of commonly neglected downstream information from the provided offline trajectories. We find that we are able to more than halve the training time of the considered offline algorithms on standard benchmarks, and surprisingly also achieve greater stability. We further build on the importance of having consistent policy and value functions to propose novel hybrid algorithms, TD3+BC+CQL and EDAC+BC, that regularize both the actor and the critic towards the behavior policy. This helps to more reliably improve on the behavior policy when learning from limited human demonstrations. Code is available at <a class="link-external link-https" href="https://github.com/AdamJelley/EfficientOfflineRL" rel="external noopener nofollow">this https URL</a>
Machine Learning
What problem does this paper attempt to address?
This paper discusses how to combine supervised learning and temporal-difference learning in offline reinforcement learning. The research finds that using either supervised learning or unsupervised temporal-difference learning alone has limitations. The paper proposes a new approach that pretrains the behavior policy and critic using supervised learning, and then improves them using offline reinforcement learning. The key innovation is that pretraining the critic improves training efficiency and stability, and reduces instability when learning from the behavior policy. Specifically, the paper uses supervised learning's Monte Carlo value error for pretraining, leveraging the provided offline trajectory information to improve initial value estimation. Experiments show that this approach can reduce training time by more than half on standard benchmarks and improve stability. Additionally, the paper proposes hybrid algorithms such as TD3+BC+CQL and EDAC+BC, which use regularization policies and critics to make them closer to the behavior policy and more reliably improve behavior policies with limited human demonstration data. The paper further discusses how to apply this approach to reinforcement learning for reward maximization and entropy-regularized reinforcement learning algorithms such as TD3+BC and EDAC, demonstrating performance improvements in complex tasks including the Adroit environment with limited data. These findings emphasize the importance of having consistent policies and value functions, as well as the value of pretraining in offline reinforcement learning.