On Robust Reinforcement Learning with Lipschitz-Bounded Policy Networks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Barbara, Nicholas H., Wang, Ruigang, Manchester, Ian R.
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910815826739200
author Barbara, Nicholas H.
Wang, Ruigang
Manchester, Ian R.
author_facet Barbara, Nicholas H.
Wang, Ruigang
Manchester, Ian R.
contents This paper presents a study of robust policy networks in deep reinforcement learning. We investigate the benefits of policy parameterizations that naturally satisfy constraints on their Lipschitz bound, analyzing their empirical performance and robustness on two representative problems: pendulum swing-up and Atari Pong. We illustrate that policy networks with smaller Lipschitz bounds are more robust to disturbances, random noise, and targeted adversarial attacks than unconstrained policies composed of vanilla multi-layer perceptrons or convolutional neural networks. However, the structure of the Lipschitz layer is important. We find that the widely-used method of spectral normalization is too conservative and severely impacts clean performance, whereas more expressive Lipschitz layers such as the recently-proposed Sandwich layer can achieve improved robustness without sacrificing clean performance.
format Preprint
id arxiv_https___arxiv_org_abs_2405_11432
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On Robust Reinforcement Learning with Lipschitz-Bounded Policy Networks
Barbara, Nicholas H.
Wang, Ruigang
Manchester, Ian R.
Machine Learning
Systems and Control
This paper presents a study of robust policy networks in deep reinforcement learning. We investigate the benefits of policy parameterizations that naturally satisfy constraints on their Lipschitz bound, analyzing their empirical performance and robustness on two representative problems: pendulum swing-up and Atari Pong. We illustrate that policy networks with smaller Lipschitz bounds are more robust to disturbances, random noise, and targeted adversarial attacks than unconstrained policies composed of vanilla multi-layer perceptrons or convolutional neural networks. However, the structure of the Lipschitz layer is important. We find that the widely-used method of spectral normalization is too conservative and severely impacts clean performance, whereas more expressive Lipschitz layers such as the recently-proposed Sandwich layer can achieve improved robustness without sacrificing clean performance.
title On Robust Reinforcement Learning with Lipschitz-Bounded Policy Networks
topic Machine Learning
Systems and Control
url https://arxiv.org/abs/2405.11432