We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CL

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computation and Language

Title: Weight Distillation: Transferring the Knowledge in Neural Network Parameters

Abstract: Knowledge distillation has been proven to be effective in model acceleration and compression. It allows a small network to learn to generalize in the same way as a large network. Recent successes in pre-training suggest the effectiveness of transferring model parameters. Inspired by this, we investigate methods of model acceleration and compression in another line of research. We propose Weight Distillation to transfer the knowledge in the large network parameters through a parameter generator. Our experiments on WMT16 En-Ro, NIST12 Zh-En, and WMT14 En-De machine translation tasks show that weight distillation can train a small network that is 1.88~2.94x faster than the large network but with competitive performance. With the same sized small network, weight distillation can outperform knowledge distillation by 0.51~1.82 BLEU points.
Comments: accepted by ACL2021
Subjects: Computation and Language (cs.CL)
Cite as: arXiv:2009.09152 [cs.CL]
  (or arXiv:2009.09152v3 [cs.CL] for this version)

Submission history

From: Ye Lin [view email]
[v1] Sat, 19 Sep 2020 03:23:26 GMT (49kb)
[v2] Wed, 26 May 2021 04:22:37 GMT (51kb)
[v3] Mon, 19 Jul 2021 04:37:21 GMT (52kb)

Link back to: arXiv, form interface, contact.