Mar, 2024

概率对比学习用于长尾视觉识别

TL;DR在研究中,我们提出了一种新颖的概率对比学习算法(ProCo),它通过估计特征空间中每个类别的样本数据分布并相应地采样对比对,以克服由于数据不平衡而导致的标准有监督学习算法性能下降的难题。我们引入了一个合理简单的假设,即在对比学习中,归一化特征遵循单位空间上的一组 von Mises-Fisher(vMF)分布的混合分布,利用该分布参数的估计,我们可以采样无限数量的对比对,并得到了对比损失的闭合形式,从而实现高效优化。