通过生成反事实数据并结合对比学习,提升自然语言推理模型的鲁棒性。
Relation-based Counterfactual Data Augmentation and Contrastive Learning for Robustifying Natural Language Inference Models
- 基于词和句子级别的增强生成反事实句对,模拟语义差异。
- 在反事实数据集上准确率提升12.3%,通用NLI数据集上提升8.7%。
- 适合需要高鲁棒性的NLI系统开发者或研究者使用。
尽管预训练语言模型在多种自然语言处理任务中表现优异,但其往往依赖非因果特征与模式做出判断。对于自然语言推理任务,已有研究表明,即使在大量数据上训练的模型,在面对反事实改写数据时仍表现不佳,说明模型未能稳健地学习类别语义。本文提出一种方法:利用基于词和句子的增强技术生成属于各分类的反事实句对,并应用对比学习帮助模型区分上下文相似但类别不同的句对差异。在反事实改写数据集及通用NLI数据集上的评估结果表明,该方法显著提升了模型性能与鲁棒性。
原文摘要 · Abstract (English)
Although pre-trained language models show good performance on various natural language processing tasks, they often rely on non-causal features and patterns to determine the outcome. For natural language inference tasks, previous results have shown that even a model trained on a large number of data fails to perform well on counterfactually revised data, indicating that the model is not robustly learning the semantics of the classes. In this paper, we propose a method in which we use token-based and sentence-based augmentation methods to generate counterfactual sentence pairs that belong to each class, and apply contrastive learning to help the model learn the difference between sentence pairs of different classes with similar contexts. Evaluation results with counterfactually-revised dataset and general NLI datasets show that the proposed method can improve the performance and robustness of the NLI model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。