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

语言建模的语境温度

Contextual Temperature for Language Modeling

National Tsing Hua University(国立清华大学);Google(谷歌公司)

Pei-Hsin Wang, Sheng-Iou Hsieh, Shih-Chieh Chang, Yu-Ting Chen, Jia-Yu Pan, Wei Wei, Da-Chang Juan

摘要:

温度定标作为一种控制分布光滑性的有效方法,已被广泛应用,这有助于模型在各种任务中的性能。应用温度缩放的当前实践假设固定的或手动制作的动态变化的时间表。然而,我们的研究表明,每个类的个人最佳轨迹可以随着上下文而变化。为此,我们提出了上下文温度,一种广义的方法,学习每个词汇的最佳温度轨迹的上下文。实验结果表明,该方法显着提高了国家的最先进的语言模型,实现了55.31和62.89的测试集上的Penn Treebank和WikiText-2,分别困惑。深入的分析表明,学习的温度时间表的行为变化显着的词汇,最佳的时间表有助于控制的不确定性。这些证据进一步证明了所提出的方法的必要性及其相对于固定温度计划的优势。

英文摘要:

Temperature scaling has been widely used as an effective approach to control the smoothness of a distribution, which helps the model performance in various tasks. Current practices to apply temperature scaling assume either a fixed, or a manually-crafted dynamically changing schedule. However, our studies indicate that the individual optimal trajectory for each class can change with the context. To this end, we propose contextual temperature, a generalized approach that learns an optimal temperature trajectory for each vocabulary over the context. Experimental results confirm that the proposed method significantly improves state-of-the-art language models, achieving a perplexity of 55.31 and 62.89 on the test set of Penn Treebank and WikiText-2, respectively. In-depth analyses show that the behaviour of the learned temperature schedules varies dramatically by vocabulary, and that the optimal schedules help in controlling the uncertainties. These evidences further justify the need for the proposed method and its advantages over fixed temperature schedules.

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

↑