On the Theory of Continual Learning with Gradient Descent for Neural Networks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Taheri, Hossein, Ghosh, Avishek, Mazumdar, Arya
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915944006156288
author Taheri, Hossein
Ghosh, Avishek
Mazumdar, Arya
author_facet Taheri, Hossein
Ghosh, Avishek
Mazumdar, Arya
contents Continual learning, the ability of a model to adapt to an ongoing sequence of tasks without forgetting earlier ones, is a central goal of artificial intelligence. To better understand its underlying mechanisms, we study the limitations of continual learning in a tractable yet representative setting. Specifically, we analyze one-hidden-layer quadratic neural networks trained by gradient descent on a sequence of XOR-cluster datasets with Gaussian noise, where different tasks correspond to clusters with orthogonal means. Our analysis is based on a tight characterization of gradient descent dynamics for the training loss, which yields explicit bounds on the rate of train-time forgetting as functions of the number of iterations, sample size, number of tasks, and hidden-layer width. We then leverage an algorithmic stability framework to bound the generalization gap, leading to corresponding guarantees on test-time forgetting. Together, our results provide the first closed-form guarantees for forgetting in continual learning with neural networks and show how key problem parameters jointly govern forgetting dynamics. Numerical experiments corroborate our theoretical results.
format Preprint
id arxiv_https___arxiv_org_abs_2510_05573
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle On the Theory of Continual Learning with Gradient Descent for Neural Networks
Taheri, Hossein
Ghosh, Avishek
Mazumdar, Arya
Machine Learning
Information Theory
Continual learning, the ability of a model to adapt to an ongoing sequence of tasks without forgetting earlier ones, is a central goal of artificial intelligence. To better understand its underlying mechanisms, we study the limitations of continual learning in a tractable yet representative setting. Specifically, we analyze one-hidden-layer quadratic neural networks trained by gradient descent on a sequence of XOR-cluster datasets with Gaussian noise, where different tasks correspond to clusters with orthogonal means. Our analysis is based on a tight characterization of gradient descent dynamics for the training loss, which yields explicit bounds on the rate of train-time forgetting as functions of the number of iterations, sample size, number of tasks, and hidden-layer width. We then leverage an algorithmic stability framework to bound the generalization gap, leading to corresponding guarantees on test-time forgetting. Together, our results provide the first closed-form guarantees for forgetting in continual learning with neural networks and show how key problem parameters jointly govern forgetting dynamics. Numerical experiments corroborate our theoretical results.
title On the Theory of Continual Learning with Gradient Descent for Neural Networks
topic Machine Learning
Information Theory
url https://arxiv.org/abs/2510.05573