Model Inversion with Layer-Specific Modeling and Alignment for Data-Free Continual Learning
About
Continual learning (CL) aims to incrementally train a model on a sequence of tasks while retaining performance on prior ones. However, storing and replaying data is often infeasible due to privacy or security constraints and impractical for arbitrary pre-trained models. Data-free CL seeks to update models without access to previous data. Beyond regularization, we employ model inversion to synthesize data from the trained model, enabling replay without storing samples. Yet, model inversion in predictive models faces two challenges: (1) generating inputs solely from compressed output labels causes drift between synthetic and real data, and replaying such data can erode prior knowledge; (2) inversion is computationally expensive since each step backpropagates through the full model. These issues are amplified in large pre-trained models such as CLIP. To improve efficiency, we propose Per-layer Model Inversion (PMI), inspired by faster convergence in single-layer optimization. PMI provides strong initialization for full-model inversion, substantially reducing iterations. To mitigate feature shift, we model class-wise features via Gaussian distributions and contrastive model, ensuring alignment between synthetic and real features. Combining PMI and feature modeling, our approach enables continual learning of new classes by generating pseudo-images from semantic-aware projected features, achieving strong effectiveness and compatibility across multiple CL settings.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Class-incremental learning | CIFAR-100 20 tasks | -- | 58 | |
| Task-Incremental Learning | Tiny-ImageNet 20 tasks | Average Accuracy25.67 | 54 | |
| Class-incremental learning | CUB-200 (test) | Alast81.64 | 51 | |
| Task-Incremental Learning | CIFAR-100 20 tasks | Accuracy (ACC)32.6 | 40 | |
| Class-incremental learning | CIFAR100 5 Tasks | Accuracy52.38 | 31 | |
| Class-incremental learning | Tiny-ImageNet 10 tasks | -- | 31 | |
| Incremental Learning | CIFAR-100 (test) | Average Accuracy88.35 | 27 | |
| Class-incremental learning | Tiny-ImageNet 20 tasks | -- | 25 | |
| Incremental Learning | CIFAR-100 10 task | Avg Incremental Acc43.9 | 17 | |
| Incremental Learning | Tiny-ImageNet 5 task | Average Incremental Accuracy37.9 | 15 |