arXiv reaDer
スパース MDOD: 二部マッチングを使用しないエンドツーエンドのマルチオブジェクト検出器のトレーニング
Sparse MDOD: Training End-to-End Multi-Object Detector without Bipartite Matching
最近のエンドツーエンドのマルチオブジェクト検出器は、非最大抑制 (NMS) を使用した重複境界ボックスの削除などの手作業のプロセスを削除することで、推論パイプラインを簡素化します。ただし、トレーニングでは、検出器の出力から損失を計算するために二部マッチングが必要です。エンド ツー エンド学習の中心にある指向性とは対照的に、二部マッチングは、エンド ツー エンド検出器のトレーニングを複雑でヒューリスティックで信頼性の高いものにします。この論文では、二部マッチングなしでエンドツーエンドのマルチオブジェクト検出器をトレーニングする方法を提案します。この目的のために、混合モデルを使用した密度推定問題として、エンドツーエンドのマルチオブジェクト検出に取り組みます。 Sparse Mixture Density Object Detector (Sparse MDOD) と呼ばれる提案された検出器は、混合モデルを使用して境界ボックスの分布を推定します。スパース MDOD は、負の対数尤度と、提案された正則化項、予測の重複を防ぐ最大成分最大化 (MCM) 損失を最小化することによってトレーニングされます。トレーニング中、二部マッチングなどの追加手順は必要なく、損失はネットワーク出力から直接計算されます。さらに、当社のスパース MDOD は、有名なマルチオブジェクト検出ベンチマークである MS-COCO の既存の検出器よりも優れています。
Recent end-to-end multi-object detectors simplify the inference pipeline by removing the hand-crafted process such as the duplicate bounding box removal using non-maximum suppression (NMS). However, in the training, they require bipartite matching to calculate the loss from the output of the detector. Contrary to the directivity, which is at the heart of end-to-end learning, the bipartite matching makes the training of the end-to-end detector complex, heuristic, and reliant. In this paper, we propose a method to train an end-to-end multi-object detector without bipartite matching. To this end, we approach end-to-end multi-object detection as a density estimation problem using a mixture model. Our proposed detector, called Sparse Mixture Density Object Detector (Sparse MDOD), estimates the distribution of bounding boxes using a mixture model. Sparse MDOD is trained by minimizing the negative log-likelihood and our proposed regularization term, maximum component maximization (MCM) loss that prevents duplicated predictions. During training, no additional procedure such as bipartite matching is needed, and the loss is directly computed from the network outputs. Moreover, our Sparse MDOD outperforms the existing detectors on MS-COCO, a renowned multi-object detection benchmark.
updated: Mon Sep 12 2022 01:10:50 GMT+0000 (UTC)
published: Wed May 18 2022 04:20:23 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アソシエイト