Sep, 2023

SCoRe:面向真实世界类别不平衡场景的子模块组合表示学习

TL;DR在真实世界中,类别不平衡的情况下的表示学习已经成为深度学习进化中的一项具有挑战性的任务。缺乏稀有类别的视觉和结构特征限制了现代神经网络学习具有区分性的特征聚类。本文引入了 SCoRe(子模组合表示学习)框架,提出了一系列子模组合损失函数,以克服对比学习中与类别不平衡相关的挑战。我们在两个图像分类基准数据集(CIFAR-10 和 MedMNIST 的子集)以及一个真实世界中的道路目标检测基准数据集(India Driving Dataset)上进行了实验,实验结果表明,像 Facility Location、Graph-Cut 和 Log Determinant 等新引入的目标相对于最先进的度量学习方法在类别不平衡的分类任务中的性能提高了最高达 7.6%,在目标检测任务中的性能提高了最高达 19.4%。