Source-linked AI summary

Continual Learning Through Synaptic Intelligence

Friedemann Zenke, Ben Poole, Surya Ganguli

arXiv:1703.04200v3cs.LGq-bio.NCstat.ML

TL;DR

Changing data distributions expose catastrophic forgetting in conventional neural networks, whereas biological networks continually adapt. This paper introduces synapses that estimate their contribution to past tasks and consolidate important parameters during later learning. The approach substantially reduces forgetting while retaining computational efficiency across continual classification tasks.

  • Problem

    Artificial neural networks struggle to learn from changing data distributions without forgetting previously learned tasks.

    Source: §1 · p. 1
  • Method

    The method gives each synapse internal dynamics that estimate task importance online and penalizes changes to important parameters during subsequent tasks.

    Source: §1 · p. 1, §3 · p. 4
  • Results

    Synaptic consolidation retains high accuracy on earlier tasks across nine additional tasks, while solving later tasks nearly as well as joint training and comparably to EWC.

    Source: §5.2 · p. 7
  • Takeaways & Limitations

    Estimating synapse importance and constraining important parameters can enable continual learning with minimal interference to previously learned tasks.

    Source: §6 · p. 7

    The importance estimate can overestimate true parameter contributions under stochastic gradient descent, and its simple relationship to the Hessian does not extend generally beyond quadratic losses.

    Source: §3 · p. 3, §4 · p. 5

Abstract

from arXiv · show

While deep learning has led to remarkable advances across diverse applications, it struggles in domains where the data distribution changes over the course of learning. In stark contrast, biological neural networks continually adapt to changing domains, possibly by leveraging complex molecular machinery to solve many tasks simultaneously. In this study, we introduce intelligent synapses that bring some of this biological complexity into artificial neural networks. Each synapse accumulates task relevant information over time, and exploits this information to rapidly store new memories without forgetting old ones. We evaluate our approach on continual learning of classification tasks, and show that it dramatically reduces forgetting while maintaining computational efficiency.

1. Introduction

Artificial neural networks typically learn on fixed datasets and struggle when data distributions change, unlike biological networks that continually adapt. The paper addresses this gap by giving synapses internal dynamics that track task relevance and protect prior knowledge.

  • ANNs are typically trained on a dataset and then frozen, so adapting to changed data distributions requires retraining and risks catastrophic forgetting.Source: §1 · p. 1
  • Biological neural networks continually acquire knowledge from non-stationary data, suggesting that biological mechanisms absent from ANNs may support continual learning.Source: §1 · p. 1
  • Biological synapses use molecular machinery across spatial and temporal scales, but few studies had shown how this complexity benefits learning in ANNs.Source: §1 · p. 1
  • The proposed synapses track past and current parameters while estimating each synapse’s importance for solving earlier tasks.Source: §1 · p. 1
  • Penalizing changes to important synapses lets future tasks rely mainly on previously unimportant synapses, reducing forgetting during sequential classification learning.Source: §1 · p. 1

2. Prior work

Prior work addresses catastrophic forgetting through architectural, functional, and structural changes. These approaches reduce interference by modifying network structure, preserving outputs, or penalizing parameter changes, with distinct computational and scalability costs.

Architectural approaches

  • Architectural methods reduce task interference by freezing or modifying weights, but network-copying approaches grow in complexity with the number of tasks.Approaches include freezing weights, reducing shared-layer learning rates, changing nonlinearities, sparsifying gradients, and augmenting copied networks.Source: §2 · p. 2

Functional approaches

  • Functional methods preserve old input-output mappings through distillation or hidden-activation regularization, requiring an additional old-network forward pass for each new data point.Source: §2 · p. 2

Structural approaches

  • Structural methods penalize parameter changes using measures such as EWC’s Fisher-weighted quadratic penalty, but exact diagonal Fisher computation scales linearly with output dimensionality.Source: §2 · p. 2

3. Synaptic framework

