谷歌提出可量化评估NLG模型性能的BLEURT指标

by

https://static.cnbetacdn.com/topics/681e56b636122b6.png

过去几年,自然语言生成(NLG)模型得到了巨大的发展,通读、总结文本、以及参与对话的能力都已大幅提升。为便于研究团队评估不同 NLG 模型的性能,搜索巨头谷歌(Google)特地提出了一个名叫 BLEURT 量化指标。通常情况下,我们可以通过人工和自动化流程来评估 NLG 模型,比如双语评估学习(BLEU)。前者的缺点是太费人力,后者的优势是具有更高的准确性。


https://static.cnbetacdn.com/article/2020/0527/d7ca851ac9b26d2.jpg

(来自:MIT Tech Review

谷歌研究人员称,BLEURT 是一种针对自然语言模型(NLG)的全新自动化评估指标,可为不同模型打出可靠的评分,结果接近、甚至超越了人类指标。

据悉,BLEURT 的核心为机器学习。对于任何 ML 模型,最重要的就是训练用的数据有多丰富。然而对于 NLG 模型来说,其训练数据是相当有限的。

https://static.cnbetacdn.com/article/2020/0527/6af3a17132e6bbc.jpg

实际上,在 WMT Metrics Task 数据集中(目前人类汇聚的最大集合),也仅收集了涵盖新闻领域的大约 26 万数据。

若将之用作唯一的训练数据集,那 WMT 度量任务数据集将失去训练模型的通用性和鲁棒性。为攻克这一问题,研究人员采取了转移学习的方法。

首先,研究团队使用了 BERT 的上下文词,且其已顺利聚合到 Yis 和 BERTscore 等 NLG 量化工具中。

接着,研究人员介绍了一种新颖的预训练方案,以提升 BLEURT 的鲁棒性和准确度,同时有助于应对模型的质量偏移。

https://static.cnbetacdn.com/article/2020/0527/827ee1f30ed5c3a.jpg

在微调人工量化标准前,BLEURT 借助了数以百万计的合成句子,对 NLG 模型展开了“预热”训练。其通过来自维基百科的句子、加上随机扰动来生成训练数据。

研究团队未手机人工评分,而是使用了相关文献(含 BLEU)中的指标与模型集合,能够以极低的代价来扩大训练示例的数量,然后对 BLEURT 进行了两次预训练。

其一阶段目标是语言建模,二阶段目标则是评估 NLG 模型,此后团队在 WMT 指标数据集上对模型进行了微调。一旦受过训练,BLEURT 就会试着与竞争方案对抗,以证明其由于当前的指标。

https://static.cnbetacdn.com/article/2020/0527/aaf46b0d7f80a58.jpg

据悉,BLUERT 在 Python 3 上运行,且依赖于 TensorFlow,详情可参阅 GitHub 项目介绍页(传送门)。有关这项研究的详情,可翻看 ArXiv 上的预印本。

最后,研究人员还总结了其它结果,比如 BLEURT 试图“捕获表面重叠以外的 NLG 质量”,该指标在两项学术基准评估中获得了 SOTA 的评价。