arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.04873cs.CL
收藏

CORE:一个基于检索-编辑的反事实数据生成框架

CORE: A Retrieve-then-Edit Framework for Counterfactual Data Generation

Indian Institute of Technology, Madras(印度理工学院马德拉斯分校);Paul G. Allen School of Computer Science & Engineering, University of Washington(华盛顿大学保罗·G·艾伦计算机科学与工程学院);Allen Institute of Artificial Intelligence(艾伦人工智能研究所);Meta AI

Tanay Dixit, Bhargavi Paranjape, Hannaneh Hajishirzi, Luke Zettlemoyer

摘要:

反事实数据增强(CDA)--即,在训练过程中添加最小扰动输入-有助于减少模型对虚假相关性的依赖,并提高对分布外(OOD)数据的泛化能力。先前关于生成反事实的工作只考虑了有限类的扰动,限制了它们的有效性。我们提出了通过检索和编辑(CORE),一个检索增强生成的框架,用于创建不同的反事实扰动CDA的反事实生成。对于每个训练样本,CORE首先使用学习的双编码器对任务相关的未标记文本语料库执行稠密检索,并提取相关的反事实摘录。CORE然后将这些内容合并到一个具有少样本学习能力的大语言模型的提示中,以进行反事实编辑。在自然发生的数据上调节语言模型编辑会导致不同的扰动。自然语言推理和情感分析基准测试的实验表明,CORE反事实比其他DA方法更有效地提高了对OOD数据的泛化能力。我们还表明,核心检索框架可以用来鼓励多样性,在手动创作的扰动

英文摘要:

Counterfactual data augmentation (CDA) -- i.e., adding minimally perturbed inputs during training -- helps reduce model reliance on spurious correlations and improves generalization to out-of-distribution (OOD) data. Prior work on generating counterfactuals only considered restricted classes of perturbations, limiting their effectiveness. We present COunterfactual Generation via Retrieval and Editing (CORE), a retrieval-augmented generation framework for creating diverse counterfactual perturbations for CDA. For each training example, CORE first performs a dense retrieval over a task-related unlabeled text corpus using a learned bi-encoder and extracts relevant counterfactual excerpts. CORE then incorporates these into prompts to a large language model with few-shot learning capabilities, for counterfactual editing. Conditioning language model edits on naturally occurring data results in diverse perturbations. Experiments on natural language inference and sentiment analysis benchmarks show that CORE counterfactuals are more effective at improving generalization to OOD data compared to other DA approaches. We also show that the CORE retrieval framework can be used to encourage diversity in manually authored perturbations

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

↑