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

Continual Named Entity Recognition without Catastrophic Forgetting

About

Continual Named Entity Recognition (CNER) is a burgeoning area, which involves updating an existing model by incorporating new entity types sequentially. Nevertheless, continual learning approaches are often severely afflicted by catastrophic forgetting. This issue is intensified in CNER due to the consolidation of old entity types from previous steps into the non-entity type at each step, leading to what is known as the semantic shift problem of the non-entity type. In this paper, we introduce a pooled feature distillation loss that skillfully navigates the trade-off between retaining knowledge of old entity types and acquiring new ones, thereby more effectively mitigating the problem of catastrophic forgetting. Additionally, we develop a confidence-based pseudo-labeling for the non-entity type, \emph{i.e.,} predicting entity types using the old model to handle the semantic shift of the non-entity type. Following the pseudo-labeling process, we suggest an adaptive re-weighting type-balanced learning strategy to handle the issue of biased type distribution. We carried out comprehensive experiments on ten CNER settings using three different datasets. The results illustrate that our method significantly outperforms prior state-of-the-art approaches, registering an average improvement of $6.3$\% and $8.0$\% in Micro and Macro F1 scores, respectively.

Duzhen Zhang, Wei Cong, Jiahua Dong, Yahan Yu, Xiuyi Chen, Yonggang Zhang, Zhen Fang• 2023

Related benchmarks

TaskDatasetResultRank
Named Entity RecognitionOntoNotes 5
A_T Score48.36
44
Named Entity Recognitioni2b2
A_T26.38
44
Named Entity RecognitionOntoNotes FG-8-PG-1 5.0 (test)
A_T Score40.58
11
Named Entity RecognitionOntoNotes FG-8-PG-2 5.0 (test)
A_T50.7
11
Named Entity RecognitionOntoNotes FG-1-PG-1 5.0 (test)
A_T29.5
11
Named Entity RecognitionOntoNotes FG-2-PG-2 5.0 (test)
A_T Score41.01
11
Named Entity Recognitioni2b2 FG-2-PG-2 bert-base-cased
At31.44
11
Named Entity Recognitioni2b2 bert-base-cased (FG-1-PG-1)
At Score17.72
11
Named Entity Recognitioni2b2 FG-8-PG-1 bert-base-cased
At Score5
11
Named Entity Recognitioni2b2 bert-base-cased (FG-8-PG-2)
At Score23.03
11
Showing 10 of 10 rows

Other info

Follow for update