Assessing Generalization in TD methods for Deep Reinforcement LearningDownload PDF

25 Sept 2019 (modified: 05 May 2023)ICLR 2020 Conference Blind SubmissionReaders: Everyone
TL;DR: Empirical investigation showing TD, in particular TD(0), may be preventing generalization in DeepRL
Abstract: Current Deep Reinforcement Learning (DRL) methods can exhibit both data inefficiency and brittleness, which seem to indicate that they generalize poorly. In this work, we experimentally analyze this issue through the lens of memorization, and show that it can be observed directly during training. More precisely, we find that Deep Neural Networks (DNNs) trained with supervised tasks on trajectories capture temporal structure well, but DNNs trained with TD(0) methods struggle to do so, while using TD(lambda) targets leads to better generalization.
Code: https://anonymous.4open.science/r/366905cc-d617-4467-bb12-de95c08deaad/
Keywords: reinforcement learning, deep learning, generalization
Original Pdf: pdf
9 Replies

Loading