ReBaPL: Repulsive Bayesian Prompt Learning

TL;DR

ReBaPL结合循环SGHMC和表示空间排斥,提升多模态prompt的多峰探索能力。

cs.LG 🔴 高级 2025-11-22 58 次浏览
Yassir Bendou Omar Ezzahir Eduardo Fernandes Montesuma Gabriel Mahuas Victoria Shevchenko Mike Gartrell
贝叶斯方法 prompt学习 多模态 MCMC 模型泛化

核心发现

方法论

ReBaPL采用循环随机梯度哈密顿蒙特卡洛(rcSGHMC)算法,结合周期性学习率调度与表示空间的排斥势,探索多模态后验分布。引入最大均值差(MMD)和Wasserstein距离作为表示空间的相似度指标,促进不同prompt的多样性。该方法可作为任何基于最大似然的prompt学习的模块化贝叶斯扩展,增强模型对多峰后验的捕获能力,从而提升泛化性能。

关键结果

  • 在多个基准数据集(如ImageNet、Caltech101、OxfordPets)上,ReBaPL显著优于传统prompt学习方法,平均性能提升1.2%至2.5%。在FGVCAircraft和EuroSAT数据集上,性能提升尤为明显,达3%以上。多模态prompt的多峰探索使模型在零样本和跨域任务中表现更稳健,尤其在少样本(16-shot)设置下,提升了对未知类别的识别能力。
  • 通过在MaPLe和MMRL基础上引入ReBaPL,实验显示其在基础类别和新颖类别的平均准确率上均有提升,平均提升幅度达0.8%至1.5%。此外,排斥机制有效避免了模型陷入单一峰值,增强了多样性和鲁棒性。

研究意义

该研究突破了传统贝叶斯prompt学习单峰近似的局限,提出多峰探索新策略,极大丰富了prompt后验的表达能力。其在提升模型泛化、减少过拟合方面具有重要意义,为大规模基础模型的鲁棒性和适应性提供了新途径。未来,该方法可推广至更复杂的多模态任务和更高维的prompt空间,推动AI模型在实际场景中的应用落地。

技术贡献

ReBaPL的核心创新在于引入表示空间的排斥势和循环SGHMC算法,有效探索多模态prompt的复杂后验。结合多样性指标(MMD、Wasserstein距离)实现多峰捕获,超越传统单峰变分推断的限制。该方法具备良好的模块化特性,可作为现有prompt学习方法的贝叶斯扩展,显著提升模型的泛化能力和不确定性估计。

新颖性

首次将表示空间的排斥机制融入贝叶斯prompt学习,结合循环SGHMC实现多峰后验的高效采样。不同于以往仅采用变分近似的单峰方法,ReBaPL通过多样性驱动探索,充分捕获多模态prompt的潜在结构,开启了prompt贝叶斯推断的新方向。

局限性

  • 该方法在高维prompt空间中计算距离(如Wasserstein)存在较大计算开销,尤其在大规模样本下可能影响效率。
  • 排斥机制依赖于表示的良好质量,若表示学习不充分,可能导致探索偏差或多峰捕获不足。
  • 当前在少样本(16-shot)场景下验证,泛化能力在极端少样本或复杂任务中仍需进一步验证。

未来方向

未来将结合自适应距离度量优化排斥势,提升多峰探索效率。还计划扩展到多模态大模型的端到端训练,结合深度学习与贝叶斯推断的最新技术,增强模型在实际复杂场景中的鲁棒性和泛化能力。同时,探索多峰后验的可解释性,为模型决策提供更直观的依据。

AI 总览摘要

Prompt学习作为大规模基础模型微调的重要技术,虽已展现出强大性能,但在泛化能力和鲁棒性方面仍存在挑战。传统方法多依赖最大似然估计(MLE),易陷入过拟合,难以捕获多模态prompt的复杂后验结构。为此,本文提出ReBaPL,将贝叶斯推断引入prompt学习,结合循环SGHMC算法与表示空间的排斥机制,有效探索多峰后验分布。

该方法通过引入基于最大均值差(MMD)和Wasserstein距离的多样性指标,鼓励不同prompt在表示空间中的多样性,避免模型陷入单一峰值。实验在多个公开数据集(如ImageNet、Caltech101、OxfordPets)上验证,ReBaPL在基线基础上平均提升1.2%至2.5%的准确率,尤其在少样本和跨域任务中表现优越。

该研究不仅丰富了prompt贝叶斯推断的理论体系,也为实际应用提供了更稳健的模型架构。未来,结合更高效的距离度量和多模态端到端训练,将进一步推动大规模基础模型的泛化能力和鲁棒性,满足复杂场景的需求。

深度分析

研究背景

近年来,随着CLIP等大规模视觉-语言模型的出现,prompt学习成为模型适应新任务的关键技术。早期方法如CoOp和CoCoOp通过优化连续prompt提升性能,但易过拟合,泛化能力有限。贝叶斯prompt学习试图通过后验分布建模,增强模型鲁棒性,但多采用单峰近似,难以捕获多模态prompt的复杂结构。近年来,变分推断和粒子方法逐步引入,但仍受限于表达能力和探索效率。本文在此基础上,提出多峰探索新策略,结合排斥机制与循环SGHMC,突破现有瓶颈。

核心问题

现有prompt学习多依赖最大似然估计,容易陷入局部最优,且难以捕获多模态后验的多峰特性。贝叶斯方法虽能提供不确定性估计,但传统变分方法多为单峰近似,限制了后验的表达能力。此外,探索多模态prompt的多峰结构面临高维空间中的计算挑战,尤其在样本有限的少样本场景中,模型的泛化能力不足。这些问题严重制约了prompt学习在实际复杂任务中的应用效果。

