ACL 2024findings2 citations

Flexible Weight Tuning and Weight Fusion Strategies for Continual Named Entity Recognition

Yahan Yu, Duzhen Zhang, Xiuyi Chen, Chenhui Chu

Abstract

Continual Named Entity Recognition (CNER) is dedicated to sequentially learning new entity types while mitigating catastrophic forgetting of old entity types. Traditional CNER approaches commonly employ knowledge distillation to retain old knowledge within the current model. However, because only the representations of old and new models are constrained to be consistent, the reliance solely on distillation in existing methods still suffers from catastrophic forgetting. To further alleviate the forgetting issue of old entity types, this paper introduces flexible Weight Tuning (WT) and Weight Fusion (WF) strategies for CNER. The WT strategy, applied at each training step, employs a learning rate schedule on the parameters of the current model. After learning the current task, the WF strategy dynamically integrates knowledge from both the current and previous models for inference. Notably, these two strategies are model-agnostic and seamlessly integrate with existing State-Of-The-Art (SOTA) models. Extensive experiments demonstrate that the WT and WF strategies consistently enhance the performance of previous SOTA methods across ten CNER settings in three datasets.

BibTeX
@inproceedings{yu-etal-2024-flexible,
    title = "Flexible Weight Tuning and Weight Fusion Strategies for Continual Named Entity Recognition",
    author = "Yu, Yahan  and
      Zhang, Duzhen  and
      Chen, Xiuyi  and
      Chu, Chenhui",
    editor = "Ku, Lun-Wei  and
      Martins, Andre  and
      Srikumar, Vivek",
    booktitle = "Findings of the Association for Computational Linguistics: ACL 2024",
    month = aug,
    year = "2024",
    address = "Bangkok, Thailand",
    publisher = "Association for Computational Linguistics",
    url = "https://aclanthology.org/2024.findings-acl.79/",
    doi = "10.18653/v1/2024.findings-acl.79",
    pages = "1351--1358"
}