动态调整多模态对比学习温度与边界,提升长尾数据下的性能
MM-TS: Multi-Modal Temperature and Margin Schedules for Contrastive Learning with Long-Tail Data
- 根据样本局部分布动态调节对比损失中的温度参数
- 在四个数据集上实现新最优结果,显著提升长尾类别表现
- 融合温度调度与最大间隔框架,统一主流多模态学习方法
对比学习已成为单模态和多模态框架中的基础方法,通过拉近正样本对、推开负样本对来学习表示。在单模态场景中,温度参数可调控这种吸引力与排斥力的强度。本文提出多模态温度与边界调度(MM-TS),将单模态温度调度扩展至多模态对比学习。该方法在训练过程中动态调整对比损失中的温度,调节多模态空间中的吸引与排斥力。考虑到标准多模态数据集常呈长尾分布,我们根据每个样本的局部分布特性调整温度:密集簇中的样本采用更高温度以更好保持其语义结构。此外,我们证明温度调度可有效整合进最大间隔框架,从而统一信息瓶颈(InfoNCE)损失与最大间隔目标这两种主流多模态对比学习范式。我们在Flickr30K、MSCOCO、EPIC-KITCHENS-100和YouCook2四个广泛使用的图像-视频-语言数据集上评估,结果表明动态温度与边界调度显著提升性能,并达到该领域的最新最佳水平。
原文摘要 · Abstract (English)
Contrastive learning has become a fundamental approach in both uni-modal and multi-modal frameworks. This learning paradigm pulls positive pairs of samples closer while pushing negatives apart. In the uni-modal setting (e.g., image-based learning), previous research has shown that the strength of these forces can be controlled through the temperature parameter. In this work, we propose Multi-Modal Temperature and Margin Schedules (MM-TS), extending the concept of uni-modal temperature scheduling to multi-modal contrastive learning. Our method dynamically adjusts the temperature in the contrastive loss during training, modulating the attraction and repulsion forces in the multi-modal setting. Additionally, recognizing that standard multi-modal datasets often follow imbalanced, long-tail distributions, we adapt the temperature based on the local distribution of each training sample. Specifically, samples from dense clusters are assigned a higher temperature to better preserve their semantic structure. Furthermore, we demonstrate that temperature scheduling can be effectively integrated within a max-margin framework, thereby unifying the two predominant approaches in multi-modal contrastive learning: InfoNCE loss and max-margin objective. We evaluate our approach on four widely used image- and video-language datasets, Flickr30K, MSCOCO, EPIC-KITCHENS-100, and YouCook2, and show that our dynamic temperature and margin schedules improve performance and lead to new state-of-the-art results in the field.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。