arXiv:2411.05895cs.CL2024-11被引 4

小模型和大模型协同提升文档级事件论元抽取效果

One Small and One Large for Document-level Event Argument Extraction

  • 用小模型捕捉事件共现与结构关系,减少冗余干扰
  • 大模型通过新提示实现生成式抽取,性能超越传统微调
  • 小模型的有效设计可迁移至大模型,提升整体表现

文档级事件论元抽取(EAE)因输入长度增加面临两大挑战:难以区分事件语义边界,以及冗余信息干扰。为此,我们提出两种方法。第一种是基于小语言模型(SLMs)的共现与结构事件论元抽取模型(CsEAE),包含共现感知模块,通过上下文标注和共现事件提示整合全文事件信息;以及结构感知模块,通过建立触发词所在句与其他句子间的结构关系来降低干扰。第二种方法为大语言模型(LLMs)设计新提示,将抽取任务转化为生成任务,弥补其在监督微调(SFT)下性能不足的问题,并在多个数据集上微调以提升泛化能力。最后,我们将CsEAE的洞察应用于LLMs,进一步提升性能。在Rams、WikiEvents和MLEE数据集上的实验表明,CsEAE在Arg-C F1上分别优于基线PAIE 2.1%、2.3%和3.2%;同时,优化后的LLM在文档级数据集上的表现已接近小模型水平。

原文摘要 · Abstract (English)

Document-level Event Argument Extraction (EAE) faces two challenges due to increased input length: 1) difficulty in distinguishing semantic boundaries between events, and 2) interference from redundant information. To address these issues, we propose two methods. The first method introduces the Co and Structure Event Argument Extraction model (CsEAE) based on Small Language Models (SLMs). CsEAE includes a co-occurrences-aware module, which integrates information about all events present in the current input through context labeling and co-occurrences event prompts extraction. Additionally, CsEAE includes a structure-aware module that reduces interference from redundant information by establishing structural relationships between the sentence containing the trigger and other sentences in the document. The second method introduces new prompts to transform the extraction task into a generative task suitable for Large Language Models (LLMs), addressing gaps in EAE performance using LLMs under Supervised Fine-Tuning (SFT) conditions. We also fine-tuned multiple datasets to develop an LLM that performs better across most datasets. Finally, we applied insights from CsEAE to LLMs, achieving further performance improvements. This suggests that reliable insights validated on SLMs are also applicable to LLMs. We tested our models on the Rams, WikiEvents, and MLEE datasets. The CsEAE model achieved improvements of 2.1\%, 2.3\%, and 3.2\% in the Arg-C F1 metric compared to the baseline, PAIE~\cite{PAIE}. For LLMs, we demonstrated that their performance on document-level datasets is comparable to that of SLMs~\footnote{All code is available at https://github.com/simon-p-j-r/CsEAE}.

事件抽取小模型大模型生成式抽取

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