提升CLIP少样本适配的不确定性估计,让模型更可信。
BayesAdapter: enhanced uncertainty estimation in CLIP few-shot adaptation
- 用贝叶斯推断替代点估计,捕捉参数空间变异性
- 在多个任务上实现更好校准与选择性分类性能
- 适合需要可靠置信度评估的落地应用
大规模预训练视觉语言模型(VLM)带来了机器学习范式变革,其中最流行的模型之一CLIP在分类任务中展现出卓越的零样本和迁移学习能力。为将CLIP迁移到下游任务,适配器(adapters)提供了一种参数高效的方法,无需对大模型进行反向传播(不同于提示学习)。然而,现有CLIP适配器主要关注判别性能,其不确定性估计质量被忽视。本文发现,当前最优适配器的判别性能与不确定性估计能力并不总相关,而后者对真实场景的安全部署至关重要。我们进一步表明,其中一个适配器可通过更通用的概率框架中的最大后验(MAP)推断获得。基于此,我们提出BayesAdapter,利用贝叶斯推断估计完整概率分布而非单一点估计,更充分地捕捉参数空间的变异性。在全面的实证评估中,我们的方法在预测不确定性估计方面表现优异,在校准性和选择性分类上显著领先。代码将在论文接收后公开。
原文摘要 · Abstract (English)
The emergence of large pre-trained vision-language models (VLMs) represents a paradigm shift in machine learning, with unprecedented results in a broad span of visual recognition tasks. CLIP, one of the most popular VLMs, has exhibited remarkable zero-shot and transfer learning capabilities in classification. To transfer CLIP to downstream tasks, adapters constitute a parameter-efficient approach that avoids backpropagation through the large model (unlike related prompt learning methods). However, CLIP adapters have been developed to target discriminative performance, and the quality of their uncertainty estimates has been overlooked. In this work we show that the discriminative performance of state-of-the-art CLIP adapters does not always correlate with their uncertainty estimation capabilities, which are essential for a safe deployment in real-world scenarios. We also demonstrate that one of such adapters is obtained through MAP inference from a more general probabilistic framework. Based on this observation we introduce BayesAdapter, which leverages Bayesian inference to estimate a full probability distribution instead of a single point, better capturing the variability inherent in the parameter space. In a comprehensive empirical evaluation we show that our approach obtains high quality uncertainty estimates in the predictions, standing out in calibration and selective classification. Our code will be publicly available upon acceptance of the paper.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。