arXiv reaDer
オブジェクト検出のための一般的なインスタンスの蒸留
General Instance Distillation for Object Detection
近年、知識の蒸留がモデル圧縮の効果的なソリューションであることが証明されています。このアプローチにより、軽量の学生モデルは、面倒な教師モデルから抽出された知識を習得できます。ただし、以前の検出の蒸留方法は、さまざまな検出フレームワークの一般化が弱く、インスタンス間の貴重な関係情報を無視して、グラウンドトゥルース(GT)に大きく依存しています。そこで、GTで区別される正または負を考慮せずに、識別インスタンスに基づく検出タスクのための新しい蒸留方法を提案します。これは、一般インスタンス蒸留(GID)と呼ばれます。私たちのアプローチには、蒸留のための特徴ベース、関係ベース、および応答ベースの知識を最大限に活用するための一般的なインスタンス選択モジュール(GISM)が含まれています。広範な結果は、学生モデルが大幅なAPの改善を達成し、さまざまな検出フレームワークで教師よりも優れていることを示しています。具体的には、ResNet-50を使用したRetinaNetは、COCOデータセットのGIDを使用したmAPで39.1%を達成します。これは、ベースラインの36.2%を2.9%超えており、APが38.1%のResNet-101ベースの教師モデルよりも優れています。
In recent years, knowledge distillation has been proved to be an effective solution for model compression. This approach can make lightweight student models acquire the knowledge extracted from cumbersome teacher models. However, previous distillation methods of detection have weak generalization for different detection frameworks and rely heavily on ground truth (GT), ignoring the valuable relation information between instances. Thus, we propose a novel distillation method for detection tasks based on discriminative instances without considering the positive or negative distinguished by GT, which is called general instance distillation (GID). Our approach contains a general instance selection module (GISM) to make full use of feature-based, relation-based and response-based knowledge for distillation. Extensive results demonstrate that the student model achieves significant AP improvement and even outperforms the teacher in various detection frameworks. Specifically, RetinaNet with ResNet-50 achieves 39.1% in mAP with GID on COCO dataset, which surpasses the baseline 36.2% by 2.9%, and even better than the ResNet-101 based teacher model with 38.1% AP.
updated: Wed Mar 03 2021 11:41:26 GMT+0000 (UTC)
published: Wed Mar 03 2021 11:41:26 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アソシエイト