arXiv reaDer
マスクされた生成的蒸留
Masked Generative Distillation
知識の蒸留は、さまざまなタスクにうまく適用されています。現在の蒸留アルゴリズムは通常、教師の出力を模倣することによって生徒のパフォーマンスを向上させます。この論文は、教師が生徒の特徴の回復を導くことによって生徒の表現力を向上させることもできることを示しています。この観点から、単純なMasked Generative Distillation(MGD)を提案します。これは、生徒の特徴のランダムピクセルをマスクし、単純なブロックを介して教師の完全な特徴を生成するように強制します。 MGDは、真に一般的な機能ベースの蒸留方法であり、画像分類、オブジェクト検出、セマンティックセグメンテーション、インスタンスセグメンテーションなどのさまざまなタスクで利用できます。広範なデータセットを使用してさまざまなモデルを実験した結果、すべての生徒が優れた改善を達成していることがわかりました。特に、ResNet-18を69.90%から71.69%にImageNetトップ1の精度で、RetinaNetとResNet-50バックボーンを37.4から41.0バウンディングボックスmAPに、SOLOをResNet-50に基づいて33.1から36.2にマスクmAPに、DeepLabV3をResNetに基づいて- 1873.20から76.02mIoUまで。コードはhttps://github.com/yzd-v/MGDで入手できます。
Knowledge distillation has been applied to various tasks successfully. The current distillation algorithm usually improves students' performance by imitating the output of the teacher. This paper shows that teachers can also improve students' representation power by guiding students' feature recovery. From this point of view, we propose Masked Generative Distillation (MGD), which is simple: we mask random pixels of the student's feature and force it to generate the teacher's full feature through a simple block. MGD is a truly general feature-based distillation method, which can be utilized on various tasks, including image classification, object detection, semantic segmentation and instance segmentation. We experiment on different models with extensive datasets and the results show that all the students achieve excellent improvements. Notably, we boost ResNet-18 from 69.90% to 71.69% ImageNet top-1 accuracy, RetinaNet with ResNet-50 backbone from 37.4 to 41.0 Boundingbox mAP, SOLO based on ResNet-50 from 33.1 to 36.2 Mask mAP and DeepLabV3 based on ResNet-18 from 73.20 to 76.02 mIoU. Our codes are available at https://github.com/yzd-v/MGD.
updated: Tue Jul 05 2022 06:35:10 GMT+0000 (UTC)
published: Tue May 03 2022 14:30: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アソシエイト