通过注意力调控提升小样本增量学习的泛化能力
CASP: Few-Shot Class-Incremental Learning with CLS Token Attention Steering Prompts
- 利用CLS token注意力机制动态调节自注意力权重
- 在多个数据集上实现领先性能,且无需增量阶段微调
- 适合资源受限场景下的持续学习应用
小样本增量学习(FSCIL)要求模型在极少样本下快速适应新类别,同时避免灾难性遗忘。现有基于提示的方法虽有进展,但在极端少样本条件下仍面临泛化能力不足的问题。本文受CLS token类人注意力机制启发,提出CLS Token Attention Steering Prompts(CASP),在查询、键和值投影中引入可学习的共享偏置参数,显式调控自注意力权重。为增强泛化,还设计了注意力扰动策略,并在浅层特征空间执行流形令牌混合,合成潜在新类特征以保留未来任务的表征能力。在CUB200、CIFAR100和ImageNet-R数据集上的实验表明,CASP在标准与细粒度FSCIL设置下均优于现有方法,且增量阶段无需微调,参数开销显著降低。
原文摘要 · Abstract (English)
Few-shot class-incremental learning (FSCIL) presents a core challenge in continual learning, requiring models to rapidly adapt to new classes with very limited samples while mitigating catastrophic forgetting. Recent prompt-based methods, which integrate pretrained backbones with task-specific prompts, have made notable progress. However, under extreme few-shot incremental settings, the model's ability to transfer and generalize becomes critical, and it is thus essential to leverage pretrained knowledge to learn feature representations that can be shared across future categories during the base session. Inspired by the mechanism of the CLS token, which is similar to human attention and progressively filters out task-irrelevant information, we propose the CLS Token Attention Steering Prompts (CASP). This approach introduces class-shared trainable bias parameters into the query, key, and value projections of the CLS token to explicitly modulate the self-attention weights. To further enhance generalization, we also design an attention perturbation strategy and perform Manifold Token Mixup in the shallow feature space, synthesizing potential new class features to improve generalization and reserve the representation capacity for upcoming tasks. Experiments on the CUB200, CIFAR100, and ImageNet-R datasets demonstrate that CASP outperforms state-of-the-art methods in both standard and fine-grained FSCIL settings without requiring fine-tuning during incremental phases and while significantly reducing the parameter overhead.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。