Adaptive Additive Parameter Updates of Vision Transformers for Few-Shot Continual Learning

Kyle Stein, Andrew Arash Mahyari, Guillermo Francia, Eman El-Sheikh
4/11/2025
cs.CV

No official code found. Best implementation: argonauts18/Adaptive-Additive-Parameter-Updates-of-Vision-Transformers-for-Few-Shot-Continual-Learning (0 stars, MIT, last commit 2025-04-25).

Abstract

Integrating new class information without losing previously acquired knowledge remains a central challenge in artificial intelligence, often referred to as catastrophic forgetting. Few-shot class incremental learning (FSCIL) addresses this by first training a model on a robust dataset of base classes and then incrementally adapting it in successive sessions using only a few labeled examples per novel class. However, this approach is prone to overfitting on the limited new data, which can compromise overall performance and exacerbate forgetting. In this work, we propose a simple yet effective novel FSCIL framework that leverages a frozen Vision Transformer (ViT) backbone augmented with parameter-efficient additive updates. Our approach freezes the pre-trained ViT parameters and selectively injects trainable weights into the self-attention modules via an additive update mechanism. This design updates only a small subset of parameters to accommodate new classes without sacrificing the representations learned during the base session. By fine-tuning a limited number of parameters, our method preserves the generalizable features in the frozen ViT while reducing the risk of overfitting. Furthermore, as most parameters remain fixed, the model avoids overwriting previously learned knowledge when small novel data batches are introduced. Extensive experiments on benchmark datasets demonstrate that our approach yields state-of-the-art performance compared to baseline FSCIL methods.

View on arXivView PDF

Code Implementations(1)

Similar Papers

Cite this paper

@article{stein2025adaptive,
  title  = {Adaptive Additive Parameter Updates of Vision Transformers for Few-Shot Continual Learning},
  author = {Kyle Stein and Andrew Arash Mahyari and Guillermo Francia and Eman El-Sheikh},
  year   = {2025},
  eprint = {2504.08982},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2504.08982v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2504.08982.svg)](https://codeofpaper.com/paper/2504.08982)

Discussion