arXiv reaDer
分離された機能による物体検出器の蒸留
Distilling Object Detectors via Decoupled Features
知識の蒸留は、複雑な教師ネットワークからコンパクトな学生ネットワークに情報を継承し、強力なパフォーマンスを維持するために広く使用されているパラダイムです。画像分類とは異なり、オブジェクト検出器は、セマンティック情報が依存する機能が絡み合っている複数の損失関数を備えた、はるかに洗練されています。この論文では、オブジェクトを除く領域から派生した特徴の情報も、既存のアプローチでは通常無視されている学生検出器を蒸留するために不可欠であることを指摘します。さらに、異なる地域からの特徴は、蒸留中に異なる重要性で割り当てられるべきであることを明らかにします。この目的のために、より優れた学生検出器を学習するための分離機能(DeFeat)を介した新しい蒸留アルゴリズムを紹介します。具体的には、2つのレベルの分離された特徴が処理され、有用な情報が学生に埋め込まれます。つまり、首から分離された特徴と分類ヘッドから分離された提案です。さまざまなバックボーンを持つさまざまな検出器での広範な実験により、提案されたDeFeatは、オブジェクト検出のための最先端の蒸留方法を超えることができることが示されています。たとえば、DeFeatは、COCOベンチマークでResNet50ベースのFaster R-CNNを37.4%から40.9%mAPに改善し、ResNet50ベースのRetinaNetを36.5%から39.7%mAPに改善します。私たちの実装はhttps://github.com/ggjy/DeFeat.pytorchで入手できます。
Knowledge distillation is a widely used paradigm for inheriting information from a complicated teacher network to a compact student network and maintaining the strong performance. Different from image classification, object detectors are much more sophisticated with multiple loss functions in which features that semantic information rely on are tangled. In this paper, we point out that the information of features derived from regions excluding objects are also essential for distilling the student detector, which is usually ignored in existing approaches. In addition, we elucidate that features from different regions should be assigned with different importance during distillation. To this end, we present a novel distillation algorithm via decoupled features (DeFeat) for learning a better student detector. Specifically, two levels of decoupled features will be processed for embedding useful information into the student, i.e., decoupled features from neck and decoupled proposals from classification head. Extensive experiments on various detectors with different backbones show that the proposed DeFeat is able to surpass the state-of-the-art distillation methods for object detection. For example, DeFeat improves ResNet50 based Faster R-CNN from 37.4% to 40.9% mAP, and improves ResNet50 based RetinaNet from 36.5% to 39.7% mAP on COCO benchmark. Our implementation is available at https://github.com/ggjy/DeFeat.pytorch.
updated: Fri Mar 26 2021 13:58:49 GMT+0000 (UTC)
published: Fri Mar 26 2021 13:58:49 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アソシエイト