Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Benechehab, Abdelhakim, Thomas, Albert, Kégl, Balázs
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909093434753024
author Benechehab, Abdelhakim
Thomas, Albert
Kégl, Balázs
author_facet Benechehab, Abdelhakim
Thomas, Albert
Kégl, Balázs
contents We consider the problem of offline reinforcement learning where only a set of system transitions is made available for policy optimization. Following recent advances in the field, we consider a model-based reinforcement learning algorithm that infers the system dynamics from the available data and performs policy optimization on imaginary model rollouts. This approach is vulnerable to exploiting model errors which can lead to catastrophic failures on the real system. The standard solution is to rely on ensembles for uncertainty heuristics and to avoid exploiting the model where it is too uncertain. We challenge the popular belief that we must resort to ensembles by showing that better performance can be obtained with a single well-calibrated autoregressive model on the D4RL benchmark. We also analyze static metrics of model-learning and conclude on the important model properties for the final performance of the agent.
format Preprint
id arxiv_https___arxiv_org_abs_2402_02858
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning
Benechehab, Abdelhakim
Thomas, Albert
Kégl, Balázs
Machine Learning
We consider the problem of offline reinforcement learning where only a set of system transitions is made available for policy optimization. Following recent advances in the field, we consider a model-based reinforcement learning algorithm that infers the system dynamics from the available data and performs policy optimization on imaginary model rollouts. This approach is vulnerable to exploiting model errors which can lead to catastrophic failures on the real system. The standard solution is to rely on ensembles for uncertainty heuristics and to avoid exploiting the model where it is too uncertain. We challenge the popular belief that we must resort to ensembles by showing that better performance can be obtained with a single well-calibrated autoregressive model on the D4RL benchmark. We also analyze static metrics of model-learning and conclude on the important model properties for the final performance of the agent.
title Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning
topic Machine Learning
url https://arxiv.org/abs/2402.02858