解决视觉语言模型少样本学习中的类别不平衡问题,提升性能。
PRiSM: Prototype Regularization for Few-Shot VLMs
- 提出基于原型正则化的插件式方法,增强分类边界
- 在类不平衡和多类别场景下,性能显著优于现有方法
- 适合处理真实世界中数据分布不均的少样本任务
无需训练的少样本适配方法在视觉语言模型(VLMs)中受到广泛关注。然而,现有基准依赖于适应数据统计的强假设,例如类别平衡。我们质疑这些简化假设,提出更贴近现实的基准,通过狄利克雷采样控制类别平衡度和有效类别数量。令人意外的是,在此设定下,先进方法性能大幅下降,且标签样本越多,降幅越明显。为此,我们提出PRiSM,一种可作为插件模块部署的类别原型正则化方法,显著提升性能。该方法优化包含多项目标的损失函数,包括最大化类间距离的正则项,以及促进支持特征对齐和保持基线原型保真度的项。此外,我们设计了一种高效计算的块极大极小优化器,通过格什戈林圆定理推导出有效的块局部利普希茨常数(即海森矩阵谱范数的上界),实现快速收敛。大量实验表明,PRiSM能显著改进多个无训练基线,在严重类别不平衡和高类别数场景下优势明显。
原文摘要 · Abstract (English)
Training-free few-shot adaptation methods have gained significant attention recently in the context of Vision-language Models (VLMs). Yet, current benchmarks rely on strong assumptions about the statistics of the adaptation data, e.g., class balance. We question these simplifying assumptions and introduce a more realistic benchmark that varies both the levels of class balance and the effective number of classes in few-shot tasks via Dirichlet sampling. Surprisingly, under our setting, we observe substantial drops in the performances of state-of-the-art methods, more so when the number of labeled samples increases. To mitigate this, we introduce PRiSM, a class-prototype regularization that can be deployed as a plug and play module on top of any existing baseline method, significantly improving performances. Our method optimizes a novel multi-term loss, which includes a regularizer maximizing inter-class pairwise distances, along with additional terms promoting support-feature alignment and fidelity to the baseline prototypes. Furthermore, we introduce an effective and computationally efficient block Majorize-Minimize optimizer for our objective. More specifically, we derive a valid blockwise Lipschitz constant (i.e., a bound on the Hessian's spectral norm), which can be computed efficiently via the Gershgorin circle theorem. Extensive experiments show that PRiSM improves several training-free baselines, with large gains when dealing with severe class imbalance and high numbers of classes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。