arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2102.02557cs.CL
收藏

自适应半参数语言模型

Adaptive Semiparametric Language Models

DeepMind(深度思维)

Dani Yogatama, Cyprien de Masson d'Autume, Lingpeng Kong

摘要:

我们提出了一种语言模型,它结合了一个大的参数神经网络(即,Transformer)与集成架构中的非参数情景记忆组件。我们的模型通过缓存本地隐藏状态(类似于transformer-XL)来使用扩展的短期上下文,并通过在每个时间步检索一组最近邻居令牌来使用全局长期记忆。我们设计了一个门控函数来自适应地组合多个信息源来进行预测。这种机制允许模型根据上下文在特定的基础上使用本地上下文、短期记忆或长期记忆(或它们的任何组合)。基于单词和基于字符的语言建模数据集上的实验表明,与强基线相比,我们所提出的方法是有效的。

英文摘要:

We present a language model that combines a large parametric neural network (i.e., a transformer) with a non-parametric episodic memory component in an integrated architecture. Our model uses extended short-term context by caching local hidden states -- similar to transformer-XL -- and global long-term memory by retrieving a set of nearest neighbor tokens at each timestep. We design a gating function to adaptively combine multiple information sources to make a prediction. This mechanism allows the model to use either local context, short-term memory, or long-term memory (or any combination of them) on an ad hoc basis depending on the context. Experiments on word-based and character-based language modeling datasets demonstrate the efficacy of our proposed method compared to strong baselines.

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

↑