HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

CrossWeigh: Training Named Entity Tagger from Imperfect Annotations

Zihan Wang; Jingbo Shang; Liyuan Liu; Lihao Lu; Jiacheng Liu; Jiawei Han

CrossWeigh: Training Named Entity Tagger from Imperfect Annotations

Abstract

Everyone makes mistakes. So do human annotators when curating labels for named entity recognition (NER). Such label mistakes might hurt model training and interfere model comparison. In this study, we dive deep into one of the widely-adopted NER benchmark datasets, CoNLL03 NER. We are able to identify label mistakes in about 5.38% test sentences, which is a significant ratio considering that the state-of-the-art test F1 score is already around 93%. Therefore, we manually correct these label mistakes and form a cleaner test set. Our re-evaluation of popular models on this corrected test set leads to more accurate assessments, compared to those on the original test set. More importantly, we propose a simple yet effective framework, CrossWeigh, to handle label mistakes during NER model training. Specifically, it partitions the training data into several folds and train independent NER models to identify potential mistakes in each fold. Then it adjusts the weights of training data accordingly to train the final NER model. Extensive experiments demonstrate significant improvements of plugging various NER models into our proposed framework on three datasets. All implementations and corrected test set are available at our Github repo: https://github.com/ZihanWangKi/CrossWeigh.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
named-entity-recognition-ner-on-conll-2003CrossWeigh + Pooled Flair
F1: 93.43
named-entity-recognition-on-conllPooled Flair
F1: 94.13
named-entity-recognition-on-conllCrossWeigh + Pooled Flair
F1: 94.28
named-entity-recognition-on-wnut-2017CrossWeigh + Pooled Flair
F1: 50.03

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
CrossWeigh: Training Named Entity Tagger from Imperfect Annotations | Papers | HyperAI