arXiv:2511.01694cs.LGcs.AI2025-11被引 1

用卡尔曼滤波实现贝叶斯自然梯度微调,提升CLIP在少样本下的泛化能力。

Bayesian Natural Gradient Fine-Tuning of CLIP Models via Kalman Filtering

  • 基于卡尔曼滤波近似自然梯度,融合二阶优化与贝叶斯推断。
  • 在多个数据集上实现比现有方法更优的分布内/外性能,尤其提升OOD鲁棒性。
  • 首次将卡尔曼滤波用于CLIP微调,兼顾效率与不确定性量化,适合小样本场景。

视觉-语言预训练模型如CLIP在多模态数据挖掘中已建立新基准。然而,在标签数据稀缺时,少样本微调仍面临挑战,尤其在分布外(OOD)数据上的泛化性能不佳。现有方法多依赖一阶梯度优化器,存在收敛慢、对学习率敏感、泛化能力差等问题。相比之下,二阶方法利用损失曲面的局部曲率信息调节更新步长,对含尖锐极值点的非凸损失函数尤为有效。自然梯度方向能提供更高效、更具影响力的迭代更新。自然梯度下降(NGD)通过逆费舍尔信息矩阵(FIM)预处理标准梯度获得,但对大模型计算成本高昂。为此,本文提出一种基于卡尔曼滤波的贝叶斯近似方法来实现CLIP模型的自然梯度微调。该方法结合二阶优化与贝叶斯推断优势,提升泛化能力并提供不确定性量化。在多个图像分类数据集上的实验表明,本方法在保持或超越先进基线的分布内(ID)性能的同时,显著提升分布外(OOD)鲁棒性。据我们所知,这是首次成功将卡尔曼滤波应用于基于CLIP的模型微调,实现了更稳健高效的视觉-语言学习。

原文摘要 · Abstract (English)

Vision-language pre-trained models, such as CLIP, have established new benchmarks in multimodal data mining. In such models, few-shot fine-tuning is a major challenge to achieve optimal performance on both in-distribution (ID) and out-of-distribution (OOD) datasets, especially when labeled data is scarce. Most existing fine-tuning approaches rely on first-order gradient-based optimizers, which typically suffer from slow convergence, sensitivity to step-size hyperparameters, and poor generalization in OOD settings. In contrast, second-order methods utilize local curvature information of the loss landscape to adjust the update step size. This is particularly beneficial for CLIP models, whose non-convex loss functions often contain sharp critical points. In such cases, natural gradient direction can offer more substantial and efficient per-iteration updates when fine-tuning with limited data. Natural Gradient Descent (NGD) is obtained by preconditioning the standard gradient with the inverse Fisher Information Matrix (FIM), which is computationally expensive for large models. To address this, we propose a Bayesian approximation of NGD using a Kalman filter for CLIP models. Our method combines the benefits of second-order optimization with Bayesian inference, which enhances generalization while providing uncertainty quantification. Extensive experiments conducted on diverse image classification datasets demonstrate that our algorithm consistently achieves superior--or comparable--ID performance and improved OOD robustness compared to state-of-the-art baselines. To the best of our knowledge, this work represents the first successful application of Kalman filtering to fine-tuning CLIP-based models, which enables more robust and efficient learning in vision-language tasks.

CLIP贝叶斯优化自然梯度卡尔曼滤波

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。