arXiv reaDer
選択的クエリ再収集によるクエリベースのオブジェクト検出の強化されたトレーニング
Enhanced Training of Query-Based Object Detection via Selective Query Recollection
この論文では、クエリベースのオブジェクト検出器が、中間段階で正しく予測しながら、最後のデコード段階で誤って予測する現象を調査します。トレーニング プロセスを確認し、見落とされた現象が 2 つの制限にあると考えます: トレーニングの強調の欠如と、デコード シーケンスからのカスケード エラーです。クエリベースのオブジェクト検出器のシンプルで効果的なトレーニング戦略である選択的クエリ再収集 (SQR) を設計して提示します。デコード段階が深くなるにつれて中間クエリを累積的に収集し、シーケンシャル構造とは別にクエリを下流のステージに選択的に転送します。このように、SQR は後の段階でトレーニングを重視し、後の段階で前の段階からの中間クエリを直接処理できるようにします。 SQR は、さまざまなクエリベースのオブジェクト検出器に簡単にプラグインでき、推論パイプラインを変更せずにパフォーマンスを大幅に向上させることができます。その結果、さまざまな設定 (バックボーン、クエリ数、スケジュール) にわたって Adamixer、DAB-DETR、および Deformable-DETR に SQR を適用し、一貫して 1.4 ~ 2.8 AP の改善をもたらします。
This paper investigates a phenomenon where query-based object detectors mispredict at the last decoding stage while predicting correctly at an intermediate stage. We review the training process and attribute the overlooked phenomenon to two limitations: lack of training emphasis and cascading errors from decoding sequence. We design and present Selective Query Recollection (SQR), a simple and effective training strategy for query-based object detectors. It cumulatively collects intermediate queries as decoding stages go deeper and selectively forwards the queries to the downstream stages aside from the sequential structure. Such-wise, SQR places training emphasis on later stages and allows later stages to work with intermediate queries from earlier stages directly. SQR can be easily plugged into various query-based object detectors and significantly enhances their performance while leaving the inference pipeline unchanged. As a result, we apply SQR on Adamixer, DAB-DETR, and Deformable-DETR across various settings (backbone, number of queries, schedule) and consistently brings 1.4-2.8 AP improvement.
updated: Thu Mar 02 2023 18:54:31 GMT+0000 (UTC)
published: Thu Dec 15 2022 02:45:57 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アソシエイト