arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.00581cs.CLcs.LG
收藏

多尺度Transformer语言模型

Multi-scale Transformer Language Models

Facebook AI Research(Facebook人工智能研究院);Mila, Université de Montréal(蒙特利尔大学Mila)

Sandeep Subramanian, Ronan Collobert, Marc'Aurelio Ranzato, Y-Lan Boureau

摘要:

我们研究多尺度Transformer语言模型,学习表示的文本在多个尺度,并提出了三种不同的架构,有一个归纳的偏见,以处理语言的层次性质。大规模语言建模基准的实验经验证明有利的可能性与内存占用权衡,例如,我们表明,它是可能的训练一个分层的变体与30层,具有23%的内存占用和更好的困惑,相比香草Transformer与不到一半的层数,在多伦多BookCorpus。我们分析了在多个尺度上学习表示在内存占用,计算时间和困惑方面的优势,这是特别有吸引力的Transformers的运行时间和内存使用相对于序列长度的二次缩放。

英文摘要:

We investigate multi-scale transformer language models that learn representations of text at multiple scales, and present three different architectures that have an inductive bias to handle the hierarchical nature of language. Experiments on large-scale language modeling benchmarks empirically demonstrate favorable likelihood vs memory footprint trade-offs, e.g. we show that it is possible to train a hierarchical variant with 30 layers that has 23% smaller memory footprint and better perplexity, compared to a vanilla transformer with less than half the number of layers, on the Toronto BookCorpus. We analyze the advantages of learned representations at multiple scales in terms of memory footprint, compute time, and perplexity, which are particularly appealing given the quadratic scaling of transformers' run time and memory usage with respect to sequence length.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