提升少样本视觉语言模型适配的准确性与可靠性。
Calibrated Cache Model for Few-Shot Vision-Language Model Adaptation
- 引入三重校准模块,优化图像间相似性、样本关系建模和置信度动态调整。
- 在11个少样本分类数据集上达到当前最优性能,显著降低高不确定性输出风险。
- 适用于需要高效少样本适配的视觉语言模型场景,尤其适合资源受限环境。
基于缓存的方法在适配视觉语言模型(VLMs)方面表现出高效性与有效性,但现有缓存模型忽略了三个关键问题:1)预训练的VLM主要优化图像-文本相似性,忽视了图像-图像相似性,导致预训练与适配阶段存在差距;2)当前缓存模型基于Nadaraya-Watson估计器,未充分建模训练样本间的复杂关系;3)在样本有限条件下,缓存模型生成的logits具有高不确定性,直接使用不考虑置信度可能带来风险。本文提出三种校准模块:相似性校准通过无标签图像优化图像-图像相似性,采用可学习投影层与残差连接,并通过自监督对比损失优化;权重校准引入精度矩阵到权重函数中,将缓存模型转化为高斯过程(GP)回归器,更准确地建模样本关系;置信度校准利用GP回归计算的预测方差,动态重缩放缓存模型的logits,实现按置信度自适应调整。此外,为降低GP的高复杂度,提出分组学习策略。设计了无需训练和需训练两种变体。在11个少样本分类数据集上的大量实验表明,所提方法达到当前最优性能。
原文摘要 · Abstract (English)
Cache-based approaches stand out as both effective and efficient for adapting vision-language models (VLMs). Nonetheless, the existing cache model overlooks three crucial aspects. 1) Pre-trained VLMs are mainly optimized for image-text similarity, neglecting the importance of image-image similarity, leading to a gap between pre-training and adaptation. 2) The current cache model is based on the Nadaraya-Watson (N-W) estimator, which disregards the intricate relationships among training samples while constructing weight function. 3) Under the condition of limited samples, the logits generated by cache model are of high uncertainty, directly using these logits without accounting for the confidence could be problematic. This work presents three calibration modules aimed at addressing the above challenges. Similarity Calibration refines the image-image similarity by using unlabeled images. We add a learnable projection layer with residual connection on top of the pre-trained image encoder of CLIP and optimize the parameters by minimizing self-supervised contrastive loss. Weight Calibration introduces a precision matrix into the weight function to adequately model the relation between training samples, transforming the existing cache model to a Gaussian Process (GP) regressor, which could be more accurate than N-W estimator. Confidence Calibration leverages the predictive variances computed by GP Regression to dynamically re-scale the logits of cache model, ensuring that the cache model's outputs are appropriately adjusted based on their confidence levels. Besides, to reduce the high complexity of GPs, we further propose a group-based learning strategy. Integrating the above designs, we propose both training-free and training-required variants. Extensive experiments on 11 few-shot classification datasets validate that the proposed methods can achieve state-of-the-art performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。