Layers Matter: Why Continual Learning Regularization Should Be Layer-Adaptive
Organizations: German Research Center for Artificial Intelligence (DFKI) · RPTU University Kaiserslautern-Landau
Abstract
Continual learning regularizers like EWC fight forgetting by penalizing changes from previous-task parameters with per-parameter importance, typically diagonal Fisher values. Per-parameter looks more flexible than per-layer, but each layer's diagonal Fisher is a weak summary of its actual curvature, missing the top-eigenvalue information that controls forgetting. Adversarial bit-flip attacks and Hessian-spectrum studies show that this missing per-layer sensitivity spans orders of magnitude in neural networks. Under a block-diagonal Hessian assumption, the layer-level analogue of EWC's existing diagonal assumption, we prove three things. Forgetting decomposes as a sum of per-layer terms weighted by each layer's top Hessian eigenvalue. Diagonal-Fisher weights cannot recover this eigenvalue. For instance, two layers with identical Fisher averages can have top eigenvalues differing by a factor as large as the layer width. For the same level of forgetting, uniform regularization loses new-task performance by an amount scaling with the layer condition number. Our theoretical analysis leads to a simple recipe: protect early layers strongly, let deeper layers move. We apply this recipe to EWC and SLCA and show clear improvements in average performance and forgetting metrics.
Figures & tables
| ImageNet-21k pretrain | MoCoV3 pretrain | ||||
|---|---|---|---|---|---|
| Method | Memory-free | Last-acc | Inc-acc | Last-acc | Inc-acc |
| Joint training (upper bound) | – | 93.22 | — | 89.11 | — |
| GDumb [ 36 ] | 81.92 | 89.46 | 69.72 | 80.95 | |
| DER++ [ 3 ] | 84.50 | 91.49 | 63.64 | 79.55 | |
| BiC [ 54 ] | 88.45 | 93.37 | 80.57 | 89.39 | |
| L2P [ 53 ] | 82.76 | 88.48 | — | — | |
Appendix figures & tables23 assets
Supplementary material from the paper’s appendix.
Appendix
| Backbone | mean adjacent | |||
|---|---|---|---|---|
| ResNet-18 | 6 | 55.1 | 0.975 | 9.4 |
| ResNet-34 | 6 | 75.4 | 0.967 | 12.9 |
| ResNet-50 | 6 | 195.3 | 1.211 † | 27.7 † |
| ResNet-101 | 6 | 181.4 | 1.337 † | 23.6 † |
| ViT-B/16 | 14 | 31.5 | 0.867 | 2.6 |
| ViT-L/16 | 26 | 22.3 | 0.764 | 1.11 |
| Backbone | blockdiag/full | -bound/full | bound holds |
|---|---|---|---|
| MediumCNN | 3/3 | ||
| ResNet-18 | 3/3 | ||
| ResNet-50 | 3/3 |
| ResNet-50 | ViT-B/16 | ||||
|---|---|---|---|---|---|
| block | block | ||||
| stem | 9,536 | 39.05 | patch_embed | 590,592 | 78.21 |
| layer1 | 215,808 | 3,167.48 | block0 | 7,087,872 | 10.70 |
| layer2 | 1,219,584 | 519.36 | block1 | 7,087,872 | 2.85 |
| layer3 | 7,098,368 | 139.32 | block2-11 | 7M each | 3–7 |
| layer4 | 14,964,736 | 57.73 | head | 76,900 | 6.61 |
| Backbone | Params (M) | Most-sensitive | Least-sensitive | ||
|---|---|---|---|---|---|
| ResNet-18 | 6 | 11.2 | 55.1 | layer1 | stem |
| ResNet-34 | 6 | 21.3 | 75.4 | layer3 | stem |
| ResNet-50 | 6 | 23.7 | 195.3 | layer1 | fc |
| ResNet-101 | 6 | 42.7 | 181.4 | layer1 | layer4 |
| ViT-B/16 | 14 | 85.7 | 31.5 | patch_embed | block1 |
| ViT-L/16 | 26 | 303.2 | 22.3 | block9 | block0 |
| Pretraining | Schedule | Last-acc | Inc-acc | Forgetting | |
|---|---|---|---|---|---|
| ImageNet-21k | uniform ( , ) | 5 | |||
| ImageNet-21k | geometric, ours ( , ) | 5 | |||
| ImageNet-21k | measured ( ) | 3 | |||
| MoCoV3 | uniform ( , ) | 3 | |||
| MoCoV3 | geometric, ours ( , ) | 3 | |||
| MoCoV3 | measured ( ) | 3 |
| uniform ( ) | measured- schedule | |
| ResNet-50 / Split-CIFAR-100 | ||
| 50 | — | |
| 200 | ||
| 1000 | ||
| 5000 | ||
| ResNet-18 / Split-CIFAR-100 | ||
| Arm | ResNet-18 | ResNet-50 |
|---|---|---|
| uniform ( ) | ||
| static measured | ||
| online | ||
| online |
| Method | Avg-acc | Forgetting |
|---|---|---|
| SeqFT | ||
| Uniform EWC ( ) | ||
| Measured- EWC ( ) |
| Experiment | # jobs | Wallclock/job | GPU-h (approx.) |
|---|---|---|---|
| EWC sweep, SmallCNN/CIFAR-10 (3 seeds, 6 4 grid) | 72 | 5–10 min | 10 |
| EWC sweep, MediumCNN/CIFAR-100 (3 seeds, 6 4 grid) | 72 | 10–20 min | 20 |
| EWC sweep, ResNet-18/CIFAR-100 (3 seeds, 6 4 grid) | 72 | 20–40 min | 40 |
| EWC sweep, ResNet-50/CIFAR-100 (3 seeds, 6 4 grid) | 72 | 30–60 min | 60 |
| EWC measured- , ResNet-18/CIFAR-100 (3 seeds, 4 , Appendix K ) | 12 | 20–40 min | 7 |
| EWC measured- , ResNet-50/CIFAR-100 (3 seeds, 4 , Appendix K ) | 12 | 30–60 min | 10 |
| Method | MediumCNN | ResNet-18 | ResNet-50 |
|---|---|---|---|
| ER ( ) | |||
| ER ( ) | |||
| ER ( ) | |||
| EWC uniform (best ) | |||
| EWC geometric (best , ) | |||
| ER+EWC ( , , ) |