Skip to main navigation Skip to search Skip to main content

VPT-NSP2++: Importance-Aware Visual Prompt Tuning in Null Space for Continual Learning

  • Shizhou Zhang
  • , Yue Lu
  • , De Cheng
  • , Yinghui Xing
  • , Nannan Wang
  • , Peng Wang
  • , Yanning Zhang
  • Northwestern Polytechnical University Xian
  • Xidian University

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

Continual learning (CL) enables AI models to adapt to evolving environments while mitigating catastrophic forgetting, which is a critical capability for dynamic real-world applications. With the growing popularity of pre-trained Vision Transformer (ViT) models and visual prompt tuning (VPT) technique in CL, this work explores a CL method on top of the ViT-based foundation model, through VPT mechanism with theoretical guarantees. Inspired by the orthogonal projection method, we aim to leverage this approach for VPT to enhance CL performance, particularly in long-term scenarios. However, since the orthogonal projection is originally designed for linear operations in CNNs, applying it to ViTs poses challenges induced by the non-linear self-attention mechanism and the distribution drift within LayerNorm. To address these issues, we deduced two orthogonality conditions to achieve the prompt gradient orthogonal projection, which provide a theoretical guarantee of maintaining stability. Considering the strict orthogonal constraints can diminish model capacity and reduce plasticity, we further propose an importance-aware orthogonal regularization framework. By applying varying degrees of orthogonal constraints to different parameters based on their importance to old and new tasks, the framework adaptively enhances model capacity and thereby promotes long-sequence CL while improving the stability-plasticity trade-off. To implement the proposed approach, a null-space-based approximation solution is employed to efficiently achieve the prompt gradient orthogonal projection. Extensive experiments on various class-incremental learning benchmarks demonstrate that our method achieves state-of-the-art performance across diverse CL scenarios.

Original languageEnglish
Pages (from-to)4318-4335
Number of pages18
JournalIEEE Transactions on Pattern Analysis and Machine Intelligence
Volume48
Issue number4
DOIs
StatePublished - 2026

Keywords

  • Continual learning
  • catastrophic forgetting
  • null space
  • orthogonal projection
  • visual prompt tuning

Fingerprint

Dive into the research topics of 'VPT-NSP2++: Importance-Aware Visual Prompt Tuning in Null Space for Continual Learning'. Together they form a unique fingerprint.

Cite this