ProgGen: Generating Named Entity Recognition Datasets Step-by-step with Self-Reflexive Large Language Models

Yuzhao Heng, Chunyuan Deng, Yitong Li, Yue Yu, Yinghao Li +2 more
3/17/2024
cs.CLcs.LG

No official code found. Best implementation: StefanHeng/ProgGen (17 stars, MIT, last commit 2024-03-29).

Abstract

Although Large Language Models (LLMs) exhibit remarkable adaptability across domains, these models often fall short in structured knowledge extraction tasks such as named entity recognition (NER). This paper explores an innovative, cost-efficient strategy to harness LLMs with modest NER capabilities for producing superior NER datasets. Our approach diverges from the basic class-conditional prompts by instructing LLMs to self-reflect on the specific domain, thereby generating domain-relevant attributes (such as category and emotions for movie reviews), which are utilized for creating attribute-rich training data. Furthermore, we preemptively generate entity terms and then develop NER context data around these entities, effectively bypassing the LLMs' challenges with complex structures. Our experiments across both general and niche domains reveal significant performance enhancements over conventional data generation methods while being more cost-effective than existing alternatives.

View on arXivView PDF

Code Implementations(1)

StefanHeng/ProgGenCommunity100%
MIT

Code for paper "ProgGen: Generating Named Entity Recognition Datasets Step-by-step with Self-Reflexive Large Language Models"

174PythonMar 14, 20242 years agoMIT
data-generationefficient-nlpfew-shot-learninglarge-language-modelslow-resource-nlp+3 more

Similar Papers

Cite this paper

@article{heng2024proggen,
  title  = {ProgGen: Generating Named Entity Recognition Datasets Step-by-step with Self-Reflexive Large Language Models},
  author = {Yuzhao Heng and Chunyuan Deng and Yitong Li and Yue Yu and Yinghao Li and Rongzhi Zhang and Chao Zhang},
  year   = {2024},
  eprint = {2403.11103},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.11103v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.11103.svg)](https://codeofpaper.com/paper/2403.11103)

Discussion