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

Task Relation Distillation and Prototypical Pseudo Label for Incremental Named Entity Recognition

About

Incremental Named Entity Recognition (INER) involves the sequential learning of new entity types without accessing the training data of previously learned types. However, INER faces the challenge of catastrophic forgetting specific for incremental learning, further aggravated by background shift (i.e., old and future entity types are labeled as the non-entity type in the current task). To address these challenges, we propose a method called task Relation Distillation and Prototypical pseudo label (RDP) for INER. Specifically, to tackle catastrophic forgetting, we introduce a task relation distillation scheme that serves two purposes: 1) ensuring inter-task semantic consistency across different incremental learning tasks by minimizing inter-task relation distillation loss, and 2) enhancing the model's prediction confidence by minimizing intra-task self-entropy loss. Simultaneously, to mitigate background shift, we develop a prototypical pseudo label strategy that distinguishes old entity types from the current non-entity type using the old model. This strategy generates high-quality pseudo labels by measuring the distances between token embeddings and type-wise prototypes. We conducted extensive experiments on ten INER settings of three benchmark datasets (i.e., CoNLL2003, I2B2, and OntoNotes5). The results demonstrate that our method achieves significant improvements over the previous state-of-the-art methods, with an average increase of 6.08% in Micro F1 score and 7.71% in Macro F1 score.

Duzhen Zhang, Hongliu Li, Wei Cong, Rongtao Xu, Jiahua Dong, Xiuyi Chen• 2023

Related benchmarks

TaskDatasetResultRank
Named Entity Recognitioni2b2
A_T42.98
44
Named Entity RecognitionOntoNotes 5
A_T Score50.34
44
Named Entity RecognitionOntoNotes FG-8-PG-2 5.0 (test)
A_T52.54
11
Named Entity Recognitioni2b2 FG-2-PG-2 bert-base-cased
At44.53
11
Named Entity RecognitionOntoNotes FG-8-PG-1 5.0 (test)
A_T Score41.58
11
Named Entity Recognitioni2b2 FG-8-PG-1 bert-base-cased
At Score26.83
11
Named Entity Recognitioni2b2 bert-base-cased (FG-8-PG-2)
At Score41.43
11
Named Entity RecognitionOntoNotes FG-1-PG-1 5.0 (test)
A_T34.84
11
Named Entity RecognitionOntoNotes FG-2-PG-2 5.0 (test)
A_T Score46.95
11
Named Entity Recognitioni2b2 bert-base-cased (FG-1-PG-1)
At Score28.05
11
Showing 10 of 10 rows

Other info

Follow for update