
摘要
我们提出了一种利用动态评估来改进神经序列模型的方法。通过基于梯度下降的机制,模型能够适应最近的历史数据,从而为重复出现的序列模式分配更高的概率。在我们的对比实验中,动态评估的表现优于现有的适应方法。动态评估将Penn Treebank和WikiText-2数据集上的词级困惑度分别提升至51.1和44.3,并将text8和Hutter Prize数据集上的字符级交叉熵分别降低至1.19比特/字符和1.08比特/字符,达到了当前最佳水平。
代码仓库
benkrause/dynamic-evaluation
官方
pytorch
GitHub 中提及
sacmehta/PRU
pytorch
GitHub 中提及
benkrause/dynamiceval-transformer
tf
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| language-modelling-on-hutter-prize | mLSTM + dynamic eval | Bit per Character (BPC): 1.08 Number of params: 46M |
| language-modelling-on-penn-treebank-word | AWD-LSTM + dynamic eval | Params: 24M Test perplexity: 51.1 Validation perplexity: 51.6 |
| language-modelling-on-text8 | mLSTM + dynamic eval | Bit per Character (BPC): 1.19 Number of params: 45M |
| language-modelling-on-wikitext-2 | AWD-LSTM + dynamic eval | Number of params: 33M Test perplexity: 44.3 Validation perplexity: 46.4 |