提出CONEC-LoRA,解决领域增量学习中的知识遗忘与分类不准问题。
Continual Knowledge Consolidation LORA for Domain Incremental Learning
- 通过共享与特定任务LoRA的融合提取共性知识与领域特性。
- 在4个基准上性能超越已有方法超5个百分点,显著提升准确率。
- 适合关注持续学习中模型稳定性与泛化能力的研究者。
领域增量学习(DIL)旨在应对新领域不断到来的同时避免灾难性遗忘。尽管参数高效微调(PEFT)方法已出现,但现有工作生成的任务特定LoRA忽略了跨任务间的共享知识,推理时任务特定LoRA选择不当导致准确率显著下降;且依赖线性或原型分类器,泛化能力有限。本文提出持续知识融合低秩适配(CONEC-LoRA),通过任务共享与任务特定LoRA之间的融合,提取共性知识并保留领域特异性信息。不同于以往方法,CONEC-LoRA引入随机分类器,其参数从分布中采样,提高正确分类概率。此外,设计辅助网络以最优预测推理时的任务特定LoRA,并采用不同深度网络结构,每层连接局部分类器以利用中间表示。该模块结合球生成损失与变换模块,缓解合成样本偏差问题。大量实验表明,CONEC-LoRA在4个主流基准上均优于先前方法,性能提升超过5%。
原文摘要 · Abstract (English)
Domain Incremental Learning (DIL) is a sub-branch of continual learning that aims to address the never-ending arrival of new domains without catastrophic forgetting. Despite the advent of parameter-efficient fine-tuning (PEFT) approaches, prior works create task-specific LoRAs that overlook shared knowledge across tasks. Inaccurate selection of task-specific LoRAs during inference leads to significant drops in accuracy, while existing works rely on linear or prototype-based classifiers, which have suboptimal generalization powers. Our paper proposes continual knowledge consolidation low-rank adaptation (CONEC-LoRA) addressing the DIL problems. CONEC-LoRA is developed from consolidations between task-shared LORA to extract common knowledge and task-specific LORA to incorporate domain-specific knowledge. Unlike existing approaches, CONEC-LoRA integrates the concept of a stochastic classifier whose parameters are sampled from a distribution, thus enhancing the likelihood of correct classifications. Last but not least, an auxiliary network is deployed to optimally predict the task-specific LoRAs for inference and implements the concept of a different-depth network structure in which every layer is connected with a local classifier to leverage intermediate representations. This module integrates the ball-generator loss and transformation module to address the synthetic sample bias problem. Our rigorous experiments demonstrate the advantage of CONEC-LoRA over prior art on 4 popular benchmark problems with over 5% margins.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。