1段階のオブジェクト検出器は、分類損失とローカリゼーション損失を同時に最適化することによって訓練されます。前者は、多数のアンカーによる極端な前景と背景のクラスの不均衡の問題に多く苦しんでいます。このホワイトペーパーでは、1段階検出器の分類タスクをランキングタスクに置き換える新しいフレームワークを提案し、ランキング問題に平均精度損失(APロス)を採用することにより、この問題を軽減します。非微分性と非凸性のため、AP損失は直接最適化できません。この目的のために、パーセプトロン学習でのエラー駆動型更新スキームとディープネットワークでの逆伝播アルゴリズムをシームレスに組み合わせた新しい最適化アルゴリズムを開発します。提案アルゴリズムの良好な収束特性を理論的および経験的に検証します。実験結果は、ネットワークアーキテクチャを変更することなく、さまざまなベンチマークでのさまざまな種類の分類損失に対するAP損失に基づく最先端の1ステージ検出器の顕著なパフォーマンスの改善を示しています。コードはhttps://github.com/cccorn/AP-lossで入手できます。
One-stage object detectors are trained by optimizing classification-loss and localization-loss simultaneously, with the former suffering much from extreme foreground-background class imbalance issue due to the large number of anchors. This paper alleviates this issue by proposing a novel framework to replace the classification task in one-stage detectors with a ranking task, and adopting the Average-Precision loss (AP-loss) for the ranking problem. Due to its non-differentiability and non-convexity, the AP-loss cannot be optimized directly. For this purpose, we develop a novel optimization algorithm, which seamlessly combines the error-driven update scheme in perceptron learning and backpropagation algorithm in deep networks. We verify good convergence property of the proposed algorithm theoretically and empirically. Experimental results demonstrate notable performance improvement in state-of-the-art one-stage detectors based on AP-loss over different kinds of classification-losses on various benchmarks, without changing the network architectures. Code is available at https://github.com/cccorn/AP-loss.