arXiv reaDer
インスタンス-オブジェクト検出のための条件付き知識抽出
Instance-Conditional Knowledge Distillation for Object Detection
知識の蒸留は分類で大きな成功を収めていますが、それでも検出は困難です。検出用の一般的な画像では、さまざまな場所からの表現が検出ターゲットにさまざまな影響を与える可能性があり、蒸留のバランスをとることが困難になります。この論文では、必要な知識、つまりすべてのインスタンスの分類とローカリゼーションの両方の観点から有益な知識を抽出するための条件付き蒸留フレームワークを提案します。フレームワークは、学習可能な条件付きデコードモジュールを導入します。このモジュールは、各ターゲットインスタンスをクエリとして指定して情報を取得します。具体的には、条件情報をクエリとしてエンコードし、教師の表現をキーとして使用します。クエリとキーの間の注意は、ローカリゼーション認識に敏感な補助タスクによって導かれる、さまざまな機能の寄与を測定するために使用されます。広範な実験により、私たちの方法の有効性が実証されています。さまざまな設定で印象的な改善が見られます。特に、ResNet-50バックボーンを使用したRetinaNetを1xスケジュールで37.4から40.7 mAP(+3.3)にブーストし、3xスケジュールでResNet-101バックボーンを使用した教師(40.4 mAP)を上回ります。コードはhttps://github.com/megvii-research/ICDでリリースされています。
Knowledge distillation has shown great success in classification, however, it is still challenging for detection. In a typical image for detection, representations from different locations may have different contributions to detection targets, making the distillation hard to balance. In this paper, we propose a conditional distillation framework to distill the desired knowledge, namely knowledge that is beneficial in terms of both classification and localization for every instance. The framework introduces a learnable conditional decoding module, which retrieves information given each target instance as query. Specifically, we encode the condition information as query and use the teacher's representations as key. The attention between query and key is used to measure the contribution of different features, guided by a localization-recognition-sensitive auxiliary task. Extensive experiments demonstrate the efficacy of our method: we observe impressive improvements under various settings. Notably, we boost RetinaNet with ResNet-50 backbone from 37.4 to 40.7 mAP (+3.3) under 1x schedule, that even surpasses the teacher (40.4 mAP) with ResNet-101 backbone under 3x schedule. Code has been released on https://github.com/megvii-research/ICD.
updated: Thu Jan 06 2022 13:21:29 GMT+0000 (UTC)
published: Mon Oct 25 2021 08:23:29 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アソシエイト