arXiv:2409.13717cs.CLcs.AI2024-09被引 1

提出新框架,让模型更准地从文档中提取实体关系。

DiVA-DocRE: A Discriminative and Voice-Aware Paradigm for Document-Level Relation Extraction

  • 将文档级关系抽取转为判别式任务,聚焦每种关系
  • 在Re-DocRED和DocRED上达当前最优性能
  • 特别关注主动被动语态,适合真实场景应用

大型语言模型在文本理解与生成方面的能力显著提升了信息抽取水平。文档级关系三元组抽取(DocRTE)是信息系统中的关键任务,旨在从文档中提取实体及其语义关系。然而,现有方法多针对句子级关系抽取(SentRTE),通常只处理单句内的有限关系与三元组。部分方法将关系作为提示模板中的候选项,导致处理效率低且性能不佳。为此,我们提出判别式与语音感知范式DiVA。DiVA仅需两步:先进行文档级关系抽取(DocRE),再根据关系识别主语与宾语实体,无需额外处理,直接输入文档即可获得三元组。该流程更贴近真实场景。创新之处在于将DocRE转化为判别任务,使模型关注每种关系及三元组中常被忽视的主动/被动语态问题。在Re-DocRED和DocRED数据集上的实验表明,该方法在文档级关系三元组抽取任务上达到当前最优效果。

原文摘要 · Abstract (English)

The remarkable capabilities of Large Language Models (LLMs) in text comprehension and generation have revolutionized Information Extraction (IE). One such advancement is in Document-level Relation Triplet Extraction (DocRTE), a critical task in information systems that aims to extract entities and their semantic relationships from documents. However, existing methods are primarily designed for Sentence level Relation Triplet Extraction (SentRTE), which typically handles a limited set of relations and triplet facts within a single sentence. Additionally, some approaches treat relations as candidate choices integrated into prompt templates, resulting in inefficient processing and suboptimal performance when determining the relation elements in triplets. To address these limitations, we introduce a Discriminative and Voice Aware Paradigm DiVA. DiVA involves only two steps: performing document-level relation extraction (DocRE) and then identifying the subject object entities based on the relation. No additional processing is required simply input the document to directly obtain the triplets. This streamlined process more accurately reflects real-world scenarios for triplet extraction. Our innovation lies in transforming DocRE into a discriminative task, where the model pays attention to each relation and to the often overlooked issue of active vs. passive voice within the triplet. Our experiments on the Re-DocRED and DocRED datasets demonstrate state-of-the-art results for the DocRTE task.

关系抽取大模型文档级

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。