arXiv reaDer
LGD:オブジェクト検出のためのラベルガイド付き自己蒸留
LGD: Label-guided Self-distillation for Object Detection
この論文では、LGD(ラベルガイド付き自己蒸留)と呼ばれる、一般的なオブジェクト検出のための最初の自己蒸留フレームワークを提案します。以前の研究は、実際のシナリオでは利用できない可能性のある有益な知識を提供するために、強力な事前訓練を受けた教師に依存しています。代わりに、学生の表現と通常のラベルのみに基づいて有益な知識を生成します。私たちのフレームワークには、スパースラベル外観エンコーダー、オブジェクト間関係アダプター、オブジェクト内知識マッパーが含まれています。これらは、トレーニングフェーズで暗黙の教師を共同で形成し、ラベルと進化する学生の表現に動的に依存します。それらは検出器でエンドツーエンドでトレーニングされ、推論で破棄されます。実験的に、LGDは、さまざまな検出器、データセット、およびインスタンスのセグメンテーションなどの広範なタスクで適切な結果を取得します。たとえば、MS-COCOデータセットでは、LGDはResNet-50を使用してRetinaNetを2倍のシングルスケールトレーニングで36.2%から39.0%mAP(+ 2.8%)に改善します。これは、2倍のマルチスケールトレーニングの下でResNeXt-101 DCN v2を使用したFCOSのようなはるかに強力な検出器を46.1%から47.9%(+ 1.8%)にブーストします。 LGDは、従来の教師ベースの方法FGFIと比較して、事前に訓練された教師を必要とせずにパフォーマンスが向上するだけでなく、本来の学生の学習を超えて51%の訓練コストを削減します。コードはhttps://github.com/megvii-research/LGDで入手できます。
In this paper, we propose the first self-distillation framework for general object detection, termed LGD (Label-Guided self-Distillation). Previous studies rely on a strong pretrained teacher to provide instructive knowledge that could be unavailable in real-world scenarios. Instead, we generate an instructive knowledge based only on student representations and regular labels. Our framework includes sparse label-appearance encoder, inter-object relation adapter and intra-object knowledge mapper that jointly form an implicit teacher at training phase, dynamically dependent on labels and evolving student representations. They are trained end-to-end with detector and discarded in inference. Experimentally, LGD obtains decent results on various detectors, datasets, and extensive tasks like instance segmentation. For example in MS-COCO dataset, LGD improves RetinaNet with ResNet-50 under 2x single-scale training from 36.2% to 39.0% mAP (+ 2.8%). It boosts much stronger detectors like FCOS with ResNeXt-101 DCN v2 under 2x multi-scale training from 46.1% to 47.9% (+ 1.8%). Compared with a classical teacher-based method FGFI, LGD not only performs better without requiring pretrained teacher but also reduces 51% training cost beyond inherent student learning. Codes are available at https://github.com/megvii-research/LGD.
updated: Sun Apr 24 2022 11:50:05 GMT+0000 (UTC)
published: Thu Sep 23 2021 16:55:01 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アソシエイト