Spectral normalisation for deep reinforcement learning: an optimisation perspective
File(s)2105.05246v1.pdf (2.28 MB)
Accepted version
Author(s)
Type
Conference Paper
Abstract
Most of the recent deep reinforcement learning
advances take an RL-centric perspective and focus on refinements of the training objective. We
diverge from this view and show we can recover
the performance of these developments not by
changing the objective, but by regularising the
value-function estimator. Constraining the Lipschitz constant of a single layer using spectral
normalisation is sufficient to elevate the performance of a Categorical-DQN agent to that of a
more elaborated RAINBOW agent on the challenging Atari domain. We conduct ablation studies
to disentangle the various effects normalisation
has on the learning dynamics and show that is
sufficient to modulate the parameter updates to
recover most of the performance of spectral normalisation. These findings hint towards the need
to also focus on the neural component and its
learning dynamics to tackle the peculiarities of
Deep Reinforcement Learning.
advances take an RL-centric perspective and focus on refinements of the training objective. We
diverge from this view and show we can recover
the performance of these developments not by
changing the objective, but by regularising the
value-function estimator. Constraining the Lipschitz constant of a single layer using spectral
normalisation is sufficient to elevate the performance of a Categorical-DQN agent to that of a
more elaborated RAINBOW agent on the challenging Atari domain. We conduct ablation studies
to disentangle the various effects normalisation
has on the learning dynamics and show that is
sufficient to modulate the parameter updates to
recover most of the performance of spectral normalisation. These findings hint towards the need
to also focus on the neural component and its
learning dynamics to tackle the peculiarities of
Deep Reinforcement Learning.
Editor(s)
Meila, M
Zhang, T
Date Issued
2021-07-01
Date Acceptance
2021-07-01
Citation
INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 139, 2021, 139, pp.1-11
ISSN
2640-3498
Publisher
JMLR-JOURNAL MACHINE LEARNING RESEARCH
Start Page
1
End Page
11
Journal / Book Title
INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 139
Volume
139
Sponsor
Wellcome Trust
Identifier
http://gateway.webofknowledge.com/gateway/Gateway.cgi?GWVersion=2&SrcApp=PARTNER_APP&SrcAuth=LinksAMR&KeyUT=WOS:000683104603068&DestLinkType=FullRecord&DestApp=ALL_WOS&UsrCustomerID=1ba7043ffcc86c417c072aa74d649202
Grant Number
200790/Z/16/Z
Source
International Conference on Machine Learning (ICML)
Subjects
Science & Technology
Technology
Computer Science, Artificial Intelligence
Computer Science
ENVIRONMENT
Publication Status
Published
Start Date
2021-07-18
Finish Date
2021-07-24
Coverage Spatial
ELECTR NETWORK
Date Publish Online
2021-07-18