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

预训练Transformers的校准

Calibration of Pre-trained Transformers

The University of Texas at Austin(德克萨斯大学奥斯汀分校)

Shrey Desai, Greg Durrett

摘要:

预先训练的Transformers现在在自然语言处理中无处不在,但是尽管它们具有高端的任务性能,但很少有人知道它们是否经过校准。具体来说,这些模型的后验概率是否提供了一个准确的经验测量,来衡量模型在给定示例中正确的可能性?我们专注于BERT和RoBERTa在这项工作中,并分析他们的校准跨越三个任务:自然语言推理,释义检测,和常识推理。对于每一项任务,我们都考虑了域内和具有挑战性的域外设置,在这些设置中,模型面临更多的不确定示例。我们证明:(1)当开箱即用时,预训练模型在域内进行校准,与基线相比,其域外校准误差可以低3.5倍;(2)温度缩放在进一步减少域内校准误差方面是有效的,并且使用标签平滑故意增加经验不确定性有助于校准域外后验。

英文摘要:

Pre-trained Transformers are now ubiquitous in natural language processing, but despite their high end-task performance, little is known empirically about whether they are calibrated. Specifically, do these models' posterior probabilities provide an accurate empirical measure of how likely the model is to be correct on a given example? We focus on BERT and RoBERTa in this work, and analyze their calibration across three tasks: natural language inference, paraphrase detection, and commonsense reasoning. For each task, we consider in-domain as well as challenging out-of-domain settings, where models face more examples they should be uncertain about. We show that: (1) when used out-of-the-box, pre-trained models are calibrated in-domain, and compared to baselines, their calibration error out-of-domain can be as much as 3.5x lower; (2) temperature scaling is effective at further reducing calibration error in-domain, and using label smoothing to deliberately increase empirical uncertainty helps calibrate posteriors out-of-domain.

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

↑