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

基于预训练语言模型的虚假相关鲁棒性实证研究

An Empirical Study on Robustness to Spurious Correlations using Pre-trained Language Models

Toyota Technological Institute at Chicago(芝加哥丰田技术研究所);Amazon AI(亚马逊人工智能);New York University(纽约大学)

Lifu Tu, Garima Lalwani, Spandana Gella, He He

摘要:

最近的工作表明,预先训练的语言模型(如BERT)提高了对数据集中虚假相关性的鲁棒性。被这些结果所吸引,我们发现他们成功的关键是从少量的反例中推广,其中虚假的相关性不成立。当这样的少数例子很少时,预先训练的模型表现得和从头开始训练的模型一样差。在极端少数的情况下,我们建议使用多任务学习(MTL)来提高泛化能力。我们对自然语言推理和释义识别的实验表明,MTL与正确的辅助任务显着提高了具有挑战性的例子的性能,而不会损害分布性能。此外,我们表明,从MTL的增益主要来自于改进的推广少数的例子。我们的研究结果突出了数据多样性的重要性,克服虚假的相关性。

英文摘要:

Recent work has shown that pre-trained language models such as BERT improve robustness to spurious correlations in the dataset. Intrigued by these results, we find that the key to their success is generalization from a small amount of counterexamples where the spurious correlations do not hold. When such minority examples are scarce, pre-trained models perform as poorly as models trained from scratch. In the case of extreme minority, we propose to use multi-task learning (MTL) to improve generalization. Our experiments on natural language inference and paraphrase identification show that MTL with the right auxiliary tasks significantly improves performance on challenging examples without hurting the in-distribution performance. Further, we show that the gain from MTL mainly comes from improved generalization from the minority examples. Our results highlight the importance of data diversity for overcoming spurious correlations.

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

↑