arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04437cs.CLcs.LG

学习使用语言模型提取结构化实体

Learning to Extract Structured Entities Using Language Models

McGill University(麦吉尔大学);Mila - Quebec AI Institute(米拉-魁北克人工智能研究所);Microsoft Research(微软研究院);ETH Zürich(苏黎世联邦理工学院)

Haolun Wu, Ye Yuan, Liana Mikaelyan, Alexander Meulemans, Xue Liu, James Hensman, Bhaskar Mitra

AI总结:

本文提出以实体为中心的结构化实体抽取任务、AESOP 评估指标及多阶段 MuSEE 模型,实验表明其效果与效率优于基线。

摘要:

机器学习的最新进展显著影响了信息抽取领域,其中语言模型(LMs)在从非结构化文本中提取结构化信息方面发挥着关键作用。以往工作通常将信息抽取表示为以三元组为中心的任务,并使用精确率和召回率等经典指标进行评估。我们将该任务重新表述为以实体为中心,从而能够使用多样化指标,从不同视角提供更多洞察。我们通过引入结构化实体抽取,并提出旨在恰当评估模型性能的近似实体集重叠(Approximate Entity Set OverlaP,AESOP)指标,为该领域作出贡献。随后,我们介绍一种新的多阶段结构化实体抽取(Multistage Structured Entity Extraction,MuSEE)模型,该模型利用 LMs 的能力,通过将抽取任务分解为多个阶段来提升效果和效率。定量评估和人工并排评估证实,我们的模型优于基线,为结构化实体抽取未来的发展提供了有前景的方向。我们的源代码和数据集可在 https://github.com/microsoft/Structured-Entity-Extraction 获取。

英文摘要:

Recent advances in machine learning have significantly impacted the field of information extraction, with Language Models (LMs) playing a pivotal role in extracting structured information from unstructured text. Prior works typically represent information extraction as triplet-centric and use classical metrics such as precision and recall for evaluation. We reformulate the task to be entity-centric, enabling the use of diverse metrics that can provide more insights from various perspectives. We contribute to the field by introducing Structured Entity Extraction and proposing the Approximate Entity Set OverlaP (AESOP) metric, designed to appropriately assess model performance. Later, we introduce a new Multistage Structured Entity Extraction (MuSEE) model that harnesses the power of LMs for enhanced effectiveness and efficiency by decomposing the extraction task into multiple stages. Quantitative and human side-by-side evaluations confirm that our model outperforms baselines, offering promising directions for future advancements in structured entity extraction. Our source code and datasets are available at https://github.com/microsoft/Structured-Entity-Extraction.

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

↑