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

APOLLO:一种面向语言模型逻辑推理自适应预训练的简单方法

APOLLO: A Simple Approach for Adaptive Pretraining of Language Models for Logical Reasoning

University of Southern California(南加州大学);Microsoft Cognitive Service Research(微软认知服务研究);University of Notre Dame(圣母大学)

Soumya Sanyal, Yichong Xu, Shuohang Wang, Ziyi Yang, Reid Pryzant, Wenhao Yu, Chenguang Zhu, Xiang Ren

AI总结:

本文提出 APOLLO,利用逻辑推理关键词筛选 Wikipedia 子集,并结合词性选择性掩码与蕴含/矛盾句子分类损失进行自适应预训练,从而在不依赖任务格式的情况下提升语言模型的通用逻辑推理能力。

摘要:

文本的逻辑推理是一项重要能力,它要求理解文本中呈现的信息及其相互联系,然后通过这些信息进行推理以推断新的结论。以往提升语言模型逻辑推理能力的工作需要对训练数据进行复杂处理(例如,将符号知识与文本对齐),从而产生了针对特定任务的数据增强方案,限制了通用逻辑推理技能的学习。在本工作中,我们提出了 APOLLO,这是一种经过自适应预训练、具备更强逻辑推理能力的语言模型。我们基于一组逻辑推理关键词选取 Wikipedia 的一个子集,用于语言模型的持续预训练。我们使用两种自监督损失函数:一种是改进的掩码语言建模损失,其中仅掩码特定词性的词,这些词相比基础语言理解可能需要更多推理;另一种是句子级分类损失,用于教模型区分句子的蕴含与矛盾类型。所提出的训练范式既简单,又独立于任务格式。我们在两个逻辑推理数据集上将 APOLLO 与以往基线进行比较,证明了其有效性。APOLLO 在 ReClor 上表现相当,并在 LogiQA 上优于基线。代码库已公开发布。

英文摘要:

Logical reasoning of text is an important ability that requires understanding the information present in the text, their interconnections, and then reasoning through them to infer new conclusions. Prior works on improving the logical reasoning ability of language models require complex processing of training data (e.g., aligning symbolic knowledge to text), yielding task-specific data augmentation solutions that restrict the learning of general logical reasoning skills. In this work, we propose APOLLO, an adaptively pretrained language model that has improved logical reasoning abilities. We select a subset of Wikipedia, based on a set of logical inference keywords, for continued pretraining of a language model. We use two self-supervised loss functions: a modified masked language modeling loss where only specific parts-of-speech words, that would likely require more reasoning than basic language understanding, are masked, and a sentence-level classification loss that teaches the model to distinguish between entailment and contradiction types of sentences. The proposed training paradigm is both simple and independent of task formats. We demonstrate the effectiveness of APOLLO by comparing it with prior baselines on two logical reasoning datasets. APOLLO performs comparably on ReClor and outperforms baselines on LogiQA. The code base has been made publicly available.

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

↑