arXiv reaDer
AI-KD: 自己知識蒸留のための敵対的学習と暗黙の正則化
AI-KD: Adversarial learning and Implicit regularization for self-Knowledge Distillation
敵対的学習と暗黙的蒸留 (AI-KD) の敵対的学習および暗黙的正則化と名付けられた、敵対的学習と暗黙的蒸留によってトレーニング手順を正則化する、新しい敵対的ペナルティ付き自己知識蒸留法を提示します。私たちのモデルは、事前トレーニング済みおよび以前のエポック予測確率からの決定論的および進歩的な知識を抽出するだけでなく、敵対的学習を使用して決定論的予測分布の知識を転送します。動機は、自己知識蒸留法がソフトターゲットで予測確率を正則化することですが、正確な分布を予測するのは難しい場合があります。私たちの方法は、訓練された手順で弁別器をだますように学生モデルが訓練されている間、事前訓練されたモデルと学生モデルの間の分布を区別するために弁別器を展開します。したがって、学生モデルは、事前トレーニング済みモデルの予測確率を学習できるだけでなく、事前トレーニング済みモデルと学生モデルの間の分布を調整することもできます。複数のデータセットのネットワーク アーキテクチャを使用して提案された方法の有効性を実証し、提案された方法が最先端の方法よりも優れたパフォーマンスを達成することを示します。
We present a novel adversarial penalized self-knowledge distillation method, named adversarial learning and implicit regularization for self-knowledge distillation (AI-KD), which regularizes the training procedure by adversarial learning and implicit distillations. Our model not only distills the deterministic and progressive knowledge which are from the pre-trained and previous epoch predictive probabilities but also transfers the knowledge of the deterministic predictive distributions using adversarial learning. The motivation is that the self-knowledge distillation methods regularize the predictive probabilities with soft targets, but the exact distributions may be hard to predict. Our method deploys a discriminator to distinguish the distributions between the pre-trained and student models while the student model is trained to fool the discriminator in the trained procedure. Thus, the student model not only can learn the pre-trained model's predictive probabilities but also align the distributions between the pre-trained and student models. We demonstrate the effectiveness of the proposed method with network architectures on multiple datasets and show the proposed method achieves better performance than state-of-the-art methods.
updated: Sun Nov 20 2022 10:30:58 GMT+0000 (UTC)
published: Sun Nov 20 2022 10:30:58 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アソシエイト