S-Prompts Learning with Pre-trained Transformers: An Occam's Razor for Domain Incremental Learning

Yabin Wang, Zhiwu Huang, Xiaopeng Hong
7/26/2022
cs.CVcs.LG

Official code: iamwangyabin/S-Prompts (105 stars, MIT, last commit 2024-09-26). 1 more implementation below.

Abstract

State-of-the-art deep neural networks are still struggling to address the catastrophic forgetting problem in continual learning. In this paper, we propose one simple paradigm (named as S-Prompting) and two concrete approaches to highly reduce the forgetting degree in one of the most typical continual learning scenarios, i.e., domain increment learning (DIL). The key idea of the paradigm is to learn prompts independently across domains with pre-trained transformers, avoiding the use of exemplars that commonly appear in conventional methods. This results in a win-win game where the prompting can achieve the best for each domain. The independent prompting across domains only requests one single cross-entropy loss for training and one simple K-NN operation as a domain identifier for inference. The learning paradigm derives an image prompt learning approach and a novel language-image prompt learning approach. Owning an excellent scalability (0.03% parameter increase per domain), the best of our approaches achieves a remarkable relative improvement (an average of about 30%) over the best of the state-of-the-art exemplar-free methods for three standard DIL tasks, and even surpasses the best of them relatively by about 6% in average when they use exemplars. Source code is available at \url{https://github.com/iamwangyabin/S-Prompts}.

View on arXivView PDF

Code Implementations(2)

MIT

Code for NeurIPS 2022 paper “S-Prompts Learning with Pre-trained Transformers: An Occam’s Razor for Domain Incremental Learning“

10513PythonSep 16, 20222 years agoMIT
LAMDA-CL/PyCILCommunity100%
PyTorch

PyCIL: A Python Toolbox for Class-Incremental Learning

1,096158PythonDec 23, 20218 months agoNOASSERTION
continual-learningdeep-learningincremental-learninglifelong-learningmachine-learning+5 more

Similar Papers

Cite this paper

@article{wang2022sprompts,
  title  = {S-Prompts Learning with Pre-trained Transformers: An Occam's Razor for Domain Incremental Learning},
  author = {Yabin Wang and Zhiwu Huang and Xiaopeng Hong},
  year   = {2022},
  eprint = {2207.12819},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2207.12819v2}
}

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/2207.12819.svg)](https://codeofpaper.com/paper/2207.12819)

Discussion