atlas

Research map / Continual learning / Domain and prompt-based

Vector Quantization Prompting for Continual Learning

NeurIPS 2024 · Continual learning › Domain and prompt-based · approach: Prompt-based continual learning

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes VQ-Prompt, a prompt-based class-incremental continual learning method that quantizes a soft-selected continuous prompt to its nearest discrete element in a prompt pool, trained end-to-end with straight-through gradients and VQ/commitment losses plus classifier bias mitigation via Gaussian representation statistics.

Problem

Prompt-based continual learning relies on non-differentiable prompt identity selection, so prompt selection cannot be optimized with task loss; prior differentiable alternatives produce continuous prompts that lack abstraction for task knowledge.

Approach

Form a continuous prompt as a softmax-weighted sum over a prompt pool, then replace it with its nearest prompt-pool element (vector quantization), training end-to-end via straight-through gradient estimation plus VQ and commitment losses, so discrete prompts are optimized with task loss.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace continuous soft prompt selection with VQ discrete prompt selection (ablation)VQ-Prompt on ImageNet-R 10-task · ImageNet-R 10-taskFAA: 77.15 (Soft-Prompt, $\tau\!=\!1.0$) -> 78.05 (VQ-Prompt-S)Helped
add representation statistics for classifier bias mitigation (ablation)VQ-Prompt-S on ImageNet-R · ImageNet-R 10-taskFAA: 78.00 $\pm$ 0.39 -> 78.71 $\pm$ 0.22Helped
add representation statistics for classifier bias mitigation (ablation)VQ-Prompt-S on ImageNet-R · ImageNet-R 5-taskFAA: 78.52 $\pm$ 0.34 -> 79.23 $\pm$ 0.29Helped
add representation statistics for classifier bias mitigation (ablation)VQ-Prompt-S on ImageNet-R · ImageNet-R 20-taskFAA: 76.19 $\pm$ 0.26 -> 78.10 $\pm$ 0.22Helped
add representation statistics for classifier bias mitigation (ablation)L2P++ on ImageNet-R · ImageNet-R 5-taskFAA: 70.83 $\pm$ 0.58 -> 74.11 $\pm$ 0.08Helped
add representation statistics for classifier bias mitigation (ablation)L2P++ on ImageNet-R · ImageNet-R 10-taskFAA: 69.29 $\pm$ 0.73 -> 72.93 $\pm$ 0.27Helped
add representation statistics for classifier bias mitigation (ablation)L2P++ on ImageNet-R · ImageNet-R 20-taskFAA: 65.89 $\pm$ 1.30 -> 70.99 $\pm$ 0.26Helped
reduce prompt length $L_p$ to excessively small values (ablation)VQ-Prompt · ImageNet-R 10-taskFAAHurt
increase prompt pool size $N$ while prompt length is small (ablation)VQ-Prompt · ImageNet-R 10-taskFAAHelped
increase prompt length $L_p$ beyond a certain threshold (ablation)VQ-Prompt · ImageNet-R 10-taskFAAMixed
vary VQ and commitment loss weights (ablation)VQ-Prompt · ImageNet-R 10-taskFAANo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-R CIFAR-100 CUB-200-2011 ImageNet-1K ImageNet-21K ImageNet-A VTAB

Benchmarks: ImageNet-R Split CIFAR-100 Split CUB-200 ImageNet-A VTAB

Metrics: FAA CAA

Baselines: FT FT++ L2P++ Deep L2P++ DualPrompt CODA-Prompt HiDe-Prompt EvoPrompt LwF BiC SLCA

Builds on: CODA-Prompt L2P DualPrompt VQ-VAE prefix tuning

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy