Neural Architecture Search(NAS)は、最適なアーキテクチャを自動的に検出することにより、ネットワーク設計の手作業を効果的に削減する大きな可能性を示しています。注目に値するのは、現在のところ、オブジェクト検出は、コンピュータービジョンで非常に重要であるにもかかわらず、NASアルゴリズムの影響を受けにくいことです。私たちの知る限り、オブジェクト検出タスクに関する最近のNAS研究のほとんどは、これらのアルゴリズムによる過剰な計算リソースのコストは言うまでもなく、結果のモデルのパフォーマンスと効率のバランスを十分にとることができません。ここでは、特徴ピラミッドネットワーク(FPN)と、調整された強化学習パラダイムを使用した単純なアンカーフリーオブジェクト検出器の予測ヘッドであるFCOS [36]を検索することにより、より優れたオブジェクト検出器を取得する効率的な方法を提案します。慎重に設計された検索スペース、検索アルゴリズム、およびネットワーク品質を評価するための戦略により、8つのV100GPUを使用して4日以内に最高の検出アーキテクチャを見つけることができます。発見されたアーキテクチャは、COCOデータセットのAPで最先端のオブジェクト検出モデル(Faster R-CNN、Retina-Net、FCOSなど)を1.0%から5.4%ポイント上回り、計算の複雑さとメモリフットプリントは同等です。 、オブジェクト検出のための提案されたNASメソッドの有効性を示しています。コードはhttps://github.com/Lausannen/NAS-FCOSで入手できます。
Neural Architecture Search (NAS) has shown great potential in effectively reducing manual effort in network design by automatically discovering optimal architectures. What is noteworthy is that as of now, object detection is less touched by NAS algorithms despite its significant importance in computer vision. To the best of our knowledge, most of the recent NAS studies on object detection tasks fail to satisfactorily strike a balance between performance and efficiency of the resulting models, let alone the excessive amount of computational resources cost by those algorithms. Here we propose an efficient method to obtain better object detectors by searching for the feature pyramid network (FPN) as well as the prediction head of a simple anchor-free object detector, namely, FCOS [36], using a tailored reinforcement learning paradigm. With carefully designed search space, search algorithms, and strategies for evaluating network quality, we are able to find top-performing detection architectures within 4 days using 8 V100 GPUs. The discovered architectures surpass state-of-the-art object detection models (such as Faster R-CNN, Retina-Net and, FCOS) by 1.0% to 5.4% points in AP on the COCO dataset, with comparable computation complexity and memory footprint, demonstrating the efficacy of the proposed NAS method for object detection. Code is available at https://github.com/Lausannen/NAS-FCOS.