用类别信息量重新定义长尾检测难题,提升小样本类别识别效果
Pursuing Better Decision Boundaries for Long-Tailed Object Detection via Category Information Amount
- 提出类别信息量概念,量化每类学习难度
- 在LVIS和COCO-LT上显著提升低频类别准确率
- 方法通用性强,对非长尾数据集也有效
在目标检测中,通常以实例数量判断数据集是否呈现长尾分布,隐含假设是样本少的类别模型表现差。然而,即使实例分布较均衡,模型仍存在类别偏差,说明实例数无法完全解释该现象。本文首次提出类别信息量的概念及其度量方式,发现其与分类准确率呈显著负相关,表明信息量更能反映学习难度。基于此,提出信息量引导的角度间隔损失(IGAM Loss),根据每类信息量动态调整决策边界,缓解长尾数据集中的类别偏差。IGAM在LVIS v1.0和COCO-LT等长尾基准数据集上表现优异,且在非长尾数据集Pascal VOC中对低频类别也有显著提升。实验验证了类别信息量作为分析工具的有效性及方法的通用性。
原文摘要 · Abstract (English)
In object detection, the instance count is typically used to define whether a dataset exhibits a long-tail distribution, implicitly assuming that models will underperform on categories with fewer instances. This assumption has led to extensive research on category bias in datasets with imbalanced instance counts. However, models still exhibit category bias even in datasets where instance counts are relatively balanced, clearly indicating that instance count alone cannot explain this phenomenon. In this work, we first introduce the concept and measurement of category information amount. We observe a significant negative correlation between category information amount and accuracy, suggesting that category information amount more accurately reflects the learning difficulty of a category. Based on this observation, we propose Information Amount-Guided Angular Margin (IGAM) Loss. The core idea of IGAM is to dynamically adjust the decision space of each category based on its information amount, thereby reducing category bias in long-tail datasets. IGAM Loss not only performs well on long-tailed benchmark datasets such as LVIS v1.0 and COCO-LT but also shows significant improvement for underrepresented categories in the non-long-tailed dataset Pascal VOC. Comprehensive experiments demonstrate the potential of category information amount as a tool and the generality of our proposed method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。