Search papers, labs, and topics across Lattice.
This paper investigates the necessity of prompt selection in task-free online continual learning, finding that existing selection strategies often fail to choose appropriate prompts. They propose SinglePrompt, a simplified approach that uses a single, fixed prompt in each self-attention block combined with cosine similarity-based logits and masking of unexposed classes. SinglePrompt achieves state-of-the-art performance on various online continual learning benchmarks, demonstrating that complex prompt selection may be unnecessary.
Ditch the prompt selection circus: a single, well-placed prompt can outperform adaptive strategies in task-free continual learning.
Task-free online continual learning has recently emerged as a realistic paradigm for addressing continual learning in dynamic, real-world environments, where data arrive in a non-stationary stream without clear task boundaries and can only be observed once. To consider such challenging scenarios, many recent approaches have employed prompt selection, an adaptive strategy that selects prompts from a pool based on input signals. However, we observe that such selection strategies often fail to select appropriate prompts, yielding suboptimal results despite additional training of key parameters. Motivated by this observation, we propose a simple yet effective SinglePrompt that eliminates the need for prompt selection and focuses on classifier optimization. Specifically, we simply (i) inject a single prompt into each self-attention block, (ii) employ a cosine similarity-based logit design to alleviate the forgetting effect inherent in the classifier weights, and (iii) mask logits for unexposed classes in the current minibatch. With this simple task-free design, our framework achieves state-of-the-art performance across various online continual learning benchmarks. Source code is available at https://github.com/efficient-learning-lab/SinglePrompt.