English

Identifying and Controlling Important Neurons in Neural Machine Translation

Computation and Language 2018-11-06 v1

Abstract

Neural machine translation (NMT) models learn representations containing substantial linguistic information. However, it is not clear if such information is fully distributed or if some of it can be attributed to individual neurons. We develop unsupervised methods for discovering important neurons in NMT models. Our methods rely on the intuition that different models learn similar properties, and do not require any costly external supervision. We show experimentally that translation quality depends on the discovered neurons, and find that many of them capture common linguistic phenomena. Finally, we show how to control NMT translations in predictable ways, by modifying activations of individual neurons.

Keywords

Cite

@article{arxiv.1811.01157,
  title  = {Identifying and Controlling Important Neurons in Neural Machine Translation},
  author = {Anthony Bau and Yonatan Belinkov and Hassan Sajjad and Nadir Durrani and Fahim Dalvi and James Glass},
  journal= {arXiv preprint arXiv:1811.01157},
  year   = {2018}
}
R2 v1 2026-06-23T05:02:54.796Z