arXiv reaDer
エンドツーエンドの高密度検出のための 1 対少数のラベル割り当て
One-to-Few Label Assignment for End-to-End Dense Detection
1 対 1 (o2o) のラベル割り当ては、トランスフォーマー ベースのエンド ツー エンド検出で重要な役割を果たします。最近では、エンド ツー エンドの高密度検出用の完全な畳み込み検出器に導入されました。ただし、o2o はポジティブ サンプルの数が限られているため、特徴学習の効率が低下する可能性があります。最近の DETR では、この問題を軽減するために追加の肯定的なサンプルが導入されていますが、デコーダーでの自己注意と相互注意の計算により、その実用的なアプリケーションが高密度で完全な畳み込み検出器に制限されています。この作業では、エンドツーエンドの密な検出のためのシンプルで効果的な1対少数(o2f)のラベル割り当て戦略を提案します。オブジェクトごとに 1 つのポジティブ アンカーと多数のネガティブ アンカーを定義することとは別に、ポジティブ サンプルとネガティブ サンプルとして同時に機能するいくつかのソフト アンカーを定義します。これらのソフト アンカーの正と負の重みは、トレーニング中に動的に調整されるため、トレーニングの初期段階では「表現学習」により貢献し、後の段階では「重複した予測の除去」により貢献できます。この方法でトレーニングされた検出器は、強力な特徴表現を学習できるだけでなく、エンドツーエンドの密な検出も実行できます。 COCO および CrowdHuman データセットでの実験は、o2f スキームの有効性を示しています。コードは https://github.com/strongwolf/o2f で入手できます。
One-to-one (o2o) label assignment plays a key role for transformer based end-to-end detection, and it has been recently introduced in fully convolutional detectors for end-to-end dense detection. However, o2o can degrade the feature learning efficiency due to the limited number of positive samples. Though extra positive samples are introduced to mitigate this issue in recent DETRs, the computation of self- and cross- attentions in the decoder limits its practical application to dense and fully convolutional detectors. In this work, we propose a simple yet effective one-to-few (o2f) label assignment strategy for end-to-end dense detection. Apart from defining one positive and many negative anchors for each object, we define several soft anchors, which serve as positive and negative samples simultaneously. The positive and negative weights of these soft anchors are dynamically adjusted during training so that they can contribute more to ``representation learning'' in the early training stage, and contribute more to ``duplicated prediction removal'' in the later stage. The detector trained in this way can not only learn a strong feature representation but also perform end-to-end dense detection. Experiments on COCO and CrowdHuman datasets demonstrate the effectiveness of the o2f scheme. Code is available at https://github.com/strongwolf/o2f.
updated: Tue Mar 21 2023 03:24:47 GMT+0000 (UTC)
published: Tue Mar 21 2023 03:24:47 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アソシエイト