arXiv reaDer
オブジェクト検出のためのアンカー剪定
Anchor Pruning for Object Detection
この論文は、一段アンカーベースの検出器における物体検出のためのアンカー剪定を提案した。剪定手法は、畳み込みニューラルネットワークの計算コストを削減するために広く使用されていますが、ほとんどの計算が行われることが多いバックボーンネットワークの最適化に重点を置く傾向があります。この作業では、特にオブジェクト検出のための追加の剪定手法であるアンカー剪定を示します。より効率的なバックボーンネットワークと、非最大抑制などの後処理ステップがボトルネックになる可能性のある組み込みシステムにオブジェクト検出器を展開する傾向が高まるにつれ、検出ヘッドで使用されるアンカーの影響がますます重要になっています。この作業では、オブジェクト検出ヘッドの多くのアンカーを、精度を損なうことなく削除できることを示します。追加の再トレーニングにより、アンカーの剪定は精度の向上につながる可能性さえあります。 SSDとMSCOCOに関する広範な実験では、検出ヘッドの効率を最大44%向上させると同時に、精度を向上させることができることが示されています。 RetinaNetとPASCALVOCに関するさらなる実験は、私たちのアプローチの一般的な有効性を示しています。また、アンカーの初期形状に関連するハイパーパラメーターを排除するためにアンカー剪定と一緒に使用できる「オーバーアンカー」モデルを紹介します。コードとモデルはhttps://github.com/Mxbonn/anchor_pruningで入手できます。
This paper proposes anchor pruning for object detection in one-stage anchor-based detectors. While pruning techniques are widely used to reduce the computational cost of convolutional neural networks, they tend to focus on optimizing the backbone networks where often most computations are. In this work we demonstrate an additional pruning technique, specifically for object detection: anchor pruning. With more efficient backbone networks and a growing trend of deploying object detectors on embedded systems where post-processing steps such as non-maximum suppression can be a bottleneck, the impact of the anchors used in the detection head is becoming increasingly more important. In this work, we show that many anchors in the object detection head can be removed without any loss in accuracy. With additional retraining, anchor pruning can even lead to improved accuracy. Extensive experiments on SSD and MS COCO show that the detection head can be made up to 44% more efficient while simultaneously increasing accuracy. Further experiments on RetinaNet and PASCAL VOC show the general effectiveness of our approach. We also introduce `overanchorized' models that can be used together with anchor pruning to eliminate hyperparameters related to the initial shape of anchors. Code and models are available at https://github.com/Mxbonn/anchor_pruning.
updated: Wed Jun 01 2022 11:35:54 GMT+0000 (UTC)
published: Thu Apr 01 2021 12:33:16 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アソシエイト