通过协方差约束提升少样本增量学习的类别分离能力。
Covariance-based Space Regularization for Few-shot Class Incremental Learning
- 用协方差约束统一各类别分布,避免特征空间混淆
- 在三个基准上达到新最佳性能,显著减少遗忘
- 方法简单易集成,适合需要持续学习的场景
少样本增量学习(FSCIL)要求模型在仅有少量标注数据的新类别上持续学习,同时保留旧类知识。由于新类别数据稀少,模型易过拟合并导致旧类灾难性遗忘。现有方法多依赖原型机制约束旧类分布、增强新类区分性,但仍存在特征空间划分不清的问题,使新类与旧类混淆或彼此难以区分。本文从协方差角度出发,提出一种简单的协方差约束损失,强制模型学习具有相同协方差矩阵的类别分布。同时引入特征空间扰动策略,对少数样本进行扰动,使其远离其他类加权分布。将扰动样本视为新类数据,迫使分类器建立明确类别边界。该方法可无缝集成至现有FSCIL框架中。在三个基准测试中验证了有效性,达到当前最优性能。
原文摘要 · Abstract (English)
Few-shot Class Incremental Learning (FSCIL) presents a challenging yet realistic scenario, which requires the model to continually learn new classes with limited labeled data (i.e., incremental sessions) while retaining knowledge of previously learned base classes (i.e., base sessions). Due to the limited data in incremental sessions, models are prone to overfitting new classes and suffering catastrophic forgetting of base classes. To tackle these issues, recent advancements resort to prototype-based approaches to constrain the base class distribution and learn discriminative representations of new classes. Despite the progress, the limited data issue still induces ill-divided feature space, leading the model to confuse the new class with old classes or fail to facilitate good separation among new classes. In this paper, we aim to mitigate these issues by directly constraining the span of each class distribution from a covariance perspective. In detail, we propose a simple yet effective covariance constraint loss to force the model to learn each class distribution with the same covariance matrix. In addition, we propose a perturbation approach to perturb the few-shot training samples in the feature space, which encourages the samples to be away from the weighted distribution of other classes. Regarding perturbed samples as new class data, the classifier is forced to establish explicit boundaries between each new class and the existing ones. Our approach is easy to integrate into existing FSCIL approaches to boost performance. Experiments on three benchmarks validate the effectiveness of our approach, achieving a new state-of-the-art performance of FSCIL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。