arXiv:2412.15571cs.LGcs.AI2024-12AAAI被引 15

用核方法增强基础模型特征,实现无重放缓存的持续学习

Continual Learning Using a Kernel-Based Method Over Foundation Models

  • 基于核函数和随机傅里叶特征增强基础模型特征
  • 仅用类均值与共享协方差矩阵更新,避免灾难性遗忘
  • 无需回放数据即达联合训练精度,适合资源受限场景

持续学习(CL)需增量式地学习一系列任务。本文研究具有挑战性的类别增量学习(CIL)设置,其面临两大难题:灾难性遗忘(CF)和跨任务类别分离(ICS)。尽管已有诸多方法,这些问题仍难以根除。本文提出一种新方法——核线性判别分析(KLDA),可有效缓解上述问题。该方法仅利用基础模型(FM)提取的强大特征,但直接使用这些特征表现不佳。为此,KLDA引入径向基函数(RBF)核及其随机傅里叶特征(RFF)以增强特征表示。当新任务到来时,仅计算每类的均值,并基于核化特征更新所有已学类别的共享协方差矩阵,分类采用线性判别分析。在文本与图像分类数据集上的实证评估表明,KLDA显著优于基线方法。尤为突出的是,在不依赖回放数据的情况下,其性能接近所有类别联合训练的上限,后者被视为CIL性能的理论最优。代码已开源:https://github.com/salehmomeni/klda。

原文摘要 · Abstract (English)

Continual learning (CL) learns a sequence of tasks incrementally. This paper studies the challenging CL setting of class-incremental learning (CIL). CIL has two key challenges: catastrophic forgetting (CF) and inter-task class separation (ICS). Despite numerous proposed methods, these issues remain persistent obstacles. This paper proposes a novel CIL method, called Kernel Linear Discriminant Analysis (KLDA), that can effectively avoid CF and ICS problems. It leverages only the powerful features learned in a foundation model (FM). However, directly using these features proves suboptimal. To address this, KLDA incorporates the Radial Basis Function (RBF) kernel and its Random Fourier Features (RFF) to enhance the feature representations from the FM, leading to improved performance. When a new task arrives, KLDA computes only the mean for each class in the task and updates a shared covariance matrix for all learned classes based on the kernelized features. Classification is performed using Linear Discriminant Analysis. Our empirical evaluation using text and image classification datasets demonstrates that KLDA significantly outperforms baselines. Remarkably, without relying on replay data, KLDA achieves accuracy comparable to joint training of all classes, which is considered the upper bound for CIL performance. The KLDA code is available at https://github.com/salehmomeni/klda.

持续学习核方法基础模型无回放

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。