arXiv reaDer
DetNAS:オブジェクト検出のためのバックボーン検索
DetNAS: Backbone Search for Object Detection
 オブジェクト検出器には通常、画像分類用に設計されたバックボーンネットワークが装備されています。画像分類タスクとオブジェクト検出タスクの間にギャップがあるため、次善の策である可能性があります。この作業では、オブジェクト検出のためのより良いバックボーンの設計にニューラルアーキテクチャ検索(NAS)を使用するDetNASを紹介します。検出トレーニングには通常ImageNetの事前トレーニングが必要であるのに対し、NASシステムは監視信号としてターゲット検出タスクの精度を必要とするため、これは重要です。検索空間内のすべての可能なネットワークを含むワンショットスーパーネットの手法に基づいて、オブジェクト検出に関するバックボーン検索のフレームワークを提案します。典型的な検出器のトレーニングスケジュール、ImageNetの事前トレーニングと検出の微調整の下でスーパーネットをトレーニングします。次に、検出タスクをガイダンスとして使用して、トレーニングされたスーパーネット上でアーキテクチャ検索が実行されます。このフレームワークにより、バックボーン上のNASは非常に効率的になります。実験では、1段式RetinaNetや2段式FPNなど、さまざまな検出器でのDetNASの有効性を示しています。経験的に、オブジェクト検出で検索されたネットワークは、ImageNet分類で検索されたネットワークと比較して一貫した優位性を示すことがわかりました。結果として得られるアーキテクチャは、COCO上の手作りのネットワークよりも優れたパフォーマンスを実現し、FLOPの複雑さを大幅に軽減します。
Object detectors are usually equipped with backbone networks designed for image classification. It might be sub-optimal because of the gap between the tasks of image classification and object detection. In this work, we present DetNAS to use Neural Architecture Search (NAS) for the design of better backbones for object detection. It is non-trivial because detection training typically needs ImageNet pre-training while NAS systems require accuracies on the target detection task as supervisory signals. Based on the technique of one-shot supernet, which contains all possible networks in the search space, we propose a framework for backbone search on object detection. We train the supernet under the typical detector training schedule: ImageNet pre-training and detection fine-tuning. Then, the architecture search is performed on the trained supernet, using the detection task as the guidance. This framework makes NAS on backbones very efficient. In experiments, we show the effectiveness of DetNAS on various detectors, for instance, one-stage RetinaNet and the two-stage FPN. We empirically find that networks searched on object detection shows consistent superiority compared to those searched on ImageNet classification. The resulting architecture achieves superior performance than hand-crafted networks on COCO with much less FLOPs complexity.
updated: Mon Dec 30 2019 14:04:15 GMT+0000 (UTC)
published: Tue Mar 26 2019 16:08:30 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アソシエイト