Improving Low Compute Language Modeling with In-Domain Embedding Initialisation

Charles Welch, Rada Mihalcea, Jonathan K. Kummerfeld


Abstract
Many NLP applications, such as biomedical data and technical support, have 10-100 million tokens of in-domain data and limited computational resources for learning from it. How should we train a language model in this scenario? Most language modeling research considers either a small dataset with a closed vocabulary (like the standard 1 million token Penn Treebank), or the whole web with byte-pair encoding. We show that for our target setting in English, initialising and freezing input embeddings using in-domain data can improve language model performance by providing a useful representation of rare words, and this pattern holds across several different domains. In the process, we show that the standard convention of tying input and output embeddings does not improve perplexity when initializing with embeddings trained on in-domain data.
Anthology ID:
2020.emnlp-main.696
Volume:
Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP)
Month:
November
Year:
2020
Address:
Online
Editors:
Bonnie Webber, Trevor Cohn, Yulan He, Yang Liu
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
8625–8634
Language:
URL:
https://aclanthology.org/2020.emnlp-main.696
DOI:
10.18653/v1/2020.emnlp-main.696
Bibkey:
Cite (ACL):
Charles Welch, Rada Mihalcea, and Jonathan K. Kummerfeld. 2020. Improving Low Compute Language Modeling with In-Domain Embedding Initialisation. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 8625–8634, Online. Association for Computational Linguistics.
Cite (Informal):
Improving Low Compute Language Modeling with In-Domain Embedding Initialisation (Welch et al., EMNLP 2020)
Copy Citation:
PDF:
https://aclanthology.org/2020.emnlp-main.696.pdf
Optional supplementary material:
 2020.emnlp-main.696.OptionalSupplementaryMaterial.zip
Video:
 https://slideslive.com/38938903
Code
 jkkummerfeld/emnlp20lm
Data
Penn Treebank