通过差异感知筛选样本,加速多模态对比学习训练
Differential-informed Sample Selection Accelerates Multimodal Contrastive Learning
- 利用当前与历史模型预测相关性的差异判断样本质量
- 在三个基准数据集上实现比现有方法更快的训练速度
- 适合需要高效训练多模态模型的研究者和工程师
基于对比学习的多模态模型的成功很大程度上依赖于大规模数据集和高昂的计算成本。样本选择作为一种替代性高效范式,有助于加速训练过程。然而,现有方法要么依赖预训练模型离线选取高质量子集,难以适应冷启动场景;要么基于实时模型预测进行在线选择,却未充分考虑噪声对应关系的影响。为此,本文提出一种新的差异感知样本选择方法(DISSect),能准确高效地识别训练中的噪声对应关系,从而加速训练。我们重新思考了噪声对应关系对对比学习的影响,提出当前模型与历史模型预测相关性的差异更能反映样本质量。基于此,构建了鲁棒的基于差异的样本选择机制,并分析其理论依据。在三个基准数据集和多种下游任务上的大量实验表明,DISSect在一致性和效率上均优于当前最先进的方法。代码已开源:https://github.com/MediaBrain-SJTU/DISSect。
原文摘要 · Abstract (English)
The remarkable success of contrastive-learning-based multimodal models has been greatly driven by training on ever-larger datasets with expensive compute consumption. Sample selection as an alternative efficient paradigm plays an important direction to accelerate the training process. However, recent advances on sample selection either mostly rely on an oracle model to offline select a high-quality coreset, which is limited in the cold-start scenarios, or focus on online selection based on real-time model predictions, which has not sufficiently or efficiently considered the noisy correspondence. To address this dilemma, we propose a novel Differential-Informed Sample Selection (DISSect) method, which accurately and efficiently discriminates the noisy correspondence for training acceleration. Specifically, we rethink the impact of noisy correspondence on contrastive learning and propose that the differential between the predicted correlation of the current model and that of a historical model is more informative to characterize sample quality. Based on this, we construct a robust differential-based sample selection and analyze its theoretical insights. Extensive experiments on three benchmark datasets and various downstream tasks demonstrate the consistent superiority of DISSect over current state-of-the-art methods. Source code is available at: https://github.com/MediaBrain-SJTU/DISSect.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。