改进少样本视觉语言模型的提示集成方法,提升分类准确性。
Cluster-Aware Prompt Ensemble Learning for Few-Shot Vision-Language Model Adaptation
- 在分类逻辑空间而非特征空间进行提示集成,更贴合视觉分布。
- 引入聚类保持正则项,防止提示向量坍缩为统一方向。
- 动态加权有缺陷提示,增强跨数据集鲁棒性,适合少样本场景。
视觉语言模型(如CLIP)通过大量图像-文本对预训练,实现零样本任务迁移。这类模型常使用上下文提示的集成来表示类别,但传统平均特征的方法会将类别中心移离真实分布,导致效果不佳。为此,本文提出簇感知提示集成学习(CAPEL)框架,保留上下文提示的簇结构。CAPEL将图像分类到多个类别簇中,每个簇由特定提示表示。不直接在特征空间集成提示,而是在分类对数空间进行集成,更契合视觉特征分布。为进一步优化提示微调并维持簇间区分能力,引入聚类保持正则项,确保提示保持差异化与特异性,避免坍缩为同一方向。此外,集成自适应提示加权技术,动态调整模糊或错误提示的注意力权重,提升在多样化数据集和任务上的鲁棒表现。
原文摘要 · Abstract (English)
Vision-language models (VLMs) such as CLIP achieve zero-shot transfer across various tasks by pre-training on numerous image-text pairs. These models often benefit from using an ensemble of context prompts to represent a class. Despite being effective, conventional prompt ensembling that averages textual features of context prompts often yields suboptimal results. This is because feature averaging shifts the class centroids away from the true class distribution. To address this issue, we propose the Cluster-Aware Prompt Ensemble Learning (CAPEL) framework, which preserves the cluster nature of context prompts. CAPEL classifies images into one of several class clusters, each represented by a distinct prompt. Instead of ensembling prompts in the feature space, we perform ensembling in the classification logits space, aligning better with the visual feature distribution. To further optimize prompt fine-tuning while maintaining cluster-specific discriminative power, we introduce a cluster-preserving regularization term. This ensures that prompts remain distinct and specialized for different clusters, preventing collapse into a uniform direction. Additionally, we integrate an adaptive prompt weighting technique to dynamically adjust the attention weights for flawed or ambiguous prompts, ensuring robust performance across diverse datasets and tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。