Successor Feature Representations - Inria - Institut national de recherche en sciences et technologies du numérique Access content directly
Journal Articles Transactions on Machine Learning Research Journal Year : 2023

Successor Feature Representations

Abstract

Transfer in Reinforcement Learning aims to improve learning performance on target tasks using knowledge from experienced source tasks. Successor Representations (SR) and their extension Successor Features (SF) are prominent transfer mechanisms in domains where reward functions change between tasks. They reevaluate the expected return of previously learned policies in a new target task to transfer their knowledge. The SF framework extended SR by linearly decomposing rewards into successor features and a reward weight vector allowing their application in high-dimensional tasks. But this came with the cost of having a linear relationship between reward functions and successor features, limiting its application to such tasks. We propose a novel formulation of SR based on learning the cumulative discounted probability of successor features, called Successor Feature Representations (SFR). Crucially, SFR allows to reevaluate the expected return of policies for general reward functions. We introduce different SFR variations, prove its convergence, and provide a guarantee on its transfer performance. Experimental evaluations based on SFR with function approximation demonstrate its advantage over SF not only for general reward functions but also in the case of linearly decomposable reward functions.

Dates and versions

hal-03426870 , version 1 (12-11-2021)

Licence

Identifiers

Cite

Chris Reinke, Xavier Alameda-Pineda. Successor Feature Representations. Transactions on Machine Learning Research Journal, 2023, pp.1-35. ⟨hal-03426870⟩
59 View
0 Download

Altmetric

Share

Gmail Mastodon Facebook X LinkedIn More