arXiv reaDer
データのない知識の蒸留を求める堅牢性と多様性
Robustness and Diversity Seeking Data-Free Knowledge Distillation
知識蒸留(KD)は、モデルの圧縮と知識の伝達において目覚ましい進歩をもたらしました。ただし、KDには、通常は実際には利用できない大量の元のデータまたはそれらの表現統計が必要です。この問題を解決するために、データフリーのKDが最近提案されました。この場合、教師と生徒のモデルは、教師からトレーニングされた合成サンプルジェネレータによって供給されます。それにもかかわらず、既存のデータフリーKD手法は、重みの微調整に依存して複数の損失のバランスを取り、生成されたサンプルの多様性を無視するため、精度と堅牢性が制限されます。この課題を克服するために、本稿では、データフリーKD(RDSKD)を求める堅牢性と多様性を提案します。ジェネレータ損失関数は、高い信頼性、クラスの多様性、およびサンプル間の多様性を備えたサンプルを生成するように作成されています。実際のデータがないと、高いサンプルの信頼性とクラスの多様性を求める目的が互いに矛盾することが多く、損失の変動が頻繁に発生します。損失の増分に指数関数的にペナルティを課すことにより、これを軽減します。 MNIST、CIFAR-10、およびSVHNデータセットを使用した実験では、DAFL、MSKD、ZSKD、DeepInversionなどの他のデータフリーKDメソッドと比較して、RDSKDがさまざまなハイパーパラメーター設定に対してより高い精度と堅牢性を実現することが示されています。
Knowledge distillation (KD) has enabled remarkable progress in model compression and knowledge transfer. However, KD requires a large volume of original data or their representation statistics that are not usually available in practice. Data-free KD has recently been proposed to resolve this problem, wherein teacher and student models are fed by a synthetic sample generator trained from the teacher. Nonetheless, existing data-free KD methods rely on fine-tuning of weights to balance multiple losses, and ignore the diversity of generated samples, resulting in limited accuracy and robustness. To overcome this challenge, we propose robustness and diversity seeking data-free KD (RDSKD) in this paper. The generator loss function is crafted to produce samples with high authenticity, class diversity, and inter-sample diversity. Without real data, the objectives of seeking high sample authenticity and class diversity often conflict with each other, causing frequent loss fluctuations. We mitigate this by exponentially penalizing loss increments. With MNIST, CIFAR-10, and SVHN datasets, our experiments show that RDSKD achieves higher accuracy with more robustness over different hyperparameter settings, compared to other data-free KD methods such as DAFL, MSKD, ZSKD, and DeepInversion.
updated: Sat Feb 06 2021 03:45:51 GMT+0000 (UTC)
published: Sat Nov 07 2020 10:57:53 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アソシエイト