t-DGR: A Trajectory-Based Deep Generative Replay Method for Continual Learning in Decision Making

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Yue, William, Liu, Bo, Stone, Peter
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910489733234688
author Yue, William
Liu, Bo
Stone, Peter
author_facet Yue, William
Liu, Bo
Stone, Peter
contents Deep generative replay has emerged as a promising approach for continual learning in decision-making tasks. This approach addresses the problem of catastrophic forgetting by leveraging the generation of trajectories from previously encountered tasks to augment the current dataset. However, existing deep generative replay methods for continual learning rely on autoregressive models, which suffer from compounding errors in the generated trajectories. In this paper, we propose a simple, scalable, and non-autoregressive method for continual learning in decision-making tasks using a generative model that generates task samples conditioned on the trajectory timestep. We evaluate our method on Continual World benchmarks and find that our approach achieves state-of-the-art performance on the average success rate metric among continual learning methods. Code is available at https://github.com/WilliamYue37/t-DGR.
format Preprint
id arxiv_https___arxiv_org_abs_2401_02576
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle t-DGR: A Trajectory-Based Deep Generative Replay Method for Continual Learning in Decision Making
Yue, William
Liu, Bo
Stone, Peter
Machine Learning
Artificial Intelligence
Neural and Evolutionary Computing
Deep generative replay has emerged as a promising approach for continual learning in decision-making tasks. This approach addresses the problem of catastrophic forgetting by leveraging the generation of trajectories from previously encountered tasks to augment the current dataset. However, existing deep generative replay methods for continual learning rely on autoregressive models, which suffer from compounding errors in the generated trajectories. In this paper, we propose a simple, scalable, and non-autoregressive method for continual learning in decision-making tasks using a generative model that generates task samples conditioned on the trajectory timestep. We evaluate our method on Continual World benchmarks and find that our approach achieves state-of-the-art performance on the average success rate metric among continual learning methods. Code is available at https://github.com/WilliamYue37/t-DGR.
title t-DGR: A Trajectory-Based Deep Generative Replay Method for Continual Learning in Decision Making
topic Machine Learning
Artificial Intelligence
Neural and Evolutionary Computing
url https://arxiv.org/abs/2401.02576