用特征投影提升视觉语言模型分类性能
Feature Projection Learning for Better Vision-Language Reasoning
- 将分类转为特征投影,构建轻量级适配器
- 在多个数据集上超越现有方法,准确率显著提升
- 适合需要高效微调的视觉语言下游任务
视觉-语言预训练模型(如CLIP)通过对比学习可提取通用视觉特征。为将此类模型知识迁移到下游任务,现有方法常面临性能有限、参数过多或训练时间长的问题。本文提出一种简单高效的适配方法FPL(Feature Projection Learning),通过投影类别原型特征至查询图像特征空间并重建图像特征图,以负平均平方重建误差作为类别得分。该方法将分类问题转化为特征投影问题,最终输出融合投影模型与原始CLIP的预测结果。大量实验证明,FPL在多个下游任务中表现优异,显著超越当前最优方法。
原文摘要 · Abstract (English)
Vision-Language Pre-Trained models, notably CLIP, that utilize contrastive learning have proven highly adept at extracting generalizable visual features. To inherit the well-learned knowledge of VLP models for downstream tasks, several approaches aim to adapt them efficiently with limited supervision. However, these methods either suffer from limited performance, excessive learnable parameters, or extended training times, all of which hinder their effectiveness in adapting the CLIP model to downstream tasks. In this work, we propose a simple yet efficient and effective method called \textit{\textbf{F}eature \textbf{P}rojection \textbf{L}earning(FPL)} to address these problems. Specifically, we develop a projection model that projects class prototype features into the query image feature space and reconstructs the query image feature map. The negative average squared reconstruction error is used as the class score. In this way, we transform the classification problem into a feature projection problem. The final output of this method is a combination of the prediction from the projection model and the original pre-trained CLIP. Comprehensive empirical evaluations confirm that FPL delivers superior accuracy, surpassing the current state-of-the-art methods by a substantial margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。