摘要
ROC曲线下面积(AUC)是评估实例级长尾学习问题的常用指标。在过去的二十年中,许多AUC优化方法被提出,旨在提高长尾分布下模型的性能。本文探索了像素级长尾语义分割场景下的AUC优化方法,这是一个更为复杂的情况。该任务对AUC优化技术提出了两大挑战。一方面,像素级任务中的AUC优化涉及损失项之间的复杂耦合,存在结构化的图像内部和图像间成对依赖关系,使理论分析变得复杂。另一方面,我们发现这种情况下AUC损失的小批量估计需要更大的批量大小,导致空间复杂度过高。为了解决这些问题,我们开发了一个像素级AUC损失函数,并对算法的泛化能力进行了基于依赖图的理论分析。此外,我们设计了一个尾部类别记忆库(T-Memory Bank)来管理显著的内存需求。最后,跨多个基准的综合实验验证了我们提出的AUCSeg方法的有效性。代码可在https://github.com/boyuh/AUCSeg获取。