GPT-NER：使用大型语言模型进行命名实体识别

news2024/10/5 18:31:00

讲在前面，chatgpt出来的时候就想过将其利用在信息抽取方面，后续也发现了不少基于这种大语言模型的信息抽取的论文，比如之前收集过的：

https://github.com/cocacola-lab/GPT4IE
https://github.com/RidongHan/Evaluation-of-ChatGPT-on-Information-Extraction
https://github.com/cocacola-lab/ChatIE
Unified Text Structuralization with Instruction-tuned Language Models

接下来继续介绍另一篇论文。

GPT-NER：通过大型语言模型的命名实体识别

GPT-NER: Named Entity Recognition via Large Language Models

https://arxiv.org/pdf/2304.10428v1.pdf

https://github.com/ShuheWang1998/GPT-NER

Part1前言

为什么使用大语言模型在NER上的表现仍然明显低于普遍的基线？

由于NER和LLMs这两个任务之间的差距：前者本质上是一个序列标记任务，而后者是一个文本生成模型。

怎么解决上述的问题呢？

GPT-NER通过将序列标签任务转换为一个可以被LLMs轻松适应的生成任务来弥补这一差距，例如，在输入文本Columbus是一个城市中寻找位置实体的任务被转换为生成文本序列@@Columbus##是一个城市，其中特殊标记@@##标志着要提取的实体。
为了有效地解决LLMs的幻觉问题，即LLMs有一个强烈的倾向，即过度自信地将NULL输入标记为实体，我们提出了一个自我验证策略，即提示LLMs询问自己所提取的实体是否符合标记的实体标签。

Part2介绍

GPTNER遵循语境学习的一般范式，可以分解为三个步骤：

（1）构建提示：对于一个给定的输入句子X，我们为X构建一个提示（用Prompt(X) 来表示）；
（2）将提示输入到大语言模型得到生成的文本序列W = {w1 ,..., wn };
（3）将文本序列W转化为实体标签序列，以获得最终的结果。

如图所示：第一句话：你是一个优秀的语言学家；第二句话：任务是从给定的句子中标记xxx实体。接下来是一些例子，然后给树了一些例子。最后再输入自己想要提取实体的句子得到结果。很容易发现，每次只能提取一种实体，因此要提取出所有的实体，必须遍历实体列表。例如GPT-3，对提示的长度有一个硬性的限制（例如GPT-3的4096个tokens)。鉴于这种有限的标记数量，我们不可能在一个提示中包括对所有实体类型的描述和演示。

1怎么提供实例样本？

如图所示：

1、一个已经训练好的ner模型提取训练数据中的实体，并为每一个实体构建(实体，句子)对。
2、将句子输入的模型中并获取实体的表示。
3、通过knn找到和实体向量最接近的几个邻居，将得到的句子视为样例。

2怎么进行自我验证?

Prompt:
I am an excellent linguist. The task is to label location entities in the given sentence.
Below are some examples.
Input:Columbus is a city
Output:@@Columbus## is a city
Input:Rare Hendrix song sells for $17
Output:
GPT-3 Output:
Rare @@Hendrix## song sells for $17

过度预测是指将不是实体的预测为实体。如上面的例子：Hendrix被识别为一个location实体，这显然是不对的。自我验证策略：给定一个由LLM提取的实体，我们要求LLM进一步验证该提取的实体是否正确，用是或否回答。比如：

“The task is to verify whether the word is a location entity extracted from the given sentence”
(1) “The input sentence: Only France and Britain backed Fischler’s proposal”,
(2) “Is the word "France" in the input sentence a location entity? Please answer with yes or no”.
(3) Yes