損失関数は、ディープネットワークベースのオブジェクト検出器のトレーニングで重要な役割を果たします。オブジェクト検出に最も広く使用されている評価指標は、ローカリゼーションと分類のサブタスクのパフォーマンスを同時にキャプチャする平均精度(AP)です。ただし、APメトリックの微分不可能な性質により、従来のオブジェクト検出器は2つのサブタスクに別々の微分可能な損失を採用します。このようなミスアライメントの問題は、パフォーマンスの低下につながる可能性があります。これに対処するために、既存の作業ではAPメトリックの代理損失を手動で設計しようとしていますが、これには専門知識が必要であり、依然として最適ではない可能性があります。この論文では、パラメータ化されたAP損失を提案します。ここでは、パラメータ化された関数を導入して、AP計算の微分不可能なコンポーネントを置き換えます。したがって、さまざまなAP近似は、統一された式のパラメーター化された関数のファミリーによって表されます。次に、自動パラメータ検索アルゴリズムを使用して、最適なパラメータを検索します。 3つの異なるオブジェクト検出器(つまり、RetinaNet、Faster R-CNN、およびDeformable DETR)を使用したCOCOベンチマークでの広範な実験は、提案されたパラメーター化AP損失が既存の手作り損失を一貫して上回っていることを示しています。コードはhttps://github.com/fundamentalvision/Parameterized-AP-Lossでリリースされています。
Loss functions play an important role in training deep-network-based object detectors. The most widely used evaluation metric for object detection is Average Precision (AP), which captures the performance of localization and classification sub-tasks simultaneously. However, due to the non-differentiable nature of the AP metric, traditional object detectors adopt separate differentiable losses for the two sub-tasks. Such a mis-alignment issue may well lead to performance degradation. To address this, existing works seek to design surrogate losses for the AP metric manually, which requires expertise and may still be sub-optimal. In this paper, we propose Parameterized AP Loss, where parameterized functions are introduced to substitute the non-differentiable components in the AP calculation. Different AP approximations are thus represented by a family of parameterized functions in a unified formula. Automatic parameter search algorithm is then employed to search for the optimal parameters. Extensive experiments on the COCO benchmark with three different object detectors (i.e., RetinaNet, Faster R-CNN, and Deformable DETR) demonstrate that the proposed Parameterized AP Loss consistently outperforms existing handcrafted losses. Code is released at https://github.com/fundamentalvision/Parameterized-AP-Loss.