The synaptic framework assigns each parameter an online importance measure based on its contribution to loss reduction during training, then protects important parameters while learning new tasks. It uses a quadratic surrogate for prior tasks, updated across tasks through cumulative importance and reference weights, while acknowledging noise and multi-task scope limitations.

3. Synaptic framework

  • Online synaptic importance protects parameters that contributed strongly to prior loss reduction, allowing less important parameters to adapt to new tasks.The framework seeks a local, online structural regularizer whose importance measure is maintained at individual synapses or parameters.Source: §3 · pp. 2–4

Importance estimation

  • Parameter-specific importance is computed from gradient-update contributions accumulated along the entire training trajectory, where the path integral equals the total loss change.The gradient field is decomposed by parameter, giving each parameter an interpretable contribution to changes in total loss.Source: §3 · p. 3

Noise and calibration

  • SGD noise typically overestimates per-parameter importance, so the strength parameter is generally chosen below one to compensate.Source: §3 · pp. 3–4

Surrogate loss

  • The quadratic surrogate matches the original task’s loss decrease, net parameter displacement, and endpoint minimum rather than using a Hessian approximation at the minimum.These three conditions uniquely determine the surrogate quadratic loss summarizing the descent trajectory.Source: Figure 2 · p. 3, §3 · p. 3

Regularization

  • The regularized objective pulls important parameters toward their previous-task reference values, with a strength parameter c trading off retention of old memories against learning new tasks.A damping parameter ξ bounds the regularization expression when parameter displacement approaches zero.Source: §3 · pp. 3–4

Implementation

  • Importance estimates are updated continuously during each task, while cumulative importance and reference weights update only at task boundaries before per-task importance is reset.Source: §3 · p. 4

Scope

  • The surrogate-loss motivation is exact for two tasks, although the method is evaluated empirically for learning additional tasks.Source: §3 · p. 4

Relation to EWC

  • Compared with EWC, the framework estimates importance online over the full learning trajectory instead of using a separate end-of-task Fisher-information estimate.Source: §3 · p. 4

4. Theoretical analysis of special cases

The analysis derives the importance matrix for quadratic training trajectories and identifies conditions under which normalized importance matches the Hessian. It also shows why low-rank structure can preserve capacity for future tasks and marks the limitation for general losses.

Quadratic training analysis

  • The time-integrated matrix Q is independent of the descent time constant τ and can be expressed in the Hessian eigenbasis.The derivation uses eigenvalues, eigenvectors, and initial-to-final parameter discrepancies to obtain an explicit formula for Q.Source: §4 · p. 4

Averaged initial conditions

  • After averaging over random initial conditions, Q reduces to the Hessian up to a scale factor removed by the normalization.The scale factor reflects the discrepancy between initial and final conditions, and the normalization removes it at zero damping.Source: §4 · pp. 4–5

Special cases

  • For a diagonal Hessian, normalized Q becomes the diagonal Hessian, while for a rank-1 Hessian it becomes the diagonal of a non-diagonal low-rank Hessian.In both special cases, normalization removes the scale of movement in parameter space.Source: §4 · p. 5

Continual-learning implication

  • Low-rank structure leaves many synaptic-weight directions unconstrained by a task, leaving excess capacity for future-task modification without interfering with old-task performance.This identifies the continual-learning relevance of the low-rank Hessian case.Source: §4 · p. 5

Scope and limitation

  • For general losses with a Hessian that varies along the trajectory, no simple correspondence between importance and the endpoint Hessian is expected, although the measures are empirically correlated.The authors relate this correlation to the comparable effectiveness of endpoint-based measures in later experiments.Source: §4 · p. 5

5. Experiments

Across split MNIST, permuted MNIST, and split CIFAR-10/100, synaptic consolidation was evaluated as a defense against catastrophic forgetting. It preserved performance on earlier tasks while retaining strong performance on new tasks, often approaching or exceeding relevant controls.

