arXiv reaDer
データフリーの高密度深層蒸留
Data-free Dense Depth Distillation
単眼深度推定 (MDE) のためのデータフリー知識蒸留 (KD) を研究します。これは、ターゲット ドメインにトレーニング データが不足している間に、トレーニング済みの教師モデルから圧縮することにより、実世界の深度知覚タスクの軽量モデルを学習します。画像分類と密回帰の本質的な違いにより、以前のデータフリー KD の方法は MDE には適用できません。実世界のタスクでの適用性を強化するために、この論文では、分布外のシミュレートされた画像を使用して KD を適用することを提案します。解決すべき主な課題は、i) 実世界のトレーニング データのオブジェクト分布に関する事前情報の欠如、および ii) シミュレートされた画像と実世界の画像の間のドメイン シフトです。これらの困難に対処するために、深層蒸留のための調整されたフレームワークを提案します。このフレームワークは、ターゲット ドメイン内のオブジェクトの分散パターンを最大限にカバーするための新しいトレーニング サンプルを生成し、変換ネットワークを利用して教師モデルに保存されている特徴統計に効率的に適応させます。さまざまな深度推定モデルと 2 つの異なるデータセットに関する広範な実験を通じて、私たちの方法がベースライン KD よりも優れた性能を発揮し、わずか 1/6 のトレーニング画像でわずかに優れたパフォーマンスを達成することさえあり、明確な優位性を示しています。
We study data-free knowledge distillation (KD) for monocular depth estimation (MDE), which learns a lightweight model for real-world depth perception tasks by compressing it from a trained teacher model while lacking training data in the target domain. Owing to the essential difference between image classification and dense regression, previous methods of data-free KD are not applicable to MDE. To strengthen its applicability in real-world tasks, in this paper, we propose to apply KD with out-of-distribution simulated images. The major challenges to be resolved are i) lacking prior information about object distribution of real-world training data, and ii) domain shift between simulated and real-world images. To cope with these difficulties, we propose a tailored framework for depth distillation. The framework generates new training samples for maximally covering distributed patterns of objects in the target domain and utilizes a transformation network to efficiently adapt them to the feature statistics preserved in the teacher model. Through extensive experiments on various depth estimation models and two different datasets, we show that our method outperforms the baseline KD by a good margin and even achieves slightly better performance with as few as 1/6 of training images, demonstrating a clear superiority.
updated: Mon Jan 09 2023 10:04:51 GMT+0000 (UTC)
published: Fri Aug 26 2022 07:10: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アソシエイト