跳转至

UMOP一阶段检测器模块,对金字塔层级目标分布不平衡的探索

参考:极市平台

作者单位: 百度

文章链接:https://arxiv.org/pdf/2109.07217.pdf

代码链接:https://github.com/zimoqingfeng/UMOP

Abstract

在目标检测领域,多级预测(比如FPN、YOLO)与重采样技术(如Focal Loss、ATSS)极大的提升了一阶段检测器性能。然而,如何通过逐级优化特征金字塔提升性能尚未进行探索。我们发现:在训练过程中,不同金字塔层级的正负样例比例是不同的,即存在Level Imbalance,而这尚未被一阶段检测器解决。

本文主要由以下三点贡献:

  • 本文首次证实:FPN性能在某种程度上会受到Level Imbalanced影响
  • 集成与所提UMOP,现有一阶段检测器可以用更少的训练迭代次数取得~1.5AP指标提升,且不会导致额外的计算负担;
  • 在COCO test-dev数据集上,最佳模型取得了截至目前最高的55.1AP指标(一阶段检测器)。

Details

带固定超参的FocalLoss不能同时对所有金字塔层级做好最好均衡。

Level-wise Resampling Paradigm(LRP)

基于独立重采样考量,对每个金字塔层级设置不同的分类损失

LRP是一种解决跨不同金字塔层级采样不平衡问题的多级优化框架,在训练过程中,对每个金字塔层级独立的计算分类损失

Progressive Focal Loss(PFL)

它基于每个金字塔层级的正样例预测渐进的调整难例挖掘度。

从分而治之的角度出发,提出了一种新的动态超参调整策略,它可以按照每个金字塔层级的收敛位置动态调整超参。对于不同层级的不平衡性,真正需要的超参数是完全不同的。