arXiv:2607.04711cs.CV2026-07中稿 · ECCV

用概率分布生成多样化提示,解决持续学习中提示坍缩问题

Learning Probabilistic Prompt for Continual Learning

论文配图:Learning Probabilistic Prompt for Continual Learning
图 1 · 摘自论文原文
  • 将每个提示建模为概率分布,通过混合分布采样实现多样性
  • 在ImageNet-R、CIFAR-100等数据集上显著提升分类准确率
  • 适合研究持续学习与提示工程的学者,尤其关注模型稳定性者

持续学习旨在从一系列包含互不重叠类别的任务中逐步学习,同时保留先前知识。基于提示的持续学习方法通过关联输入图像的查询特征来学习少量参数(即提示),以表征图像的多样化模式。然而,我们观察到现有方法存在提示坍缩问题:提示趋于高度相似,无法捕捉持续学习场景中的多样数据分布。为此,本文提出一种新型提示驱动的持续学习框架,以捕捉任务序列中图像的多样化模式。具体地,我们将每个提示建模为概率分布,并构建这些分布的混合模型,从中采样多样化提示。这使模型能有效捕获持续学习过程中的高度多样性图像分布。此外,我们引入分布正则化损失,防止训练过程中提示分布发生突变。在ImageNet-R、CIFAR-100和CUB-200等标准基准上的大量实验结果验证了该框架的有效性。

原文摘要 · Abstract (English)

Continual learning aims to progressively learn from a sequence of tasks, each containing a disjoint subset of classes, while preserving previously learned knowledge. Prompt-based continual learning methods propose to learn a small set of parameters, i.e., prompts, by associating them with a query feature of an input image. These methods optimize the prompts, attempting to represent diverse patterns of images. However, we have observed that existing prompt-based methods suffer from a prompt collapse problem, that is, the prompts tend to be highly similar to each other, thereby failing to capture the diverse data distributions in continual learning scenarios. To address this issue, we propose in this paper a novel prompt-based continual learning framework that captures diverse patterns of images across a sequence of tasks. To this end, we model each prompt as a probabilistic distribution and construct a mixture of these distributions, from which we sample diverse prompts. This enables our model to effectively capture highly diverse image distributions in the continual learning process. We also present a distribution regularization loss to prevent abrupt changes in the prompt distributions throughout the training process. We show extensive experimental results for continual learning on standard benchmarks, including ImageNet-R, CIFAR-100, and CUB-200, demonstrating the effectiveness of our framework.

持续学习提示学习概率建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。