arXiv reaDer
ワッサーシュタイン対照表現蒸留
Wasserstein Contrastive Representation Distillation
知識蒸留(KD)の主な目標は、教師ネットワークから学習したモデルの情報を学生ネットワークにカプセル化することです。学生ネットワークは前者よりもコンパクトです。蒸留にKullback-Leibler分岐を使用するなどの既存の作業では、教師ネットワークで重要な構造知識を取得できない可能性があり、特に教師と生徒が異なる分類タスクに対処するように構築されている状況では、機能を一般化する機能が不足していることがよくあります。 KDのワッサースタイン距離のプライマリ形式とデュアル形式の両方を活用するワッサースタイン対照表現蒸留(WCoRD)を提案します。デュアルフォームはグローバルな知識の伝達に使用され、教師と生徒のネットワーク間の相互情報量の下限を最大化する対照的な学習目標を生み出します。プライマリフォームは、ミニバッチ内のローカルの対照的な知識の伝達に使用され、教師と生徒のネットワーク間の機能の分布を効果的に一致させます。実験は、提案されたWCoRDメソッドが、特権情報の蒸留、モデルの圧縮、およびクロスモーダル転送に関する最先端のアプローチよりも優れていることを示しています。
The primary goal of knowledge distillation (KD) is to encapsulate the information of a model learned from a teacher network into a student network, with the latter being more compact than the former. Existing work, e.g., using Kullback-Leibler divergence for distillation, may fail to capture important structural knowledge in the teacher network and often lacks the ability for feature generalization, particularly in situations when teacher and student are built to address different classification tasks. We propose Wasserstein Contrastive Representation Distillation (WCoRD), which leverages both primal and dual forms of Wasserstein distance for KD. The dual form is used for global knowledge transfer, yielding a contrastive learning objective that maximizes the lower bound of mutual information between the teacher and the student networks. The primal form is used for local contrastive knowledge transfer within a mini-batch, effectively matching the distributions of features between the teacher and the student networks. Experiments demonstrate that the proposed WCoRD method outperforms state-of-the-art approaches on privileged information distillation, model compression and cross-modal transfer.
updated: Tue Dec 15 2020 23:43:28 GMT+0000 (UTC)
published: Tue Dec 15 2020 23:43:28 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アソシエイト