ICMLMay, 2023

探究多标签学习中宏平均 AUC 的泛化能力

TL;DR本文旨在理解和解决多标签学习中常用的宏平均(Macro-AUC)评价指标的理论问题。通过对 PASCAL VOC、Yahoo、和 MS COCO 数据集的实验结果和理论分析,发现数据集的类别不平衡是影响宏平均评价指标泛化范围的关键因素,并提出一种更通用的 McDiarmid 型浓度不等式。结果表明,相较于基于单变量损失函数的算法,基于成对损失函数和重新加权的损失函数更加稳健。