arXiv · 2111.00206
One Step at a Time: Pros and Cons of Multi-Step Meta-Gradient Reinforcement Learning
Abstract
Self-tuning algorithms that adapt the learning process online encourage more effective and robust learning. Among all the methods available, meta-gradients have emerged as a promising approach. They leverage the differentiability of the learning rule with respect to some hyper-parameters to adapt them in an online fashion. Although meta-gradients can be accumulated over multiple learning steps to avoid myopic updates, this is rarely used in practice. In this work, we demonstrate that whilst multi-step meta-gradients do provide a better learning signal in expectation, this comes at the cost of a significant increase in variance, hindering performance. In the light of this analysis, we introduce a novel method mixing multiple inner steps that enjoys a more accurate and robust meta-gradient signal, essentially trading off bias and variance in meta-gradient estimation. When applied to the Snake game, the mixing meta-gradient algorithm can cut the variance by a factor of 3 while achieving similar or higher performance.
Explore related subjects
Keep this discovery
Clément Bonnet, Paul Caron, Thomas Barrett, Ian Davies, Alexandre Laterre. 2021-10-30. One Step at a Time: Pros and Cons of Multi-Step Meta-Gradient Reinforcement Learning. https://arxiv.org/abs/2111.00206
Cite the original work for its findings. Save a collection to share your selection of sources.