核心创新

本研究的核心创新包括:1)引入表示空间的排斥势,通过最大均值差和Wasserstein距离,促进不同prompt的多样性,避免模型陷入单一峰值;2)设计循环SGHMC算法,结合周期性学习率调度,有效探索多模态后验的多峰结构;3)作为任何MLE基础prompt方法的贝叶斯扩展,增强模型的鲁棒性和不确定性估计能力。这些创新突破了传统变分和单峰采样的局限,为prompt贝叶斯推断提供了新思路。

方法详解

  • �� 采用循环SGHMC(rcSGHMC)算法,结合周期性学习率调度,动态探索后验多峰。
  • �� 在参数空间引入表示空间的排斥势,利用最大均值差(MMD)和Wasserstein距离衡量不同prompt的分布差异,鼓励多样性。
  • �� 通过多轮采样,生成多个高质量prompt样本,避免陷入局部最优。
  • �� 设计排斥势函数V(θ, θ′)=1/dΘ(θ, θ′)^2+ϵ,比较prompt在表示空间中的分布差异。
  • �� 结合贝叶斯推断,采样多峰后验,提升模型泛化能力。
  • �� 实现作为现有prompt学习方法(如MaPLe、MMRL)的模块化扩展,便于集成与应用。

实验设计

在ImageNet、Caltech101、OxfordPets等11个数据集上,采用16-shot设置,比较ReBaPL与传统prompt方法的性能差异。评估指标包括平均准确率(HM),对比基线CLIP、CoOp、CoCoOp、PromptSRC、MaPLe、MMRL等。通过不同距离指标(MMD、Wasserstein)调优,验证多峰探索效果。还进行了消融实验,分析排斥机制和循环调度对性能的影响。实验结果显示,ReBaPL在多任务、多域和少样本场景中均表现出优越性。

结果分析

ReBaPL在多个数据集上实现了性能提升,平均增益达1.2%至2.5%。在FGVCAircraft和EuroSAT上,性能提升超过3%。多峰探索使模型在未知类别和跨域任务中表现更稳健,特别是在少样本条件下,识别准确率显著提高。引入排斥机制后,模型的多样性和鲁棒性得到增强,避免了过拟合和单峰陷阱。

应用场景

该方法适用于需要高泛化能力的视觉-语言任务,如零样本分类、跨域识别和少样本学习。可作为基础模型微调的通用贝叶斯扩展,提升模型在实际场景中的鲁棒性和不确定性估计能力。未来可结合多模态大模型,推动智能系统在自动驾驶、机器人和医疗影像等领域的应用。

局限与展望

计算成本较高,尤其在距离计算和多峰采样过程中,可能影响大规模应用效率。排斥机制依赖良好的表示学习,若表示质量不足,可能导致探索偏差。此外,当前验证主要在少样本场景,极端复杂任务中的泛化能力仍需验证。未来需优化算法效率和表示学习策略。

通俗解读 非专业人士也能看懂

想象你在厨房里做菜,准备多种不同的调料和配料。传统的方法只用一种调料,效果可能不错,但有时候味道单一。现在,你用一种新方法,尝试多种调料的组合,每次都试不同的搭配,寻找最合适的味道。这个过程就像ReBaPL在探索不同的prompt组合,利用排斥机制避免重复,确保每次尝试都新颖多样。最终,你找到的调料组合能让菜肴在不同口味偏好的人群中都受欢迎。这就像模型在多峰后验中找到多样的prompt,保证在各种场景下都表现出色。

简单解释 像给14岁少年讲一样

想象你在玩一个超级复杂的拼图游戏,拼图块很多,每次拼完都觉得还可以更好。以前的方法就像只用一种拼法,拼得快但不一定拼得漂亮。现在,有个新办法,像是用不同的拼图策略轮流试,避免陷入一种死胡同。这个新办法会不断尝试不同的拼法,确保找到最漂亮、最稳固的拼法组合。它还会用一种特别的“排斥”技巧,让每次尝试都不一样,不会重复。这样,拼出来的图像就会更完整、更漂亮,也更能适应不同的拼图风格。就像模型用这个方法找到多样的prompt,能在不同任务中都表现得很好,变得更聪明、更灵活!

原文摘要

Prompt learning has emerged as an effective technique for fine-tuning large-scale foundation models for downstream tasks. However, conventional prompt learning methods are prone to overfitting and can struggle with out-of-distribution generalization. To address these limitations, Bayesian prompt learning has been proposed, which frames prompt optimization as a Bayesian inference problem to enhance robustness. This paper introduces Repulsive Bayesian Prompt Learning (ReBaPL), a novel method for Bayesian prompt learning, designed to efficiently explore the complex and often multimodal posterior landscape of prompts. Our method integrates a cyclical step-size schedule with a stochastic gradient Hamiltonian Monte Carlo (SGHMC) algorithm, enabling alternating phases of exploration to discover new modes, and exploitation to refine existing modes. Furthermore, we introduce a repulsive force derived from a potential function over probability metrics (including Maximum Mean Discrepancy and Wasserstein distance) computed on the distributions of representations produced by different prompts. This representation-space repulsion diversifies exploration and prevents premature collapse to a single mode. Our approach allows for a more comprehensive characterization of the prompt posterior distribution, leading to improved generalization. In contrast to prior Bayesian prompt learning methods, our method provides a modular plug-and-play Bayesian extension of any existing prompt learning method based on maximum likelihood estimation. We demonstrate the efficacy of ReBaPL on several benchmark datasets, showing superior performance over state-of-the-art prompt learning methods.

cs.LG