arXiv reaDer
データ効率の高い検出トランスに向けて
Towards Data-Efficient Detection Transformers
検出トランスフォーマーは、サンプルが豊富なCOCOデータセットで競争力のあるパフォーマンスを実現しました。ただし、それらのほとんどは、Cityscapesなどの小さなサイズのデータセットで大幅なパフォーマンスの低下に悩まされていることを示しています。言い換えれば、検出トランスフォーマーは一般的にデータを大量に消費します。この問題に取り組むために、データ効率の高いRCNNバリアントから代表的なDETRへの段階的な移行を通じて、データ効率に影響を与える要因を経験的に分析します。経験的結果は、ローカル画像領域からのスパース特徴サンプリングが鍵を握っていることを示唆しています。この観察に基づいて、元のモデルへの変更を最小限に抑えながら、クロスアテンションレイヤーでキーと値のシーケンスを構築する方法を変更するだけで、既存の検出トランスフォーマーのデータを大量に消費する問題を軽減します。さらに、より豊富な監視を提供し、データ効率を向上させるために、シンプルでありながら効果的なラベル拡張方法を紹介します。実験によると、私たちの方法はさまざまな検出トランスフォーマーに簡単に適用でき、サイズの小さいデータセットとサンプルが豊富なデータセットの両方でパフォーマンスが向上します。コードはhttps://github.com/encounter1997/DE-DETRsで公開されます。
Detection Transformers have achieved competitive performance on the sample-rich COCO dataset. However, we show most of them suffer from significant performance drops on small-size datasets, like Cityscapes. In other words, the detection transformers are generally data-hungry. To tackle this problem, we empirically analyze the factors that affect data efficiency, through a step-by-step transition from a data-efficient RCNN variant to the representative DETR. The empirical results suggest that sparse feature sampling from local image areas holds the key. Based on this observation, we alleviate the data-hungry issue of existing detection transformers by simply alternating how key and value sequences are constructed in the cross-attention layer, with minimum modifications to the original models. Besides, we introduce a simple yet effective label augmentation method to provide richer supervision and improve data efficiency. Experiments show that our method can be readily applied to different detection transformers and improve their performance on both small-size and sample-rich datasets. Code will be made publicly available at https://github.com/encounter1997/DE-DETRs.
updated: Thu Mar 17 2022 17:56:34 GMT+0000 (UTC)
published: Thu Mar 17 2022 17:56:34 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アソシエイト