LGAINov 28, 2024

DESIRE: Dynamic Knowledge Consolidation for Rehearsal-Free Continual Learning

arXiv:2411.19154v16 citationsh-index: 34Has Code
Originality Incremental advance
AI Analysis

This addresses catastrophic forgetting in continual learning for AI systems, though it is incremental as it builds on existing LoRA-based approaches.

The paper tackles the problem of information leakage in continual learning by proposing DESIRE, a rehearsal-free method that achieves state-of-the-art performance on multiple datasets, effectively balancing stability and plasticity.

Continual learning aims to equip models with the ability to retain previously learned knowledge like a human. Recent work incorporating Parameter-Efficient Fine-Tuning has revitalized the field by introducing lightweight extension modules. However, existing methods usually overlook the issue of information leakage caused by the fact that the experiment data have been used in pre-trained models. Once these duplicate data are removed in the pre-training phase, their performance can be severely affected. In this paper, we propose a new LoRA-based rehearsal-free method named DESIRE. Our method avoids imposing additional constraints during training to mitigate catastrophic forgetting, thereby maximizing the learning of new classes. To integrate knowledge from old and new tasks, we propose two efficient post-processing modules. On the one hand, we retain only two sets of LoRA parameters for merging and propose dynamic representation consolidation to calibrate the merged feature representation. On the other hand, we propose decision boundary refinement to address classifier bias when training solely on new class data. Extensive experiments demonstrate that our method achieves state-of-the-art performance on multiple datasets and strikes an effective balance between stability and plasticity. Our code will be publicly available.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes