Apr, 2024

EMC$^2$: 高效的 MCMC 负采样对比学习与全局收敛性

TL;DR提出了一种名为 EMC² 的高效马尔可夫链蒙特卡罗负采样方法,可用于对比学习中的负样本生成,该方法利用自适应 Metropolis-Hastings 子程序在线生成适应样本,并证明了 EMC² 在 T 次迭代中找到全局对比损失的 O (1/√T)- 稳定点。与先前的方法相比,EMC² 是第一个无论批量大小如何都能实现全局收敛(到稳定状态)并具有低计算和内存成本的算法,数值实验验证了 EMC² 在小批量训练中的有效性,并获得了与基准算法相当或更好的性能。在 STL-10 和 Imagenet-100 上报告了图像编码器预训练的结果。