English

Few-Shot Class Incremental Learning with Attention-Aware Self-Adaptive Prompt

Machine Learning 2024-07-18 v3 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Few-Shot Class-Incremental Learning (FSCIL) models aim to incrementally learn new classes with scarce samples while preserving knowledge of old ones. Existing FSCIL methods usually fine-tune the entire backbone, leading to overfitting and hindering the potential to learn new classes. On the other hand, recent prompt-based CIL approaches alleviate forgetting by training prompts with sufficient data in each task. In this work, we propose a novel framework named Attention-aware Self-adaptive Prompt (ASP). ASP encourages task-invariant prompts to capture shared knowledge by reducing specific information from the attention aspect. Additionally, self-adaptive task-specific prompts in ASP provide specific information and transfer knowledge from old classes to new classes with an Information Bottleneck learning objective. In summary, ASP prevents overfitting on base task and does not require enormous data in few-shot incremental tasks. Extensive experiments on three benchmark datasets validate that ASP consistently outperforms state-of-the-art FSCIL and prompt-based CIL methods in terms of both learning new classes and mitigating forgetting.

Keywords

Cite

@article{arxiv.2403.09857,
  title  = {Few-Shot Class Incremental Learning with Attention-Aware Self-Adaptive Prompt},
  author = {Chenxi Liu and Zhenyi Wang and Tianyi Xiong and Ruibo Chen and Yihan Wu and Junfeng Guo and Heng Huang},
  journal= {arXiv preprint arXiv:2403.09857},
  year   = {2024}
}

Comments

ECCV 2024

R2 v1 2026-06-28T15:20:55.492Z