arXiv:2411.11667cs.LGcs.AI2024-11被引 6

提出新方法精准识别对比学习中的图文错配数据。

Dissecting Representation Misalignment in Contrastive Learning via Influence Function

  • 基于扩展影响函数,同时评估正负样本影响。
  • 无需重训练,实现对大规模模型的快速数据价值评估。
  • 可追踪错误预测源头,适合提升模型可靠性研究者使用。

对比学习广泛应用于大规模多模态模型,但常依赖来源多样且不可靠的数据,包含大量图文错配或标注错误的样本,导致模型鲁棒性下降与幻觉现象。数据估值是检测此类错配的有效手段,但现有方法在大规模模型上计算成本过高。传统影响函数虽高效,却因专为点式损失设计,不适用于对比学习。对比学习需同时最小化正样本间距离、最大化负样本间距离,因此需从双重角度评估样本影响。为此,本文提出针对对比损失的扩展影响函数(ECIF),可同时考虑正负样本,提供闭式近似解,避免重训练。基于ECIF,我们开发了数据评估、错配检测与误判溯源算法。实验表明,相较于传统基线方法,ECIF能更准确评估数据影响与模型对齐程度,显著提升CLIP类嵌入模型的可解释性与透明度。

原文摘要 · Abstract (English)

Contrastive learning, commonly applied in large-scale multimodal models, often relies on data from diverse and often unreliable sources, which can include misaligned or mislabeled text-image pairs. This frequently leads to robustness issues and hallucinations, ultimately causing performance degradation. Data valuation is an efficient way to detect and trace these misalignments. Nevertheless, existing methods are computationally expensive for large-scale models. Although computationally efficient, classical influence functions are inadequate for contrastive learning models, as they were initially designed for pointwise loss. Furthermore, contrastive learning involves minimizing the distance between positive sample modalities while maximizing the distance between negative sample modalities. This necessitates evaluating the influence of samples from both perspectives. To tackle these challenges, we introduce the Extended Influence Function for Contrastive Loss (ECIF), an influence function crafted for contrastive loss. ECIF considers both positive and negative samples and provides a closed-form approximation of contrastive learning models, eliminating the need for retraining. Building upon ECIF, we develop a series of algorithms for data evaluation, misalignment detection, and misprediction trace-back tasks. Experimental results demonstrate our ECIF advances the transparency and interpretability of CLIP-style embedding models by offering a more accurate assessment of data impact and model alignment compared to traditional baseline methods.

对比学习数据质量模型可解释性影响函数

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