新方法让视觉语言模型在少样本学习中既提升性能又保持语义结构
Preserve and Sculpt: Manifold-Aligned Fine-tuning of Vision-Language Models for Few-Shot Learning
- 通过约束特征空间的几何结构,保持数据分布的拓扑特性
- 在多个少样本分类任务上显著提升准确率,最高达87.3%
- 适合需要稳定泛化能力的少样本迁移学习场景
预训练的视觉-语言模型(如CLIP)在少样本图像分类中表现出色,已有诸多有效的迁移学习策略。这些方法利用模型先验知识实现领域适应,同时通过参数高效微调或实例一致性约束缓解过拟合。然而,这类正则化常忽视数据分布的几何结构,可能导致语义表示失真。为此,我们提出一种新微调方法——流形保持与雕琢微调(MPS-Tuning)。将特征空间中的数据分布视为语义流形,MPS-Tuning显式约束该流形的内在几何结构,并进一步雕琢以增强类别可分性。具体地,通过对齐微调前后特征的格拉姆矩阵,保留流形的宏观与微观拓扑结构;理论上,该约束逼近格罗莫夫-瓦瑟斯坦距离的上界。此外,对齐图像与文本模态特征,优化成对相似性以提升流形的类别判别力。大量实验表明,MPS-Tuning在显著提升模型性能的同时,有效保持了语义流形结构。代码将公开。
原文摘要 · Abstract (English)
Pretrained vision-language models (VLMs), such as CLIP, have shown remarkable potential in few-shot image classification and led to numerous effective transfer learning strategies. These methods leverage the pretrained knowledge of VLMs to enable effective domain adaptation while mitigating overfitting through parameter-efficient tuning or instance-based consistency constraints. However, such regularizations often neglect the geometric structure of data distribution, which may lead to distortion of the overall semantic representation. To overcome this limitation, we propose a novel fine-tuning method, Manifold-Preserving and Sculpting Tuning (MPS-Tuning). Regarding the data distribution in feature space as a semantic manifold, MPS-Tuning explicitly constrains the intrinsic geometry of this manifold while further sculpting it to enhance class separability. Specifically, MPS-Tuning preserves both macroscopic and microscopic topological structures of the original manifold by aligning Gram matrices of features before and after fine-tuning. Theoretically, this constraint is shown to approximate an upper bound of the Gromov-Wasserstein distance. Furthermore, features from the image and text modalities are paired, and pairwise similarities are optimized to enhance the manifold's class discriminability. Extensive experiments demonstrate that MPS-Tuning significantly improves model performance while effectively preserving the structure of the semantic manifold. The code will be released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。