Saved in:
Bibliographic Details
Main Authors: Hafez, Muhammad Burhan, Erekmen, Kerim
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.16532
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912179108708352
author Hafez, Muhammad Burhan
Erekmen, Kerim
author_facet Hafez, Muhammad Burhan
Erekmen, Kerim
contents Central to the development of universal learning systems is the ability to solve multiple tasks without retraining from scratch when new data arrives. This is crucial because each task requires significant training time. Addressing the problem of continual learning necessitates various methods due to the complexity of the problem space. This problem space includes: (1) addressing catastrophic forgetting to retain previously learned tasks, (2) demonstrating positive forward transfer for faster learning, (3) ensuring scalability across numerous tasks, and (4) facilitating learning without requiring task labels, even in the absence of clear task boundaries. In this paper, the Task-Agnostic Policy Distillation (TAPD) framework is introduced. This framework alleviates problems (1)-(4) by incorporating a task-agnostic phase, where an agent explores its environment without any external goal and maximizes only its intrinsic motivation. The knowledge gained during this phase is later distilled for further exploration. Therefore, the agent acts in a self-supervised manner by systematically seeking novel states. By utilizing task-agnostic distilled knowledge, the agent can solve downstream tasks more efficiently, leading to improved sample efficiency. Our code is available at the repository: https://github.com/wabbajack1/TAPD.
format Preprint
id arxiv_https___arxiv_org_abs_2411_16532
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Continual Deep Reinforcement Learning with Task-Agnostic Policy Distillation
Hafez, Muhammad Burhan
Erekmen, Kerim
Machine Learning
Central to the development of universal learning systems is the ability to solve multiple tasks without retraining from scratch when new data arrives. This is crucial because each task requires significant training time. Addressing the problem of continual learning necessitates various methods due to the complexity of the problem space. This problem space includes: (1) addressing catastrophic forgetting to retain previously learned tasks, (2) demonstrating positive forward transfer for faster learning, (3) ensuring scalability across numerous tasks, and (4) facilitating learning without requiring task labels, even in the absence of clear task boundaries. In this paper, the Task-Agnostic Policy Distillation (TAPD) framework is introduced. This framework alleviates problems (1)-(4) by incorporating a task-agnostic phase, where an agent explores its environment without any external goal and maximizes only its intrinsic motivation. The knowledge gained during this phase is later distilled for further exploration. Therefore, the agent acts in a self-supervised manner by systematically seeking novel states. By utilizing task-agnostic distilled knowledge, the agent can solve downstream tasks more efficiently, leading to improved sample efficiency. Our code is available at the repository: https://github.com/wabbajack1/TAPD.
title Continual Deep Reinforcement Learning with Task-Agnostic Policy Distillation
topic Machine Learning
url https://arxiv.org/abs/2411.16532