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

Memory-efficient Continual Learning with Neural Collapse Contrastive

About

Contrastive learning has significantly improved representation quality, enhancing knowledge transfer across tasks in continual learning (CL). However, catastrophic forgetting remains a key challenge, as contrastive based methods primarily focus on "soft relationships" or "softness" between samples, which shift with changing data distributions and lead to representation overlap across tasks. Recently, the newly identified Neural Collapse phenomenon has shown promise in CL by focusing on "hard relationships" or "hardness" between samples and fixed prototypes. However, this approach overlooks "softness", crucial for capturing intra-class variability, and this rigid focus can also pull old class representations toward current ones, increasing forgetting. Building on these insights, we propose Focal Neural Collapse Contrastive (FNC^2), a novel representation learning loss that effectively balances both soft and hard relationships. Additionally, we introduce the Hardness-Softness Distillation (HSD) loss to progressively preserve the knowledge gained from these relationships across tasks. Our method outperforms state-of-the-art approaches, particularly in minimizing memory reliance. Remarkably, even without the use of memory, our approach rivals rehearsal-based methods, offering a compelling solution for data privacy concerns.

Trung-Anh Dang, Vincent Nguyen, Ngoc-Son Vu, Christel Vrain• 2024

Related benchmarks

TaskDatasetResultRank
Class-incremental learningCIFAR-10 Seq
Final Average Accuracy (FAA)75.51
53
Image ClassificationCIFAR-10 Seq
Final Average Accuracy85.94
52
Image ClassificationSeq-CIFAR-100
Accuracy56.43
52
Image ClassificationSeq-Tiny-ImageNet
Final Average Accuracy36.32
44
Task-Incremental LearningSeq-CIFAR-10
FAA85.94
28
Task-Incremental LearningCIFAR-100 Seq
FAA56.43
28
Task-Incremental LearningTiny ImageNet Seq
FF39
27
Class-incremental learningTinyImageNet Seq
Average Accuracy20.31
25
Task-Incremental LearningSeq-Tiny-ImageNet
Average Accuracy53.46
25
Task-Incremental LearningCIFAR-10 Seq
Average Accuracy96.14
25
Showing 10 of 16 rows

Other info

Follow for update