Maintaining Plasticity in Continual Learning via Regenerative Regularization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kumar, Saurabh, Marklund, Henrik, Van Roy, Benjamin
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909363391692800
author Kumar, Saurabh
Marklund, Henrik
Van Roy, Benjamin
author_facet Kumar, Saurabh
Marklund, Henrik
Van Roy, Benjamin
contents In continual learning, plasticity refers to the ability of an agent to quickly adapt to new information. Neural networks are known to lose plasticity when processing non-stationary data streams. In this paper, we propose L2 Init, a simple approach for maintaining plasticity by incorporating in the loss function L2 regularization toward initial parameters. This is very similar to standard L2 regularization (L2), the only difference being that L2 regularizes toward the origin. L2 Init is simple to implement and requires selecting only a single hyper-parameter. The motivation for this method is the same as that of methods that reset neurons or parameter values. Intuitively, when recent losses are insensitive to particular parameters, these parameters should drift toward their initial values. This prepares parameters to adapt quickly to new tasks. On problems representative of different types of nonstationarity in continual supervised learning, we demonstrate that L2 Init most consistently mitigates plasticity loss compared to previously proposed approaches.
format Preprint
id arxiv_https___arxiv_org_abs_2308_11958
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Maintaining Plasticity in Continual Learning via Regenerative Regularization
Kumar, Saurabh
Marklund, Henrik
Van Roy, Benjamin
Machine Learning
Artificial Intelligence
In continual learning, plasticity refers to the ability of an agent to quickly adapt to new information. Neural networks are known to lose plasticity when processing non-stationary data streams. In this paper, we propose L2 Init, a simple approach for maintaining plasticity by incorporating in the loss function L2 regularization toward initial parameters. This is very similar to standard L2 regularization (L2), the only difference being that L2 regularizes toward the origin. L2 Init is simple to implement and requires selecting only a single hyper-parameter. The motivation for this method is the same as that of methods that reset neurons or parameter values. Intuitively, when recent losses are insensitive to particular parameters, these parameters should drift toward their initial values. This prepares parameters to adapt quickly to new tasks. On problems representative of different types of nonstationarity in continual supervised learning, we demonstrate that L2 Init most consistently mitigates plasticity loss compared to previously proposed approaches.
title Maintaining Plasticity in Continual Learning via Regenerative Regularization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2308.11958