多智能体协作提升零样本命名实体识别准确率
A Cooperative Multi-Agent Framework for Zero-Shot Named Entity Recognition
- 拆解任务为实体识别与特征提取,捕捉上下文关联
- 自评演示有效性,避免错误示例干扰模型
- 跨领域适配强,支持多种大模型和少样本场景
零样本命名实体识别(NER)旨在从无标注文本中构建实体识别系统,面临人工干预极少的挑战。现有方法虽借助大语言模型(LLM)与提示模板实现自学习,但仍存在两大问题:(i) 忽视实体周围上下文的关联性,导致类型误判或实体遗漏;(ii) 通过浅层相似性检索的演示示例滥用,严重误导模型推理。本文提出协同多智能体框架(CMAS),由四个智能体组成:自标注器、类型相关特征提取器、演示甄别器与整体预测器。为显式建模上下文关联,将NER重构为实体识别与目标句内类型特征识别两个子任务。通过建立演示甄别器,引入自反思机制,自动评估演示对当前句子的帮助程度。实验表明,CMAS在六个基准上显著提升零样本NER性能,涵盖领域特定与通用场景,并在少样本设置及不同LLM主干网络下均表现有效。
原文摘要 · Abstract (English)
Zero-shot named entity recognition (NER) aims to develop entity recognition systems from unannotated text corpora. This task presents substantial challenges due to minimal human intervention. Recent work has adapted large language models (LLMs) for zero-shot NER by crafting specialized prompt templates. It advances model self-learning abilities by incorporating self-annotated demonstrations. However, two important challenges persist: (i) Correlations between contexts surrounding entities are overlooked, leading to wrong type predictions or entity omissions. (ii) The indiscriminate use of task demonstrations, retrieved through shallow similarity-based strategies, severely misleads LLMs during inference. In this paper, we introduce the cooperative multi-agent system (CMAS), a novel framework for zero-shot NER that uses the collective intelligence of multiple agents to address the challenges outlined above. CMAS has four main agents: (i) a self-annotator, (ii) a type-related feature (TRF) extractor, (iii) a demonstration discriminator, and (iv) an overall predictor. To explicitly capture correlations between contexts surrounding entities, CMAS reformulates NER into two subtasks: recognizing named entities and identifying entity type-related features within the target sentence. To enable controllable utilization of demonstrations, a demonstration discriminator is established to incorporate the self-reflection mechanism, automatically evaluating helpfulness scores for the target sentence. Experimental results show that CMAS significantly improves zero-shot NER performance across six benchmarks, including both domain-specific and general-domain scenarios. Furthermore, CMAS demonstrates its effectiveness in few-shot settings and with various LLM backbones.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。