5.1. Split MNIST

  • Consolidation preserved high accuracy on earlier split-MNIST tasks, whereas networks without it fell to chance after learning all digits.On Task 2, both models performed near 1, but only consolidation kept average accuracy across Tasks 1 and 2 close to 1; after all tasks, unprotected early-task accuracy dropped to chance.Source: §5.1 · p. 6

5.2. Permuted MNIST benchmark

  • Synaptic consolidation retained high permuted-MNIST accuracy across nine additional tasks and performed only slightly worse than simultaneous training, with results comparable to EWC.The figure compares consolidation and EWC against SGD variants as task count increases.Source: Figure 4 · p. 6, §5.2 · p. 7
  • Without consolidation, second-layer importance correlations increased across permuted-MNIST tasks; consolidation kept these synapse sets largely uncorrelated.The authors associate correlated importances with interference and catastrophic forgetting.Source: §5.2 · pp. 6–7

5.3. Split CIFAR-10/CIFAR-100 benchmark

  • On split CIFAR-10/100, consolidation produced similar validation accuracy across tasks and outperformed non-consolidated training except on the last task.The comparison used a CNN trained first on CIFAR-10 and then on five CIFAR-100 tasks, with five repeated runs.Source: §5.3 · p. 7
  • Consolidated networks matched or exceeded single-task-from-scratch validation accuracy on most tasks, while non-consolidated networks showed the opposite training-versus-validation pattern.The exception was Task 1, CIFAR-10, which had 10× more examples per class.Source: §5.3 · p. 7

6. Discussion

The discussion frames synaptic consolidation as an online alternative to EWC that uses higher-dimensional synaptic dynamics to protect prior knowledge. It presents the model as one specific biologically motivated response to catastrophic forgetting, not a claim that biological synapses operate identically.

6. Discussion

  • The method alleviates catastrophic forgetting by estimating each synapse’s importance for past tasks and penalizing changes to important synapses.This allows novel tasks to be learned with minimal interference to previously learned tasks.Source: §6 · p. 7
  • Unlike EWC’s offline Fisher estimate at a task’s loss minimum, this approach computes consolidation strength online across the entire parameter trajectory.Despite this distinction, the two methods yielded similar permuted-MNIST performance.Source: §6 · p. 7
  • The model requires synapses to function as higher-dimensional dynamical systems that accumulate task-relevant information and retain previous parameter values.The paper contrasts this with artificial synapses represented by single scalar weights.Source: §6 · p. 8
  • The authors present this higher-dimensional synaptic model as one specific solution that motivates broader research into biologically inspired control of learning dynamics.They explicitly make no claim that biological synapses behave like the model.Source: §6 · p. 8

A. Split CIFAR-10/100 CNN architecture

The appendix specifies the split CIFAR-10/100 CNN architecture and its associated hyperparameters. Table 1 organizes these implementation details using m for the number of splits.

A. Split CIFAR-10/100 CNN architecture

B. Additional split CIFAR–10 experiments

The split CIFAR-10 experiment compares consolidation against a control after sequential training on two task groups. Consolidation significantly improves validation performance on both tasks, including a roughly 4.5% advantage on Task B.

B. Additional split CIFAR–10 experiments

other metrics

The paper contrasts local curvature-based sensitivity metrics with an online path-integral importance measure. The path integral follows parameter trajectories and estimates curvature without additional gradient computations.

other metrics

  • Prior sensitivity measures primarily use local objective curvature, with the Hessian carrying computational overhead and potentially being negative definite.The paper motivates alternatives that are more efficient for online use.Source: other metrics · p. 10
  • The empirical Fisher approximates parameter sensitivity using gradients at the current parameters and avoids the model-distribution labels required by the full Fisher.Computing the full Fisher online would require at least one additional backpropagation pass.Source: other metrics · p. 10
  • The path-integral approach evaluates parameter importance over a trajectory rather than at a single final parameter value.This distinction makes it proportional to the Hessian diagonal for a quadratic, whereas the empirical Fisher is zero at the minimum.Source: other metrics · p. 10
Loading 1703.04200v3…