摘要
在对比学习中,通过不同增强方法生成的原始图像的两个视图被视为一个正样本对,并要求它们的相似性要高。同样,不同图像的两个视图形成一个负样本对,且鼓励它们的相似性要低。通常情况下,由单一投影头提供的单一相似性度量用来评估正负样本对的相似性。然而,由于多样化的增强策略和不同的样本内部相似性,同一图像的视图可能并不总是相似。此外,由于样本之间的相似性,不同图像的视图可能比同一图像的视图更相似。因此,强制正样本对具有高相似性和负样本对具有低相似性可能无法实现,在某些情况下,这种强制可能会对性能产生不利影响。为了解决这一挑战,我们提出使用多个投影头,每个投影头生成一组不同的特征。我们的预训练损失函数源于对给定观测样本的正样本头部后验分布的最大似然估计的解决方案。该损失结合了对正负样本对的相似性度量,每个度量由单独的自适应温度重新加权,以防止不良解决方案。我们的方法,自适应多头对比学习(AMCL),可以应用于并在实验中增强几种流行的对比学习方法,如SimCLR、MoCo和Barlow Twins。无论在各种骨干网络和线性探测时期,改进都保持一致,并且在使用多种增强方法时变得更加显著。