通过显式放大难负样本梯度,提升多模态嵌入性能
Improve Multi-Modal Embedding Learning via Explicit Hard Negative Gradient Amplifying
- 分析info-NCE损失梯度,定位难负样本贡献
- 显式增强难负样本梯度,使模型学习更区分的嵌入
- 在MMEB上超越现有方法,适合追求高精度检索的场景
近年来,多模态大语言模型(MLLMs)快速发展,基于对比学习的CLIP框架被成功扩展至MLLMs,为多种检索任务提供了强大的通用多模态嵌入。尽管如此,核心的对比学习范式仍基本沿用CLIP风格。在该框架下,难负样本的有效挖掘仍是提升性能的关键。已有工作提出离线与在线难负样本挖掘策略以提高对比学习效率。然而,各难负样本对学习过程的具体贡献尚未深入研究。本文详细分析了info-NCE损失相对于查询、正样本和负样本的梯度,揭示了难负样本在参数更新中的作用。基于此分析,我们提出显式放大难负样本相关梯度,促使模型学习更具区分性的嵌入。基于LLaVA-OneVision-7B架构训练的多模态嵌入模型,在MMEB基准上达到当前最佳性能;当集成自研的MLLM QQMM时,其在MMEB排行榜中位列第一。代码与模型已开源于https://github.com/QQ-MM/QQMM-embed。
原文摘要 · Abstract (English)
With the rapid advancement of multi-modal large language models (MLLMs) in recent years, the foundational Contrastive Language-Image Pretraining (CLIP) framework has been successfully extended to MLLMs, enabling more powerful and universal multi-modal embeddings for a wide range of retrieval tasks. Despite these developments, the core contrastive learning paradigm remains largely unchanged from CLIP-style models to MLLMs. Within this framework, the effective mining of hard negative samples continues to be a critical factor for enhancing performance. Prior works have introduced both offline and online strategies for hard negative mining to improve the efficiency of contrastive learning. While these approaches have led to improved multi-modal embeddings, the specific contribution of each hard negative sample to the learning process has not been thoroughly investigated. In this work, we conduct a detailed analysis of the gradients of the info-NCE loss with respect to the query, positive, and negative samples, elucidating the role of hard negatives in updating model parameters. Building upon this analysis, we propose to explicitly amplify the gradients associated with hard negative samples, thereby encouraging the model to learn more discriminative embeddings. Our multi-modal embedding model, trained with the proposed Explicit Gradient Amplifier and based on the LLaVA-OneVision-7B architecture, achieves state-of-the-art performance on the MMEB benchmark compared to previous methods utilizing the same MLLM backbone. Furthermore, when integrated with our self-developed MLLM, QQMM, our approach attains the top rank on the MMEB leaderboard. Code and models are released on https://github.com/QQ-MM/QQMM-embed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。