arXiv reaDer
機能が豊富なオブジェクト検出器の蒸留
Distilling Object Detectors with Feature Richness
近年、大規模なディープモデルは大きな成功を収めていますが、計算の複雑さとストレージ要件が非常に大きいため、リソースが限られたデバイスにモデルを展開することは大きな課題となっています。モデルの圧縮と加速の方法として、知識の蒸留は、教師の検出器から暗い知識を転送することにより、小さなモデルのパフォーマンスを効果的に向上させます。ただし、既存の蒸留ベースの検出方法のほとんどは、主にバウンディングボックスの近くの機能を模倣しており、2つの制限があります。まず、バウンディングボックスの外側にある有益な機能を無視します。第二に、これらの方法は、教師の検出器によって誤って背景と見なされるいくつかの機能を模倣します。上記の問題に対処するために、蒸留中の一般化された検出可能性を向上させる重要な機能を選択するための新しい機能豊富スコア(FRS)メソッドを提案します。提案された方法は、バウンディングボックスの外側の重要な特徴を効果的に取得し、バウンディングボックス内の有害な特徴を削除します。広範な実験は、私たちの方法がアンカーベースの検出器とアンカーフリーの検出器の両方で優れた性能を達成することを示しています。たとえば、ResNet-50を使用したRetinaNetは、COCO2017データセットでmAPで39.7%を達成し、ResNet-101ベースの教師検出器を0.8%上回っています。私たちの実装はhttps://github.com/duzhixing/FRSで入手できます。
In recent years, large-scale deep models have achieved great success, but the huge computational complexity and massive storage requirements make it a great challenge to deploy them in resource-limited devices. As a model compression and acceleration method, knowledge distillation effectively improves the performance of small models by transferring the dark knowledge from the teacher detector. However, most of the existing distillation-based detection methods mainly imitating features near bounding boxes, which suffer from two limitations. First, they ignore the beneficial features outside the bounding boxes. Second, these methods imitate some features which are mistakenly regarded as the background by the teacher detector. To address the above issues, we propose a novel Feature-Richness Score (FRS) method to choose important features that improve generalized detectability during distilling. The proposed method effectively retrieves the important features outside the bounding boxes and removes the detrimental features within the bounding boxes. Extensive experiments show that our methods achieve excellent performance on both anchor-based and anchor-free detectors. For example, RetinaNet with ResNet-50 achieves 39.7% in mAP on the COCO2017 dataset, which even surpasses the ResNet-101 based teacher detector 38.9% by 0.8%. Our implementation is available at https://github.com/duzhixing/FRS.
updated: Thu Dec 09 2021 03:44:23 GMT+0000 (UTC)
published: Mon Nov 01 2021 03:16:06 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アソシエイト