atlas

Research map / Continual learning / Online and replay-based

On the Diminishing Returns of Width for Continual Learning

ICML 2024 · Continual learning › Online and replay-based · approach: Theoretical analysis of forgetting · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Gives a finite-width theoretical bound showing continual-learning error (forgetting) of sparse feed-forward networks decays as W^{-beta} with width, implying diminishing returns, and validates it on rotated MNIST/SVHN/Fashion-MNIST/GTSRB experiments up to width 2^16.

Problem

Prior work observed that wider networks forget less in continual learning, but the exact finite-width relationship between width and catastrophic forgetting was not characterized, and width scaling tested only up to 2048 hidden units.

Approach

Combines a shared-active-row analysis under row-wise sparsity with an empirical assumption that wider networks move less from initialization (relative distance ~ W^{-beta}), then uses perturbation analysis to bound the output difference between sequentially trained models, yielding error O(t W^{-beta} alpha^{(1-2beta)/2}) and hence diminishing returns in width.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Increase widthFeed-forward networks · Rotated MNIST and Rotated Fashion MNISTAverage ForgettingHelped
Increase widthFeed-forward networks · Rotated SVHN, two- and three-layer networksAverage ForgettingHurt
Increase widthFeed-forward networks · Rotated SVHN, one-layer networkAverage ForgettingNo clear effect
Increase widthWide ResNet · Rotated SVHN and Rotated GTSRBAverage ForgettingHelped
Use row-wise sparsity with α=0.1 instead of dense training (ablation)Feed-forward networks · MNIST, width 2^14Average Forgetting: 27.9 -> 0.2Helped
Increase depth from 1 to 3 layers (ablation)Feed-forward networks · Rotated Fashion MNIST and Rotated GTSRBAverage ForgettingHurt
Increase depth (ablation)Feed-forward networks · Rotated MNIST and Rotated SVHNAverage ForgettingNo clear effect
Increase depth beyond 4 hidden layers (ablation)Feed-forward networks · Rotated Fashion MNIST, Rotated SVHN, and Rotated GTSRBAverage ForgettingMixed
Use Adam instead of SGD (ablation)Feed-forward networks · Continual learning benchmarksAverage ForgettingNo clear effect
Increase the number of sequential tasks (ablation)Feed-forward networks · Rotated GTSRB and Rotated Fashion MNISTAccuracyHurt
Apply dropout at probabilities 0.1, 0.2, or 0.3 (ablation)Feed-forward networks · Distance from initialization across widthsDistance from initializationNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: Rotated MNIST Rotated Fashion MNIST Rotated SVHN Rotated GTSRB

Benchmarks: Rotated MNIST Rotated Fashion MNIST Rotated SVHN Rotated GTSRB

Metrics: average forgetting average accuracy learning accuracy joint accuracy

Baselines: Wide ResNet

Builds on: Lazy training Adam Ideal Continual Learner

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy