Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Exemplar-Free Continual Learning for State Space Models

About

State-Space Models (SSMs) excel at capturing long-range dependencies with structured recurrence, making them well-suited for sequence modeling. However, their evolving internal states pose challenges in adapting them under Continual Learning (CL). This is particularly difficult in exemplar-free settings, where the absence of prior data leaves updates to the dynamic SSM states unconstrained, resulting in catastrophic forgetting. To address this, we propose Inf-SSM, a novel and simple geometry-aware regularization method that utilizes the geometry of the infinite-dimensional Grassmannian to constrain state evolution during CL. Unlike classical continual learning methods that constrain weight updates, Inf-SSM regularizes the infinite-horizon evolution of SSMs encoded in their extended observability subspace. We show that enforcing this regularization requires solving a matrix equation known as the Sylvester equation, which typically incurs $\mathcal{O}(n^3)$ complexity. We develop a $\mathcal{O}(n^2)$ solution by exploiting the structure and properties of SSMs. This leads to an efficient regularization mechanism that can be seamlessly integrated into existing CL methods. Comprehensive experiments on challenging benchmarks, including ImageNet-R and Caltech-256, demonstrate a significant reduction in forgetting while improving accuracy across sequential tasks.

Isaac Ning Lee, Leila Mahmoodi, Trung Le, Mehrtash Harandi• 2025

Related benchmarks

TaskDatasetResultRank
Exemplar-Free Class-Incremental LearningCIFAR-100 5 tasks
AA45.18
13
Exemplar-Free Class-Incremental LearningCaltech-256 5-Tasks
Average Accuracy (AA)50.75
13
Exemplar-Free Class-Incremental LearningCIFAR-100 10 tasks
Average Accuracy (AA)26.53
13
Exemplar-Free Class-Incremental LearningCaltech-256 10-Tasks
Average Accuracy (AA)39.88
13
Exemplar-Free Class-Incremental LearningImageNet-R 5-task (test)
Average Accuracy (AA)49.34
12
Exemplar-Free Class-Incremental LearningImageNet-R 5-Tasks Scenario
Average Accuracy (AA)49.34
7
Exemplar-Free Class-Incremental LearningImageNet-R 10-Tasks Scenario
Average Accuracy (AA)43.82
7
Exemplar-Free Class-Incremental LearningImageNet-R 10-task (test)
Average Accuracy (AA)27.92
6
Exemplar-Free Class-Incremental LearningImageNet-R 10 tasks
Average Accuracy (AA)43.82
6
Showing 9 of 9 rows

Other info

Follow for update