Heads Collapse, Features Stay: Why Replay Needs Big Buffers

Abstract

A persistent paradox in continual learning (CL) is that neural networks often retain linearly separable representations of past tasks even when their output predictions fail. We formalize this distinction as the gap between *deep* (feature-space) and *shallow* (classifier-level) forgetting. We reveal a critical asymmetry in Experience Replay: while minimal buffers successfully anchor feature geometry and prevent deep forgetting, mitigating shallow forgetting typically requires substantially larger buffer capacities. To explain this, we extend the Neural Collapse framework to the sequential setting. We characterize deep forgetting as a geometric drift toward out-of-distribution subspaces and prove that any non-zero replay fraction asymptotically guarantees the retention of linear separability. Conversely, we identify that the ``strong collapse'' induced by small buffers leads to rank-deficient covariances and inflated class means, effectively blinding the classifier to true population boundaries. By unifying CL with out-of-distribution detection, our work challenges the prevailing reliance on large buffers, suggesting that explicitly correcting these statistical artifacts could unlock robust performance with minimal replay.

Cite

Text

Lanzillotta et al. "Heads Collapse, Features Stay: Why Replay Needs Big Buffers." International Conference on Learning Representations, 2026.

Markdown

[Lanzillotta et al. "Heads Collapse, Features Stay: Why Replay Needs Big Buffers." International Conference on Learning Representations, 2026.](https://mlanthology.org/iclr/2026/lanzillotta2026iclr-heads/)

BibTeX

@inproceedings{lanzillotta2026iclr-heads,
  title     = {{Heads Collapse, Features Stay: Why Replay Needs Big Buffers}},
  author    = {Lanzillotta, Giulia and Meier, Damiano and Hofmann, Thomas},
  booktitle = {International Conference on Learning Representations},
  year      = {2026},
  url       = {https://mlanthology.org/iclr/2026/lanzillotta2026iclr-heads/}
}