arXiv reaDer
OTA:オブジェクト検出のための最適なトランスポート割り当て
OTA: Optimal Transport Assignment for Object Detection
オブジェクト検出におけるラベル割り当ての最近の進歩は、主に、各グラウンドトゥルース(gt)オブジェクトのポジティブ/ネガティブトレーニングサンプルを独立して定義しようとしています。この論文では、グローバルな観点からラベルの割り当てを革新的に再検討し、最適化理論でよく研究されているトピックである最適輸送(OT)問題として割り当て手順を定式化することを提案します。具体的には、各需要家(アンカー)と供給者(gt)のペア間の輸送単価を、分類と回帰損失の加重和として定義します。策定後、最適な割り当てソリューションを見つけることは、最小の輸送コストで最適な輸送計画を解決するために変換されます。これは、シンクホーンの定理を介して解決できます。 COCOでは、Optimal Transport Assignment(OTA)を備えた単一のFCOS-ResNet-50検出器は、1Xスケジューラーで40.7%mAPに達することができ、他のすべての既存の割り当て方法を上回ります。 COCOとCrowdHumanで実施された広範な実験により、提案されたOTAの有効性、特に群集シナリオでの優位性がさらに検証されます。コードはhttps://github.com/Megvii-BaseDetection/OTAで入手できます。
Recent advances in label assignment in object detection mainly seek to independently define positive/negative training samples for each ground-truth (gt) object. In this paper, we innovatively revisit the label assignment from a global perspective and propose to formulate the assigning procedure as an Optimal Transport (OT) problem -- a well-studied topic in Optimization Theory. Concretely, we define the unit transportation cost between each demander (anchor) and supplier (gt) pair as the weighted summation of their classification and regression losses. After formulation, finding the best assignment solution is converted to solve the optimal transport plan at minimal transportation costs, which can be solved via Sinkhorn-Knopp Iteration. On COCO, a single FCOS-ResNet-50 detector equipped with Optimal Transport Assignment (OTA) can reach 40.7% mAP under 1X scheduler, outperforming all other existing assigning methods. Extensive experiments conducted on COCO and CrowdHuman further validate the effectiveness of our proposed OTA, especially its superiority in crowd scenarios. The code is available at https://github.com/Megvii-BaseDetection/OTA.
updated: Fri Mar 26 2021 04:45:12 GMT+0000 (UTC)
published: Fri Mar 26 2021 04:45:12 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アソシエイト