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

面向标注不完整训练数据的关系抽取类自适应自训练方法

Class-Adaptive Self-Training for Relation Extraction with Incompletely Annotated Training Data

DAMO Academy, Alibaba Group(阿里巴巴集团达摩院);National University of Singapore(新加坡国立大学);Singapore University of Technology and Design(新加坡科技设计大学)

Qingyu Tan, Lu Xu, Lidong Bing, Hwee Tou Ng

AI总结:

针对关系抽取中训练数据标注不完整导致的假阴性问题,提出类自适应重采样自训练框架,通过按精确率和召回率重采样各类伪标签提升性能,在相关数据集上优于现有方法。

摘要:

关系抽取(RE)旨在从句子和文档中抽取关系。现有关系抽取模型通常依赖监督机器学习。但近期研究表明,许多RE数据集存在标注不完整的问题,即假阴性问题:有效关系被错误标注为“no_relation(无关系)”。用此类数据训练的模型在推理阶段不可避免会犯类似错误。自训练已被证明能有效缓解假阴性问题,但传统自训练易受确认偏差影响,且在少数类上表现不佳。为克服这一局限,我们提出了一种新型类自适应重采样自训练框架。具体而言,我们根据精确率和召回率分数对每个类的伪标签进行重采样。我们的重采样策略优先选择精确率高、召回率低的类的伪标签,在不显著降低精确率的前提下提升了整体召回率。我们在文档级和生物医学关系抽取数据集上开展了实验,结果表明,当训练数据标注不完整时,我们提出的自训练框架在Re-DocRED和ChemDisgene数据集上的表现始终优于现有主流方法。我们的代码发布于https://github.com/DAMO-NLP-SG/CAST。

英文摘要:

Relation extraction (RE) aims to extract relations from sentences and documents. Existing relation extraction models typically rely on supervised machine learning. However, recent studies showed that many RE datasets are incompletely annotated. This is known as the false negative problem in which valid relations are falsely annotated as 'no_relation'. Models trained with such data inevitably make similar mistakes during the inference stage. Self-training has been proven effective in alleviating the false negative problem. However, traditional self-training is vulnerable to confirmation bias and exhibits poor performance in minority classes. To overcome this limitation, we proposed a novel class-adaptive re-sampling self-training framework. Specifically, we re-sampled the pseudo-labels for each class by precision and recall scores. Our re-sampling strategy favored the pseudo-labels of classes with high precision and low recall, which improved the overall recall without significantly compromising precision. We conducted experiments on document-level and biomedical relation extraction datasets, and the results showed that our proposed self-training framework consistently outperforms existing competitive methods on the Re-DocRED and ChemDisgene datasets when the training data are incompletely annotated. Our code is released at https://github.com/DAMO-NLP-SG/CAST.

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

↑