arXiv reaDer
DN-DETR:クエリノイズ除去を導入してDETRトレーニングを加速
DN-DETR: Accelerate DETR Training by Introducing Query DeNoising
この論文では、DETR(DEtection TRansformer)トレーニングを高速化するための新しいノイズ除去トレーニング方法を紹介し、DETRのような方法の遅い収束の問題についての理解を深めます。収束が遅いのは、トレーニングの初期段階で一貫性のない最適化目標を引き起こす2部グラフマッチングの不安定性に起因することを示しています。この問題に対処するために、ハンガリーの損失を除いて、私たちの方法は、ノイズのあるグラウンドトゥルースバウンディングボックスをTransformerデコーダーに追加でフィードし、元のボックスを再構築するようにモデルをトレーニングします。これにより、2部グラフのマッチングの難しさが効果的に軽減され、収束が速くなります。私たちの方法は普遍的であり、数十行のコードを追加することでDETRのような方法に簡単にプラグインでき、目覚ましい改善を実現します。その結果、私たちのDN-DETRは、同じ設定で顕著な改善(+ 1.9AP)をもたらし、ResNet-50を使用したDETRのような方法の中で最高の結果(それぞれ12および50エポックのトレーニングでAP 43.4および48.6)を達成します。背骨。同じ設定でのベースラインと比較して、DN-DETRは50%のトレーニングエポックで同等のパフォーマンスを達成します。コードはhttps://github.com/FengLi-ust/DN-DETRで入手できます。
We present in this paper a novel denoising training method to speedup DETR (DEtection TRansformer) training and offer a deepened understanding of the slow convergence issue of DETR-like methods. We show that the slow convergence results from the instability of bipartite graph matching which causes inconsistent optimization goals in early training stages. To address this issue, except for the Hungarian loss, our method additionally feeds ground-truth bounding boxes with noises into Transformer decoder and trains the model to reconstruct the original boxes, which effectively reduces the bipartite graph matching difficulty and leads to a faster convergence. Our method is universal and can be easily plugged into any DETR-like methods by adding dozens of lines of code to achieve a remarkable improvement. As a result, our DN-DETR results in a remarkable improvement (+1.9AP) under the same setting and achieves the best result (AP 43.4 and 48.6 with 12 and 50 epochs of training respectively) among DETR-like methods with ResNet-50 backbone. Compared with the baseline under the same setting, DN-DETR achieves comparable performance with 50% training epochs. Code is available at https://github.com/FengLi-ust/DN-DETR.
updated: Thu Dec 08 2022 16:46:49 GMT+0000 (UTC)
published: Wed Mar 02 2022 18:50: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アソシエイト