arXiv reaDer
対照的な表現蒸留
Contrastive Representation Distillation
 多くの場合、1つのニューラルネットワークから別のニューラルネットワークに表現知識を転送したいと考えています。例としては、大きなネットワークを小さなネットワークに蒸留したり、ある感覚モダリティから別の感覚モダリティに知識を転送したり、モデルのコレクションを単一の推定器にまとめたりすることが含まれます。これらの問題に対する標準的なアプローチである知識の蒸留は、教師と生徒のネットワークの確率論的出力間のKLの相違を最小限に抑えます。この目的は、教師ネットワークの重要な構造的知識を無視することを示しています。これは、教師がデータを表現する際により有意に多くの情報を取得するように生徒を訓練するという別の目的の動機となります。この目標を対照学習として定式化します。実験は、結果として得られる新しい目標が、単一モデル圧縮、アンサンブル蒸留、クロスモーダル転送などのさまざまな知識転送タスクで、知識蒸留およびその他の最先端の蒸留器よりも優れていることを示しています。私たちの方法は、多くのトランスファータスクに新しい最先端を設定し、知識の抽出と組み合わせると教師のネットワークよりも優れている場合があります。コード:http://github.com/HobbitLong/RepDistiller。
Often we wish to transfer representational knowledge from one neural network to another. Examples include distilling a large network into a smaller one, transferring knowledge from one sensory modality to a second, or ensembling a collection of models into a single estimator. Knowledge distillation, the standard approach to these problems, minimizes the KL divergence between the probabilistic outputs of a teacher and student network. We demonstrate that this objective ignores important structural knowledge of the teacher network. This motivates an alternative objective by which we train a student to capture significantly more information in the teacher's representation of the data. We formulate this objective as contrastive learning. Experiments demonstrate that our resulting new objective outperforms knowledge distillation and other cutting-edge distillers on a variety of knowledge transfer tasks, including single model compression, ensemble distillation, and cross-modal transfer. Our method sets a new state-of-the-art in many transfer tasks, and sometimes even outperforms the teacher network when combined with knowledge distillation. Code: http://github.com/HobbitLong/RepDistiller.
updated: Mon Jan 24 2022 19:12:34 GMT+0000 (UTC)
published: Wed Oct 23 2019 17:59:18 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アソシエイト