Revisiting Weight Regularization for Low-Rank Continual Learning
Organizations: State Key Laboratory of Human-Machine Hybrid Augmented Intelligence Institute of Artificial Intelligence and Robotics, Xi’an Jiaotong University, Shaanxi, China · Computer Vision Center, Barcelona, Spain · Universitat Aut`onoma de Barcelona, Barcelona, Spain · School of Instrument Science and Engineering, Harbin Institute of Technology, Harbin, China ·
Abstract
Continual Learning (CL) with large-scale pre-trained models (PTMs) has recently gained wide attention, shifting the focus from training from scratch to continually adapting PTMs. This has given rise to a promising paradigm: parameter-efficient continual learning (PECL), where task interference is typically mitigated by assigning a task-specific module during training, such as low-rank adapters. However, weight regularization techniques, such as Elastic Weight Consolidation (EWC)-a key strategy in CL-remain underexplored in this new paradigm. In this paper, we revisit weight regularization in low-rank CL as a new perspective for mitigating task interference in PECL. Unlike existing low-rank CL methods, we mitigate task interference by regularizing a shared low-rank update through EWC, thereby keeping the storage requirement and inference costs constant regardless of the number of tasks. Our proposed method EWC-LoRA leverages a low-rank representation to estimate parameter importance over the full-dimensional space. This design offers a practical, computational- and memory-efficient solution for CL with PTMs, and provides insights that may inform the broader application of regularization techniques within PECL. Extensive experiments on various benchmarks demonstrate the effectiveness of EWC-LoRA, achieving a stability-plasticity trade-off superior to existing low-rank CL approaches. These results indicate that, even under low-rank parameterizations, weight regularization remains an effective mechanism for mitigating task interference. Code is available at: https://github.com/yaoyz96/low-rank-cl.
Figures & tables
| Strategy | Avg. | Stability | Plasticity | + Mem. | |
| w/o | 0 GB | ||||
| Precomputed | 1 GB | ||||
| Separate | 4 GB | ||||
| (Ours) | 6 GB |
| Tasks | CIFAR-100 | DomainNet | ImageNet-R | ImageNet-A | ||||
| Methods | ( ) | Avg. ( ) | ( ) | Avg. ( ) | ( ) | Avg. ( ) | ( ) | Avg. ( ) |
| Joint Train | ||||||||
| Finetune | ||||||||
| L2P | ||||||||
| DualPrompt | ||||||||
| CODA-Prompt | ||||||||
| Backbone | Method | Order-1 | Order-2 | Order-3 | Avg. |
| T5-large | O-LoRA | 74.40 | |||
| EWC-LoRA | 78.01 | 76.85 | 76.39 | ||
| LLaMA-3.2-1B-Instruct | O-LoRA | ||||
| TreeLoRA | |||||
| EWC-LoRA | 61.17 | 60.47 | 67.61 | 63.08 |
| Tasks | CIFAR-100 | DomainNet | ImageNet-R | ImageNet-A | ||||
| Methods | Stability | Plasticity | Stability | Plasticity | Stability | Plasticity | Stability | Plasticity |
| Vanilla LoRA | ||||||||
| InfLoRA | ||||||||
| SD-LoRA | ||||||||
| EWC-LoRA | ||||||||
| Methods | Memory | CIFAR-100 | DomainNet | ImageNet-R | ImageNet-A |
| Vanilla LoRA | 18 GB | 10m22s 7s | 32m8s 44s | 13m32s 105s | 0m55s 21s |
| InfLoRA | 20 GB | 11m9s 4s | 42m25s 68s | 14m33s 107s | 1m16s 22s |
| SD-LoRA | 37 GB | 12m16s 86s | 34m18s 160s | 22m53s 232s | 1m56s 36s |
| EWC-LoRA | 24 GB | 10m31s 3s | 33m10s 50s | 13m42s 104s | 0m55s 21s |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| Datasets | Methods |
| CIFAR-100 | optimizer: Adam; schedular: Cosine; batch size: 128; shuffle: False; epochs: 20; rank: 10 lr: 0.0005; classifier lr: 0.005; lr decay: 0.1 lr: 0.008; classifier lr: 0.008; lr decay: 0.1 (SD-LoRA) 0.95 (InfLoRA) |
| DomainNet | optimizer: Adam; schedular: Cosine; batch size: 128; shuffle: False; epochs: 5; rank: 30 lr: 0.0005; classifier lr: 0.005; lr decay: 0.1 lr: 0.02; classifier lr: 0.02; lr decay: 0.0 (SD-LoRA) 0.95 (InfLoRA) |
| ImageNet-R | optimizer: Adam; schedular: Cosine; batch size: 128; shuffle: False; epochs: 50; rank: 10 lr: 0.0005; classifier lr: 0.0005; lr decay: 0.1 lr: 0.01; classifier lr: 0.01; lr decay: 0.0; weight decay: 0.0005 (SD-LoRA) weight decay: 0.005 (EWC-LoRA) 0.98 (InfLoRA) |
| ImageNet-A | optimizer: Adam; schedular: Cosine; batch size: 128; shuffle: False; epochs: 10; rank: 10 lr: 0.0005; classifier lr: 0.005; lr decay: 0.1 0.98 (InfLoRA) |
| 88.22 | ||||||
| Avg. | 92.46 | |||||
| Stability | 94.51 | |||||
| Plasticity | 99.07 |
| Task 1 | Task 2 | Task 3 | Task 4 | |
| Order 1 | dbpedia | amazon | yahoo | ag |
| Order 2 | dbpedia | amazon | ag | yahoo |
| Order 3 | yahoo | amazon | ag | dbpedia |
| Tasks | ImageNet-R (N=5) | ImageNet-R (N=20) | ||
| Methods | ( ) | Avg. ( ) | ( ) | Avg. ( ) |
| Joint Train | ||||
| Finetune | ||||
| InfLoRA | ||||
| SD-LoRA | ||||
| Vanilla LoRA | ||||
| CIFAR-100 | ImageNet-R | |||
| Avg. | Avg. | |||
| 16 | 88.31 | 92.59 | 72.10 | 79.18 |
| 10 | 87.91 | 92.27 | 72.86 | 78.95 |
| 4 | 87.79 | 92.12 | 72.42 | 78.78 |
| 1 | 86.77 | 91.61 | 70.37 | 77.27 |
| CIFAR-100 | ImageNet-R | |||||||
| Avg. | Stability | Plasticity | Avg. | Stability | Plasticity | |||
| 16 | 83.05 | 89.71 | 86.25 | 100.42 | 66.05 | 76.26 | 74.17 | 101.41 |
| 10 | 82.99 | 89.74 | 87.56 | 98.86 | 66.32 | 76.35 | 75.69 | 99.86 |
| 4 | 83.35 | 90.10 | 88.70 | 98.24 | 67.52 | 76.33 | 77.27 | 97.30 |
| 1 | 83.72 | 89.94 | 89.95 | 97.36 | 68.43 | 77.09 | 81.95 | 96.75 |
| Strategy | Avg. | Stability | Plasticity | |
| Uniform | ||||
| Dataset-based |
| Estimation | Avg. | Best | |
| Exact | |||
| Exact (n=500) | |||
| Sample | |||
| Empirical |
Incremental Learning
Incremental learning studies how models update from sequential data while retaining earlier knowledge as classes, tasks, or domains change.
See all 141 papers