Co2In: a Bi-level Memory Incremental Learning Framework with Knowledge Encoding, Consolidation, and Integration
Organizations: Key Laboratory of Big Data & Artificial Intelligence in Transportation (Ministry of Education), Beijing Jiaotong University, Beijing, China, 100044. · College of Computing and Data Science, Nanyang Technological University, Singapore, 639798. · Fujian Key Laboratory of Big Data Application and Intellectualization for Tea Industry, Wuyi University, Fujian, China, 354300.
Abstract
Incremental learning (IL) aims to continually acquire new knowledge (plasticity) while retaining previously learned information (stability). However, striking a balance between plasticity and stability remains a significant challenge for intelligent systems. The human brain achieves exceptional balance, owing to various memory units that collaboratively encode and store information. Inspired by human memory mechanisms, this paper introduces an IL framework with knowledge enCoding, Consolidation, and Integration (Co2In). Co2In is designed as a bi-level memory architecture: a working memory for adaptive knowledge acquisition and a long-term memory dedicated to the persistent retention of information. The two memory modules work cooperatively during IL. The working memory first learns from new data to encode knowledge into parameters. Subsequently, Co2In performs a consolidation process to identify underlying patterns in the learned parameters and re-express them into a compact knowledge representation. Next, the knowledge representation and the identified patterns are transformed into separate network layers and integrated into the long-term memory. These designs empower Co2In to accumulate knowledge with high plasticity and stability. We evaluate Co2In on CIFAR-10, CIFAR-100, and Tiny-ImageNet under exemplar-free Class-IL and Task-IL settings. Experimental results show that Co2In achieves state-of-the-art performance with efficient memory consumption.
Figures & tables
| Notation | Description |
| General Notations | |
| norm: sum of absolute values of all elements. | |
| Frobenius norm: for a matrix . | |
| Singular Value Decomposition: , where and are orthogonal, and is diagonal. | |
| The cross-entropy loss. | |
| The concatenation operation. | |
| Methods | Venue | Buffer | CIFAR-10 | CIFAR-100 | Tiny-ImageNet | |||
| Class-IL | Task-IL | Class-IL | Task-IL | Class-IL | Task-IL | |||
| Joint | - | - | 91.29 | 98.07 | 70.31 | 91.18 | 58.07 | 82.01 |
| GEM [ 10 ] | NIPS2017 | 500 | 23.66 1.2 | 85.14 2.1 | 11.76 0.7 | 62.80 2.7 | - | - |
| DER [ 4 ] | NIPS2020 | 500 | 70.57 1.0 | 93.41 0.3 | 34.24 1.4 | 73.26 1.3 | 17.75 1.1 | 51.78 0.9 |
| DER++ [ 4 ] | NIPS2020 | 500 | 73.48 0.3 | 94.10 0.3 | 36.52 2.0 | 74.86 1.1 | 19.38 1.4 | 51.91 0.7 |
| Co2L † [ 25 ] | ICCV2021 | 500 | 74.26 | 95.90 | 30.45 | 79.61 | 20.12 | 53.04 |
| Methods | CIFAR-100 | Tiny-ImageNet | ||
| Class-IL | Task-IL | Class-IL | Task-IL | |
| (a) Co 2 In w/o representation compaction regularizer | 34.75 0.5 | 83.04 0.5 | 27.05 2.0 | 65.00 2.2 |
| (b) Co 2 In w/o basis updating | 6.88 5.9 | 36.48 17.7 | 2.90 1.0 | 27.23 3.6 |
| (c) Co 2 In w/o retraining | 36.12 0.7 | 81.78 3.0 | 28.91 0.7 | 65.68 1.9 |
| (d) Co 2 In w/o quadratic classifier | 16.90 0.5 | 79.65 0.4 | 8.81 0.1 | 69.31 0.5 |
| Co 2 In | 37.54 0.9 | 84.21 1.0 | 31.23 0.5 | 68.75 0.7 |
| Dataset | Methods | Class-IL | Task-IL |
| CIFAR-10 | OWM [ 69 ] | 47.50 0.5 | 81.00 0.1 |
| PASS [ 70 ] | 53.12 0.3 | 86.07 0.2 | |
| EFT [ 51 ] | 55.12 1.9 | 95.09 0.3 | |
| ABD [ 14 ] | 57.64 0.3 | 95.11 0.3 | |
| FAS [ 71 ] | 43.36 2.8 | 90.89 1.3 | |
| DCPOC [ 31 ] | 44.30 0.6 | 90.43 0.3 |
| Method | Partition | CIFAR-100 | |
| Class-IL (%) | Task-IL (%) | ||
| Co 2 In | 10-Split | 37.54 | 84.21 |
| Co 2 In w/ RCA | 10-Split | 29.58 | 84.18 |
| Co 2 In | 25-Split | 21.75 | 91.45 |
| Co 2 In w/ RCA | 25-Split | 27.37 | 91.03 |
| Co 2 In | 50-Split | 5.02 | 91.53 |
| Method | CIFAR-100 | Tiny-ImageNet | ||
| Class-IL | Task-IL | Class-IL | Task-IL | |
| LWF [ 33 ] | 5.57 | 5.57 | 2.22 | 2.22 |
| DER++ [ 4 ] | 5.57 | 5.57 | 2.22 | 2.22 |
| Co 2 In w/o knowledge integration | 1.45 | 1.45 | 3.23 | 3.23 |
| Co 2 In | 3.49 | 3.49 | 1.43 | 1.43 |
| Methods | 5-split CIFAR-100 | 10-split CIFAR-100 | 20-split CIFAR-100 |
| DGR [ 28 ] † | 14.4 | 8.1 | 4.1 |
| LwF [ 33 ] † | 17.0 | 9.2 | 4.7 |
| DI [ 29 ] † | 18.8 | 10.9 | 5.7 |
| ABD [ 14 ] † | 43.9 | 33.7 | 20.0 |
| PASS [ 70 ] | 45.2 | 30.8 | 17.4 |
| CRHD [ 74 ] | 46.1 | 34.3 | 21.8 |
| Methods | 20-split CIFAR-100 | 25-split Tiny-ImageNet |
| InstAParam † [ 75 ] | 51.04 | 34.64 |
| PackNet † [ 42 ] | 67.50 | 52.93 |
| BLIP † [ 76 ] | 68.17 | 47.63 |
| GEM † [ 10 ] | 68.89 | - |
| A-GEM † [ 11 ] | 61.91 | 53.32 |
| EWC † [ 38 ] | 71.66 | 52.33 |
| Method | Classifier | CIFAR-100 | Tiny-ImageNet | ||
| Class-IL | Task-IL | Class-IL | Task-IL | ||
| LWF [ 33 ] | Linear | 10.05 0.6 | 63.78 4.3 | 9.36 0.1 | 58.61 1.8 |
| LWF [ 33 ] | Quadratic | FAIL | FAIL | FAIL | FAIL |
| DER++ [ 4 ] | Linear | 7.42 0.5 | 62.21 2.6 | 6.77 0.1 | 60.91 1.3 |
| DER++ [ 4 ] | Quadratic | FAIL | FAIL | FAIL | FAIL |
| GPM [ 5 ] | Linear | 8.19 0.1 | 70.93 0.9 | 7.04 0.1 | 59.84 0.2 |
| Number of Expanded Basis | Linear Classifier | Quadratic Classifier |
| 0 | 8.98 | 8.50 |
| 1 | 12.96 | 24.85 |
| 2 | 13.84 | 36.14 |
| 4 | 14.17 | 36.70 |
| 8 | 14.86 | 37.10 |
| No Restriction | 16.90 | 37.54 |
| Method | Backbone | CIFAR-100 | Tiny-ImageNet | ||||||||
| Class-IL | Task-IL | Memory (MB) | Class-IL | Task-IL | Memory (MB) | ||||||
| Accuracy (%) | Flops | Accuracy (%) | Flops | Accuracy (%) | Flops | Accuracy (%) | Flops | ||||
| LWF [ 33 ] | standard ResNet18 | 10.05 | 5.57 | 63.78 | 5.57 | 42.50 | 9.36 | 2.22 | 58.61 | 2.22 | 42.50 |
| DER++ [ 4 ] | standard ResNet18 | 36.52 | 5.57 | 74.86 | 5.57 | 48.35 | 19.38 | 2.22 | 51.91 | 2.22 | 65.94 |
| Co 2 In | standard ResNet18 | 39.25 | 3.24 | 84.78 | 3.24 | 309.32 | 33.25 | 1.19 | 70.08 | 1.19 | 270.81 |
| Co 2 In | narrower ResNet18 | 37.54 | 3.49 | 84.21 | 3.49 | 25.53 | 31.23 | 1.43 | 68.75 | 1.43 | 23.74 |
| CIFAR-100 | |||||
| Non-decomp. layers | Class-IL | Task-IL | Memory (MB) | ||
| Accuracy (%) | Flops | Accuracy (%) | Flops | ||
| All decomposed | 37.54 | 3.49 | 84.21 | 3.49 | 25.53 |
| First 5 | 35.87 | 2.62 | 83.37 | 4.34 | 25.46 |
| First 10 | 35.60 | 2.13 | 81.09 | 5.03 | 23.95 |
| First 15 | 31.46 | 1.45 | 75.33 | 5.53 | 21.19 |
| Method | CIFAR-100 | Tiny-ImageNet | ||||
| Class-IL | Task-IL | Memory | Class-IL | Task-IL | Memory | |
| ACMap [ 59 ] | 89.91% | 98.35% | 63.28 MB | 86.01% | 96.12% | 63.28 MB |
| InfoLoRA [ 60 ] | 85.83% | 98.62% | 63.28 MB | 84.62% | 93.58% | 63.28 MB |
| Naive LoRA | 83.44% | 98.03% | 63.28 MB | 79.30% | 91.42% | 63.28 MB |
| Co 2 In (ours) | 85.96% | 98.74% | 63.98 MB | 84.38% | 96.72% | 60.38 MB |