LPT通过过滤细粒度信息和结构约束,提升视觉语言模型泛化能力。
LPT: Less-overfitting Prompt Tuning for Vision-Language Model
- 用CLIP过滤易导致过拟合的细节特征,引导提示学习基础视觉概念。
- 引入特征级结构保持约束,增强特征空间塑性,缓解过拟合。
- 适合需要强泛化能力的跨域迁移与新类别识别任务。
视觉语言模型(VLMs)在下游任务中展现出卓越的泛化能力。由于高效性,提示学习逐渐成为将VLMs迁移到下游任务的更优方法,超越传统微调。然而,在迁移过程中,这些模型容易出现严重过拟合,导致泛化能力显著下降。为此,我们提出LPT框架,专为视觉语言模型设计。具体而言,我们利用CLIP过滤可能引发过拟合的细粒度前景信息,引导提示学习基础视觉概念。此外,为进一步缓解过拟合,我们在特征层面提出结构保持(SP)约束,使模型整体特征空间结构与冻结的CLIP对齐,赋予特征空间整体可塑性,实现优化过程中的有效重塑。同时,在输出层采用分层逻辑(HL)约束,控制输出端的整体类别信息,与SP形成互补。在多个基准测试(从基础到新类别、跨数据集迁移、领域泛化)上的大量实验表明,该方法显著提升了泛化能力,并有效缓解了过拟合,优于当前最先进方法。
原文摘要 · Abstract (English)
Vision-language models (VLMs) have demonstrated exceptional generalization capabilities for downstream tasks. Due to its efficiency, prompt learning has gradually become a more effective and efficient method for transferring VLMs to downstream tasks, surpassing traditional finetuning methods. However, during the transfer process, these models are prone to severe overfitting, leading to a significant decline in generalization ability. To address this issue, we propose a framework named LPT, specifically designed for vision-language models. Specifically, we use CLIP to filter out fine-grained foreground information that may lead to overfitting, thereby guiding the prompts with basic visual concepts. Additionally, to further mitigate overfitting, we have developed a Structural Preservation (SP) constraint at the feature level, which aligns the model's overall feature space structure with the frozen CLIP, endowing the feature space with overall plasticity and enabling effective reshaping of the feature space during optimization. Moreover, we employ Hierarchical Logit (HL) constraint at the output layer to constrain the overall class information in the output, complementing the role of SP at the output end. Extensive experiments across various benchmarks (from base-to-novel, cross-dataset transfer, and domain generalization) demonstrate that our approach significantly improves generalization capability and effectively alleviates overfitting compared to state-of-the-art methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。