Target Networks and Over-Parameterization Stabilize Off-Policy Bootstrapping with Function Approximation
Abstract
We prove that the combination of a target network and over-parameterized linear function approximation establishes a weaker convergence condition for bootstrapped value estimation in certain cases, even with off-policy data. Our condition is naturally satisfied for expected updates over the entire state-action space or learning with a batch of complete trajectories from episodic Markov decision processes. Notably, using only a target network or an over-parameterized model does not provide such a convergence guarantee. Additionally, we extend our results to learning with truncated trajectories, showing that convergence is achievable for all tasks with minor modifications, akin to value truncation for the final states in trajectories. Our primary result focuses on temporal difference estimation for prediction, providing high-probability value estimation error bounds and empirical analysis on Baird’s counterexample and a Four-room task. Furthermore, we explore the control setting, demonstrating that similar convergence conditions apply to Q-learning.
Cite
Text
Che et al. "Target Networks and Over-Parameterization Stabilize Off-Policy Bootstrapping with Function Approximation." International Conference on Machine Learning, 2024.Markdown
[Che et al. "Target Networks and Over-Parameterization Stabilize Off-Policy Bootstrapping with Function Approximation." International Conference on Machine Learning, 2024.](https://mlanthology.org/icml/2024/che2024icml-target/)BibTeX
@inproceedings{che2024icml-target,
title = {{Target Networks and Over-Parameterization Stabilize Off-Policy Bootstrapping with Function Approximation}},
author = {Che, Fengdi and Xiao, Chenjun and Mei, Jincheng and Dai, Bo and Gummadi, Ramki and Ramirez, Oscar A and Harris, Christopher K and Mahmood, A. Rupam and Schuurmans, Dale},
booktitle = {International Conference on Machine Learning},
year = {2024},
pages = {6372-6396},
volume = {235},
url = {https://mlanthology.org/icml/2024/che2024icml-target/}
}