A Generalized Bootstrap Target for Value-Learning, Efficiently Combining Value and Feature Predictions

Authors

  • Anthony GX-Chen Mila - Quebec AI Institute McGill University
  • Veronica Chelu Mila - Quebec AI Institute McGill University
  • Blake A. Richards Mila - Quebec AI Institute McGill University Canada AI Chair program, CIFAR
  • Joelle Pineau Mila - Quebec AI Institute McGill University Canada AI Chair program, CIFAR Meta AI Research

DOI:

https://doi.org/10.1609/aaai.v36i6.20639

Keywords:

Machine Learning (ML)

Abstract

Estimating value functions is a core component of reinforcement learning algorithms. Temporal difference (TD) learning algorithms use bootstrapping, i.e. they update the value function toward a learning target using value estimates at subsequent time-steps. Alternatively, the value function can be updated toward a learning target constructed by separately predicting successor features (SF)—a policy-dependent model—and linearly combining them with instantaneous rewards. We focus on bootstrapping targets used when estimating value functions, and propose a new backup target, the ?-return mixture, which implicitly combines value-predictive knowledge (used by TD methods) with (successor) feature-predictive knowledge—with a parameter ? capturing how much to rely on each. We illustrate that incorporating predictive knowledge through an ??-discounted SF model makes more efficient use of sampled experience, compared to either extreme, i.e. bootstrapping entirely on the value function estimate, or bootstrapping on the product of separately estimated successor features and instantaneous reward models. We empirically show this approach leads to faster policy evaluation and better control performance, for tabular and nonlinear function approximations, indicating scalability and generality.

Downloads

Published

2022-06-28

How to Cite

GX-Chen, A., Chelu, V., Richards, B. A., & Pineau, J. (2022). A Generalized Bootstrap Target for Value-Learning, Efficiently Combining Value and Feature Predictions. Proceedings of the AAAI Conference on Artificial Intelligence, 36(6), 6829-6837. https://doi.org/10.1609/aaai.v36i6.20639

Issue

Section

AAAI Technical Track on Machine Learning I