arXiv reaDer
DETReg:オブジェクト検出のための領域事前確率による教師なし事前トレーニング
DETReg: Unsupervised Pretraining with Region Priors for Object Detection
オブジェクト検出のための最近の自己監視事前トレーニング方法は、検出アーキテクチャの重要な部分を無視して、オブジェクト検出器のバックボーンを事前トレーニングすることに主に焦点を合わせています。代わりに、オブジェクトのローカリゼーションと埋め込みコンポーネントを含む、オブジェクト検出ネットワーク全体を事前トレーニングする新しい自己監視方式であるDETRegを紹介します。事前トレーニング中に、DETRegは、教師なし領域提案ジェネレーターからのローカリゼーションと一致するようにオブジェクトのローカリゼーションを予測し、同時に、対応する機能の埋め込みを自己監視画像エンコーダーからの埋め込みと位置合わせします。 DETRファミリーの検出器を使用してDETRegを実装し、COCO、PASCAL VOC、およびAirbus Shipベンチマークで微調整すると、競合するベースラインよりも向上することを示しています。半教師ありおよび数ショットの学習設定を含む低データレジームでは、DETRegは多くの最先端の結果を確立します。たとえば、COCOでは、10ショットの検出で+6.0 APの改善、+ 3.5APの改善が見られます。ラベルの1%のみでトレーニングする場合。コードと事前トレーニング済みモデルについては、https://amirbar.net/detregのプロジェクトページにアクセスしてください。
Recent self-supervised pretraining methods for object detection largely focus on pretraining the backbone of the object detector, neglecting key parts of detection architecture. Instead, we introduce DETReg, a new self-supervised method that pretrains the entire object detection network, including the object localization and embedding components. During pretraining, DETReg predicts object localizations to match the localizations from an unsupervised region proposal generator and simultaneously aligns the corresponding feature embeddings with embeddings from a self-supervised image encoder. We implement DETReg using the DETR family of detectors and show that it improves over competitive baselines when finetuned on COCO, PASCAL VOC, and Airbus Ship benchmarks. In low-data regimes, including semi-supervised and few-shot learning settings, DETReg establishes many state-of-the-art results, e.g., on COCO we see a +6.0 AP improvement for 10-shot detection and +3.5 AP improvement when training with only 1% of the labels. For code and pretrained models, visit the project page at https://amirbar.net/detreg
updated: Wed Dec 01 2021 18:55:51 GMT+0000 (UTC)
published: Tue Jun 08 2021 17:39:14 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アソシエイト