ICCVAug, 2023

ImbSAM: 对于类不平衡识别中的锐度感知最小化的深入研究

TL;DR在真实世界的识别任务中,类别不平衡是一种常见的挑战,其中大多数类别样本很少,也被称为尾类。本文从泛化的角度解决这个挑战,并且实证地发现极具潜力的 “Sharpness-Aware Minimization (SAM)” 在类别不平衡的情况下无法解决泛化问题。通过对这一特定类型任务的调查,我们确认其泛化 bottleneck 主要出现在对训练数据有限的尾类的严重过拟合。为了克服这个 bottleneck,我们利用类别先验来限制类别无关的 SAM 的泛化范围,并提出了一种名为 “Imbalanced-SAM (ImbSAM)” 的类别感知平滑优化算法。在类别先验的指导下,我们的 ImbSAM 特别改善了对尾类的泛化。我们还通过两个典型的类别不平衡识别应用验证了 ImbSAM 的有效性:长尾分类和半监督异常检测,在这些应用中,我们的 ImbSAM 显著提高了对尾类和异常的性能。我们的代码实现可在此 https URL 找到。