arXiv reaDer
オブジェクト検出における分類とローカリゼーションを統合するランキングベースのバランスの取れた損失関数
A Ranking-based, Balanced Loss Function Unifying Classification and Localisation in Object Detection
オブジェクト検出の分類タスクとローカリゼーションタスクの両方に対して、統一された、制限された、バランスの取れた、ランキングベースの損失関数である平均ローカリゼーション-再現率-精度(aLRP)を提案します。 aLRPは、平均適合率(AP)損失が分類のためのランキングベースの損失関数(Chen et al。、2020)に精度を拡張する方法に触発された、ローカリゼーション-再現率-精度(LRP)パフォーマンスメトリック(Oksuz et al。、2018)を拡張します。 aLRPには、次の明確な利点があります。(i)aLRPは、分類タスクとローカリゼーションタスクの両方で最初のランキングベースの損失関数です。 (ii)両方のタスクにランキングを使用することにより、aLRPは、高精度の分類のために高品質のローカリゼーションを自然に実施します。 (iii)aLRPは、ポジティブとネガティブの間の証明可能なバランスを提供します。 (iv)最先端の検出器の損失関数の平均約6個のハイパーパラメーターと比較すると、aLRP Lossには1つのハイパーパラメーターしかなく、実際には調整していません。 COCOデータセットでは、aLRP Lossは、ランキングベースの前身であるAP Lossを最大約5APポイント改善し、テスト時間の増加なしで48.9 APを達成し、すべての1ステージ検出器を上回ります。コードはhttps://github.com/kemaloksuz/aLRPLossで入手できます。
We propose average Localisation-Recall-Precision (aLRP), a unified, bounded, balanced and ranking-based loss function for both classification and localisation tasks in object detection. aLRP extends the Localisation-Recall-Precision (LRP) performance metric (Oksuz et al., 2018) inspired from how Average Precision (AP) Loss extends precision to a ranking-based loss function for classification (Chen et al., 2020). aLRP has the following distinct advantages: (i) aLRP is the first ranking-based loss function for both classification and localisation tasks. (ii) Thanks to using ranking for both tasks, aLRP naturally enforces high-quality localisation for high-precision classification. (iii) aLRP provides provable balance between positives and negatives. (iv) Compared to on average ∼6 hyperparameters in the loss functions of state-of-the-art detectors, aLRP Loss has only one hyperparameter, which we did not tune in practice. On the COCO dataset, aLRP Loss improves its ranking-based predecessor, AP Loss, up to around 5 AP points, achieves 48.9 AP without test time augmentation and outperforms all one-stage detectors. Code available at: https://github.com/kemaloksuz/aLRPLoss .
updated: Thu Jan 07 2021 08:25:43 GMT+0000 (UTC)
published: Mon Sep 28 2020 19:24:51 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト