arXiv:2504.09448cs.CVcs.LG2025-04AAAI被引 8

用贝叶斯方法优化图文对齐,提升少样本下分布外数据的泛化能力。

Bayesian Cross-Modal Alignment Learning for Few-Shot Out-of-Distribution Generalization

  • 仅微调文本特征,结合贝叶斯建模与正交梯度损失
  • 在二维分布偏移下实现最优的OoD泛化性能
  • 适合关注少样本与跨域泛化的研究者

近期大模型在少样本学习中表现优异,但其在二维分布外(OoD)数据上的泛化能力尚未充分研究,包括相关性偏移和多样性偏移。即使训练数据充足,现有方法在OoD泛化上仍难超越标准经验风险最小化(ERM)。本文提出一种新的贝叶斯跨模态图文对齐学习方法(Bayes-CAL),仅通过贝叶斯建模微调文本表示,并引入梯度正交化损失与不变风险最小化(IRM)损失,以避免对基类过拟合,提升对未见类别的泛化能力。该方法通过解耦图像特征中的因果与非因果成分,实现更优的图文对齐。实验表明,Bayes-CAL在两类二维分布偏移下均达到当前最优的OoD泛化性能,且相比CLIP类模型,在未见类别上表现更稳定。代码已开源:https://github.com/LinLLLL/BayesCAL。

原文摘要 · Abstract (English)

Recent advances in large pre-trained models showed promising results in few-shot learning. However, their generalization ability on two-dimensional Out-of-Distribution (OoD) data, i.e., correlation shift and diversity shift, has not been thoroughly investigated. Researches have shown that even with a significant amount of training data, few methods can achieve better performance than the standard empirical risk minimization method (ERM) in OoD generalization. This few-shot OoD generalization dilemma emerges as a challenging direction in deep neural network generalization research, where the performance suffers from overfitting on few-shot examples and OoD generalization errors. In this paper, leveraging a broader supervision source, we explore a novel Bayesian cross-modal image-text alignment learning method (Bayes-CAL) to address this issue. Specifically, the model is designed as only text representations are fine-tuned via a Bayesian modelling approach with gradient orthogonalization loss and invariant risk minimization (IRM) loss. The Bayesian approach is essentially introduced to avoid overfitting the base classes observed during training and improve generalization to broader unseen classes. The dedicated loss is introduced to achieve better image-text alignment by disentangling the causal and non-casual parts of image features. Numerical experiments demonstrate that Bayes-CAL achieved state-of-the-art OoD generalization performances on two-dimensional distribution shifts. Moreover, compared with CLIP-like models, Bayes-CAL yields more stable generalization performances on unseen classes. Our code is available at https://github.com/LinLLLL/BayesCAL.

少样本学习分布外泛化贝叶斯方法图文对齐

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