Ben Krause, Emmanuel Kahembwe, Iain Murray, Steve Renals
This research note combines two methods that have recently improved the state of the art in language modeling: Transformers and dynamic evaluation. Transformers use stacked layers of self-attention that allow them to capture long range dependencies in sequential data. Dynamic evaluation fits models to the recent sequence history, allowing them to assign higher probabilities to re-occurring sequential patterns. By applying dynamic evaluation to Transformer-XL models, we improve the state of the art on enwik8 from 0.99 to 0.94 bits/char, text8 from 1.08 to 1.04 bits/char, and WikiText-103 from 18.3 to 16.4 perplexity points.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Language Modelling | WikiText-103 | Test perplexity | 16.4 | Transformer-XL (RMS dynamic eval) |
| Language Modelling | WikiText-103 | Validation perplexity | 15.8 | Transformer-XL (RMS dynamic eval) |
| Language Modelling | WikiText-103 | Test perplexity | 17 | Transformer-XL (SGD dynamic eval) |
| Language Modelling | WikiText-103 | Validation perplexity | 16.3 | Transformer-XL (SGD dynamic eval) |
| Language Modelling | Text8 | Bit per Character (BPC) | 1.038 | Transformer-XL + RMS dynamic eval + decay |
| Language Modelling | Hutter Prize | Bit per Character (BPC) | 0.94 | Transformer-XL + RMS dynamic eval |
| Language Modelling | enwik8 | Bit per Character (BPC) | 0.94 | Transformer-XL (24 layers, RMS dynamic eval, decay) |