SOTAVerified

NuNER: Entity Recognition Encoder Pre-training via LLM-Annotated Data

2024-02-23Code Available1· sign in to hype

Sergei Bogdanov, Alexandre Constantin, Timothée Bernard, Benoit Crabbé, Etienne Bernard

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

Large Language Models (LLMs) have shown impressive abilities in data annotation, opening the way for new approaches to solve classic NLP problems. In this paper, we show how to use LLMs to create NuNER, a compact language representation model specialized in the Named Entity Recognition (NER) task. NuNER can be fine-tuned to solve downstream NER problems in a data-efficient way, outperforming similar-sized foundation models in the few-shot regime and competing with much larger LLMs. We find that the size and entity-type diversity of the pre-training dataset are key to achieving good performance. We view NuNER as a member of the broader family of task-specific foundation models, recently unlocked by LLMs.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
Few-NERD (SUP)NuNERF1-Measure69.4Unverified
NCBI DiseaseNuNER Zero SpanF161.1Unverified
Ontonotes v5 (English)NuNERF189.1Unverified

Reproductions