文章地址:https://arxiv.org/pdf/2008.07294.pdf
1. Motivation
One-stage目标检测器的性能劣于two-stage的原因之一在于前背景目标的不平衡,有人提出balanced loss, focal loss,OHEM等来解决此问题,但它们没有考虑样本之间的分布或联系。本文提出将其建模为ranking task,把AP直接作为目标损失函数,显式地考虑了样本间联系,且对正负样本比例更不敏感。然而,由于AP的不可导和不可分解问题,标准的梯度下降不适用于AP loss。
2. Method
本文目标是将one-stage检测器(如RetinaNet和SSD)的分类loss替换为ranking based AP loss。