Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

E-NER: Evidential Deep Learning for Trustworthy Named Entity Recognition

About

Most named entity recognition (NER) systems focus on improving model performance, ignoring the need to quantify model uncertainty, which is critical to the reliability of NER systems in open environments. Evidential deep learning (EDL) has recently been proposed as a promising solution to explicitly model predictive uncertainty for classification tasks. However, directly applying EDL to NER applications faces two challenges, i.e., the problems of sparse entities and OOV/OOD entities in NER tasks. To address these challenges, we propose a trustworthy NER framework named E-NER by introducing two uncertainty-guided loss terms to the conventional EDL, along with a series of uncertainty-guided training strategies. Experiments show that E-NER can be applied to multiple NER paradigms to obtain accurate uncertainty estimation. Furthermore, compared to state-of-the-art baselines, the proposed method achieves a better OOV/OOD detection performance and better generalization ability on OOV entities.

Zhen Zhang, Mengting Hu, Shiwan Zhao, Minlie Huang, Haotian Wang, Lemao Liu, Zhirui Zhang, Zhe Liu, Bingzhe Wu• 2023

Related benchmarks

TaskDatasetResultRank
Named Entity RecognitionCoNLL 2003 (test)
F1 Score93.15
539
Named Entity RecognitionOntoNotes 5.0 (test)
F1 Score90.64
90
Named Entity RecognitionTwitter NER
F1 Score75.64
14
OOD DetectionOOD
AUC (Confidence)0.822
9
OOV DetectionOOV
AUC (Confidence)74.3
9
Typos DetectionTypos
AUC (Confidence)82.5
9
Named Entity RecognitionCoNLL Typos 2003
F1 Score83.64
8
Named Entity RecognitionCoNLL OOV 2003
F1 Score69.71
8
Showing 8 of 8 rows

Other info

Code

Follow for update