arXiv:2511.00028cs.CVcs.AI2025-11

用互信息筛选数据,提升模型在真实场景下的泛化能力

Mutual Information guided Visual Contrastive Learning

  • 基于真实分布中的互信息选择正样本
  • 在多个基准上验证了方法有效性
  • 适合追求开放环境泛化的研究者

利用InfoNCE损失的表示学习方法已显著减少人工标注需求,通过训练不变特征提取器实现。尽管不同变体均遵循数据与特征间的信息最大化原则,但数据选择与增强仍依赖人工假设或工程设计,可能次优。例如,对比学习中的数据增强多聚焦于颜色抖动,旨在模拟真实光照变化。本文探索基于真实分布中计算的互信息来选择训练数据的潜力,理论上可使学习到的特征在开放环境中具备更好泛化性。具体而言,我们选取在自然扰动(如颜色变化、运动)下具有高互信息的场景块作为对比损失的正样本。我们在多个主流表示学习框架的多个基准上评估该互信息引导的数据增强方法,证明其有效性,并确立其为未来研究的有前景方向。

原文摘要 · Abstract (English)

Representation learning methods utilizing the InfoNCE loss have demonstrated considerable capacity in reducing human annotation effort by training invariant neural feature extractors. Although different variants of the training objective adhere to the information maximization principle between the data and learned features, data selection and augmentation still rely on human hypotheses or engineering, which may be suboptimal. For instance, data augmentation in contrastive learning primarily focuses on color jittering, aiming to emulate real-world illumination changes. In this work, we investigate the potential of selecting training data based on their mutual information computed from real-world distributions, which, in principle, should endow the learned features with better generalization when applied in open environments. Specifically, we consider patches attached to scenes that exhibit high mutual information under natural perturbations, such as color changes and motion, as positive samples for learning with contrastive loss. We evaluate the proposed mutual-information-informed data augmentation method on several benchmarks across multiple state-of-the-art representation learning frameworks, demonstrating its effectiveness and establishing it as a promising direction for future research.

对比学习互信息数据增强

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