Model Zoo: A Growing "Brain" That Learns Continually

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ramesh, Rahul, Chaudhari, Pratik
Formato: Preprint
Publicado: 2021
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917657427574784
author Ramesh, Rahul
Chaudhari, Pratik
author_facet Ramesh, Rahul
Chaudhari, Pratik
contents This paper argues that continual learning methods can benefit by splitting the capacity of the learner across multiple models. We use statistical learning theory and experimental analysis to show how multiple tasks can interact with each other in a non-trivial fashion when a single model is trained on them. The generalization error on a particular task can improve when it is trained with synergistic tasks, but can also deteriorate when trained with competing tasks. This theory motivates our method named Model Zoo which, inspired from the boosting literature, grows an ensemble of small models, each of which is trained during one episode of continual learning. We demonstrate that Model Zoo obtains large gains in accuracy on a variety of continual learning benchmark problems. Code is available at https://github.com/grasp-lyrl/modelzoo_continual.
format Preprint
id arxiv_https___arxiv_org_abs_2106_03027
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle Model Zoo: A Growing "Brain" That Learns Continually
Ramesh, Rahul
Chaudhari, Pratik
Machine Learning
This paper argues that continual learning methods can benefit by splitting the capacity of the learner across multiple models. We use statistical learning theory and experimental analysis to show how multiple tasks can interact with each other in a non-trivial fashion when a single model is trained on them. The generalization error on a particular task can improve when it is trained with synergistic tasks, but can also deteriorate when trained with competing tasks. This theory motivates our method named Model Zoo which, inspired from the boosting literature, grows an ensemble of small models, each of which is trained during one episode of continual learning. We demonstrate that Model Zoo obtains large gains in accuracy on a variety of continual learning benchmark problems. Code is available at https://github.com/grasp-lyrl/modelzoo_continual.
title Model Zoo: A Growing "Brain" That Learns Continually
topic Machine Learning
url https://arxiv.org/abs/2106.03027