arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.06970cs.CL

混合多阶段解码用于少样本命名实体识别与实体感知对比学习

Hybrid Multi-stage Decoding for Few-shot NER with Entity-aware Contrastive Learning

University of Chinese Academy of Sciences(中国科学院大学);Institute of Information Engineering, Chinese Academy of Sciences(中国科学院信息工程研究所)

Congying Liu, Gaosheng Wang, Peipei Liu, Xingyuan Wei, Hongsong Zhu

AI总结:

提出MsFNER,将少样本NER分为跨度检测与分类两阶段,结合对比学习和KNN,在FewNERD上取得先进性能。

摘要:

少样本命名实体识别能够基于少量标注示例识别新类型的命名实体。以往采用词级或跨度级度量学习的方法面临计算负担和大量负样本跨度的问题。本文提出了混合多阶段解码用于少样本命名实体识别与实体感知对比学习(MsFNER),该方法将通用命名实体识别分解为两个阶段:实体跨度检测和实体分类。引入MsFNER包含3个过程:训练、微调和推理。在训练过程中,我们在源域上使用元学习分别训练并获取最优的实体跨度检测模型和实体分类模型,其中我们创建了一个对比学习模块以增强实体分类的实体表示。在微调过程中,我们在目标域的支撑数据集上对两个模型进行微调。在推理过程中,对于未标注数据,我们首先检测实体跨度,然后由实体分类模型和KNN共同确定实体跨度。我们在公开的FewNERD数据集上进行了实验,结果证明了MsFNER的先进性。

英文摘要:

Few-shot named entity recognition can identify new types of named entities based on a few labeled examples. Previous methods employing token-level or span-level metric learning suffer from the computational burden and a large number of negative sample spans. In this paper, we propose the Hybrid Multi-stage Decoding for Few-shot NER with Entity-aware Contrastive Learning (MsFNER), which splits the general NER into two stages: entity-span detection and entity classification. There are 3 processes for introducing MsFNER: training, finetuning, and inference. In the training process, we train and get the best entity-span detection model and the entity classification model separately on the source domain using meta-learning, where we create a contrastive learning module to enhance entity representations for entity classification. During finetuning, we finetune the both models on the support dataset of target domain. In the inference process, for the unlabeled data, we first detect the entity-spans, then the entity-spans are jointly determined by the entity classification model and the KNN. We conduct experiments on the open FewNERD dataset and the results demonstrate the advance of MsFNER.

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

↑