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

Context-NER:大规模上下文短语生成

Context-NER : Contextual Phrase Generation at Scale

Arizona State University(亚利桑那州立大学);Georgia Institute of Technology(佐治亚理工学院)

Himanshu Gupta, Shreyas Verma, Santosh Mashetty, Swaroop Mishra

AI总结:

本文提出CONTEXT-NER实体上下文生成任务与大规模金融领域EDGAR10-Q数据集,构建基线方法并测试多款模型,发现预微调后的T5-large在下游金融任务上性能远超原始版本及专用大模型BloombergGPT-50B。

摘要:

近年来,命名实体识别(NER)取得了显著进展,众多最先进(SOTA)模型实现了高性能。然而,极少有研究聚焦于实体上下文的生成。本文中,我们提出CONTEXT-NER任务,旨在为句子中的实体生成相关上下文,该上下文是描述实体的短语,但不一定存在于原句中。为推动该任务的研究,我们还构建了EDGAR10-Q数据集,它包含市值排名前1500的上市公司的年度和季度报告。该数据集是同类数据集中规模最大的,包含100万条句子、280万个实体,平均每条句子含35个词元,是一个颇具挑战性的数据集。我们提出了一种基线方法,将短语生成算法与基于220M参数语言模型的推理相结合,在测试集上取得了27%的ROUGE-L分数。此外,我们用ChatGPT进行了一次单样本推理,获得了30%的ROUGE-L分数,凸显了该数据集的难度。我们还评估了T5、BART等模型,它们在EDGAR10-Q上进行监督微调后,最高ROUGE-L分数达到49%。我们还发现,T5-large在EDGAR10-Q上进行预微调后,在Headline、FPB、FiQA SA等下游金融任务上取得了SOTA结果,比原始版本性能高出10.81分。令人意外的是,这个参数规模小66倍的预微调模型,性能还比金融领域专用大语言模型BloombergGPT-50B高出15分。我们希望本研究的数据集和生成的相关成果能推动该方向的进一步研究,助力开发更先进的金融文本分析语言模型。

英文摘要:

Named Entity Recognition (NER) has seen significant progress in recent years, with numerous state-of-the-art (SOTA) models achieving high performance. However, very few studies have focused on the generation of entities' context. In this paper, we introduce CONTEXT-NER, a task that aims to generate the relevant context for entities in a sentence, where the context is a phrase describing the entity but not necessarily present in the sentence. To facilitate research in this task, we also present the EDGAR10-Q dataset, which consists of annual and quarterly reports from the top 1500 publicly traded companies. The dataset is the largest of its kind, containing 1M sentences, 2.8M entities, and an average of 35 tokens per sentence, making it a challenging dataset. We propose a baseline approach that combines a phrase generation algorithm with inferencing using a 220M language model, achieving a ROUGE-L score of 27% on the test split. Additionally, we perform a one-shot inference with ChatGPT, which obtains a 30% ROUGE-L, highlighting the difficulty of the dataset. We also evaluate models such as T5 and BART, which achieve a maximum ROUGE-L of 49% after supervised finetuning on EDGAR10-Q. We also find that T5-large, when pre-finetuned on EDGAR10-Q, achieve SOTA results on downstream finance tasks such as Headline, FPB, and FiQA SA, outperforming vanilla version by 10.81 points. To our surprise, this 66x smaller pre-finetuned model also surpasses the finance-specific LLM BloombergGPT-50B by 15 points. We hope that our dataset and generated artifacts will encourage further research in this direction, leading to the development of more sophisticated language models for financial text analysis

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

↑