当前位置：首页 > article >正文

NLP-UIE(Universal Information Extraction)

article 2025/3/4 8:34:55

介绍：一种模型同时完成4种NLP任务。
原理：将几种NLP任务的输入和输出进行统一，联合训练。
任务：Entity提取、Relation关系、Event事件、Sentiment情感分类。
输入统一：通过prompt的方式去控制需要抽取的信息，（在信息加入任务类型，一同作为模型的输入,样本输入是[CLS]+prompt+[SEP]+text+[SEP]）
输出统一：使用SEL(Structed extraction Language)模板来统一输出。

SpotName：实体类型
AssoName：关系
InfoSpan：value。

好处：

多个任务共享表征，可以提升效果。
减少模型维护。
可以达到zero-shot的效果。
能实现多标签的提取。

如何接一个位置对应多个标签问题？

介绍：新增了一个线性层用于全局 Span 判断。例如已经得到[O,B-school,I-school,I-school,I-school,I-school,I-school]，然后可以使用不同位置组合使用进行MLP映射到新的label，例如位置4和5虽然都是I-school，但是经过MLP后可能是B-type]

实现：把位置4和位置5的向量concat后再MLP就得到span打分后的真实标签。

其他：

标记法：「BIO 标记法」、「BIOES 标记法」
传统的NLP无法解决一字多标签的问题。
源码：这是一个抽取式模型，而非论文中所述的生成式，是以ERNIE编码器，接双指针解码，这个结构不是两年前香侬AI提出的MRC模型吗，区别是在这个“UIE”中似乎只有start和end，少了交叉的部分，但是思想还是双指针的思想。