arXiv reaDer
Anchor Pruning for Object Detection
この論文は、一段アンカーベースの検出器における物体検出のためのアンカー剪定を提案する。剪定手法は、畳み込みニューラルネットワークの計算コストを削減するために広く使用されていますが、ほとんどの計算が行われることが多いバックボーンネットワークの最適化に重点を置く傾向があります。この作業では、特にオブジェクト検出のための追加のプルーニング手法であるアンカープルーニングについて説明します。より効率的なバックボーンネットワークと、非最大抑制などの後処理ステップがボトルネックになる可能性がある組み込みシステムにオブジェクト検出器を展開する傾向が高まるにつれ、検出ヘッドで使用されるアンカーの影響がますます重要になっています。この作業では、オブジェクト検出ヘッドの多くのアンカーが精度を損なうことなく削除できることを示します。追加の再トレーニングにより、アンカーの剪定は精度の向上につながる可能性もあります。 SSDとMSCOCOでの広範な実験により、検出ヘッドの効率を最大44%向上させると同時に、精度を向上させることができることが示されています。 RetinaNetとPASCALVOCに関するさらなる実験は、私たちのアプローチの一般的な有効性を示しています。また、アンカーの初期形状に関連するハイパーパラメータを排除するためにアンカー剪定と一緒に使用できる「オーバーアンカー」モデルを紹介します。
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.
updated: Tue Jul 20 2021 06:34:37 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)アソシエイト