通过自采样与双权重机制,有效提升视觉语言预训练在噪声数据下的鲁棒性
Robust Noisy Correspondence Learning via Self-Drop and Dual-Weight
- 将数据细分为四类,区分干净显著、干净不显著、模糊和噪声样本
- 自采样剔除噪声,双权重策略聚焦关键样本,提升模型对噪声的容忍度
- 在高噪声比下仍保持稳定性能,适合互联网收集的带噪跨模态数据
许多研究者通过众包或网络爬取获取数据以缓解跨模态匹配的数据饥渴问题。尽管无需昂贵标注,但不可避免引入错误配对,导致噪声对应问题。现有方法利用深度神经网络的记忆效应区分噪声并进行重加权,但简单降低噪声样本权重无法彻底消除其负面影响。本文提出一种新颖的自采样与双权重方法,通过四类数据划分实现精细化数据处理:干净且显著、干净但不显著、模糊、噪声。分析表明,对于视觉-语言预训练模型,少量干净样本的价值高于大量噪声样本。基于此,我们采用自采样剔除噪声样本以减轻噪声影响,并引入双权重策略使模型更关注显著样本,同时合理利用模糊样本。相比先前方法,本方案更具鲁棒性,在高噪声比例下表现更稳定。在Flickr30K、MS-COCO和Conceptual Captions三个常用数据集上的大量实验验证了该方法的有效性。
原文摘要 · Abstract (English)
Many researchers collect data from the internet through crowd-sourcing or web crawling to alleviate the data-hungry challenge associated with cross-modal matching. Although such practice does not require expensive annotations, it inevitably introduces mismatched pairs and results in a noisy correspondence problem. Current approaches leverage the memorization effect of deep neural networks to distinguish noise and perform re-weighting. However, briefly lowering the weight of noisy pairs cannot eliminate the negative impact of noisy correspondence in the training process. In this paper, we propose a novel self-drop and dual-weight approach, which achieves elaborate data processing by qua-partitioning the data. Specifically, our approach partitions all data into four types: clean and significant, clean yet insignificant, vague, and noisy. We analyze the effect of noisy and clean data pairs and find that for vision-language pre-training models, a small number of clean samples is more valuable than a majority of noisy ones. Based on this observation, we employ self-drop to discard noisy samples to effectively mitigate the impact of noise. In addition, we adopt a dual-weight strategy to ensure that the model focuses more on significant samples while appropriately leveraging vague samples. Compared to the prior works, our approach is more robust and demonstrates relatively more stable performance on noisy datasets, especially under a high noise ratio. Extensive experiments on three widely used datasets, including Flickr30K, MS-COCO, and Conceptual Captions, validate the effectiveness of our approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。