References & Citations
Computer Science > Computation and Language
Title: Byte Pair Encoding is Suboptimal for Language Model Pretraining
(Submitted on 7 Apr 2020 (v1), last revised 5 Oct 2020 (this version, v2))
Abstract: The success of pretrained transformer language models (LMs) in natural language processing has led to a wide range of pretraining setups. In particular, these models employ a variety of subword tokenization methods, most notably byte-pair encoding (BPE) (Sennrich et al., 2016; Gage, 1994), the WordPiece method (Schuster and Nakajima, 2012), and unigram language modeling (Kudo, 2018), to segment text. However, to the best of our knowledge, the literature does not contain a direct evaluation of the impact of tokenization on language model pretraining. We analyze differences between BPE and unigram LM tokenization, finding that the latter method recovers subword units that align more closely with morphology and avoids problems stemming from BPE's greedy construction procedure. We then compare the fine-tuned task performance of identical transformer masked language models pretrained with these tokenizations. Across downstream tasks and two languages (English and Japanese), we find that the unigram LM tokenization method matches or outperforms BPE. We hope that developers of future pretrained LMs will consider adopting the unigram LM method over the more prevalent BPE.
Submission history
From: Kaj Bostrom [view email][v1] Tue, 7 Apr 2020 21:21:06 GMT (61kb,D)
[v2] Mon, 5 Oct 2020 17:35:44 GMT (4618kb,D)
Link back to: arXiv, form interface, contact.