Bridging the Gap Between Target Networks and Functional Regularization

TL;DR

This paper analyzes the role of Target Networks in deep Reinforcement Learning, revealing their regularization effect, and proposes an explicit Functional Regularization method that improves stability and performance.

Contribution

It introduces a novel explicit Functional Regularization approach as a flexible alternative to Target Networks, with theoretical convergence analysis and empirical validation.

Findings

Functional Regularization can replace Target Networks effectively.

Adjusting regularization weight and update period improves performance.

The new method enhances accurate Q-value recovery.

Abstract

Bootstrapping is behind much of the successes of deep Reinforcement Learning. However, learning the value function via bootstrapping often leads to unstable training due to fast-changing target values. Target Networks are employed to stabilize training by using an additional set of lagging parameters to estimate the target values. Despite the popularity of Target Networks, their effect on the optimization is still misunderstood. In this work, we show that they act as an implicit regularizer which can be beneficial in some cases, but also have disadvantages such as being inflexible and can result in instabilities, even when vanilla TD(0) converges. To overcome these issues, we propose an explicit Functional Regularization alternative that is flexible and a convex regularizer in function space and we theoretically study its convergence. We conduct an experimental study across a range of environments, discount factors, and off-policiness data collections to investigate the effectiveness of the regularization induced by Target Networks and Functional Regularization in terms of performance, accuracy, and stability. Our findings emphasize that Functional Regularization can be used as a drop-in replacement for Target Networks and result in performance improvement. Furthermore, adjusting both the regularization weight and the network update period in Functional Regularization can result in further performance improvements compared to solely adjusting the network update period as typically done with Target Networks. Our approach also enhances the ability to networks to recover accurate $Q$-values.