arXiv:2411.11623cs.CL2024-11中稿 · IEEE/ACM Transacti…被引 6

解决联邦学习中实体类型持续新增的遗忘问题

Federated Incremental Named Entity Recognition

  • 设计局部-全局遗忘防御机制,保留旧知识
  • 新实体类型自动识别,支持动态加入
  • 适合医疗、金融等持续更新的场景

联邦命名实体识别(FNER)通过聚合分散客户端的模型更新来提升训练效果,而无需共享私有数据。然而,现有方法假设实体类型和客户端固定,难以应对实际场景。在更真实的设置中,客户端会持续接收新实体类型,同时新客户端可能不规律地加入训练。这种称为联邦增量命名实体识别的场景,导致全局模型在客户端内部和跨客户端均出现知识遗忘。为此,我们提出局部-全局遗忘防御(LGFD)模型:为缓解客户端内遗忘,设计结构化知识蒸馏损失以保持特征空间结构,并引入伪标签引导的跨类型对比损失以增强类别区分能力;为应对客户端间遗忘,提出任务切换监控器,在隐私保护下自动识别新实体类型,并存储最新旧全局模型用于知识蒸馏与伪标签生成。实验表明,LGFD显著优于基线方法。

原文摘要 · Abstract (English)

Federated Named Entity Recognition (FNER) boosts model training within each local client by aggregating the model updates of decentralized local clients, without sharing their private data. However, existing FNER methods assume fixed entity types and local clients in advance, leading to their ineffectiveness in practical applications. In a more realistic scenario, local clients receive new entity types continuously, while new local clients collecting novel data may irregularly join the global FNER training. This challenging setup, referred to here as Federated Incremental NER, renders the global model suffering from heterogeneous forgetting of old entity types from both intra-client and inter-client perspectives. To overcome these challenges, we propose a Local-Global Forgetting Defense (LGFD) model. Specifically, to address intra-client forgetting, we develop a structural knowledge distillation loss to retain the latent space's feature structure and a pseudo-label-guided inter-type contrastive loss to enhance discriminative capability over different entity types, effectively preserving previously learned knowledge within local clients. To tackle inter-client forgetting, we propose a task switching monitor that can automatically identify new entity types under privacy protection and store the latest old global model for knowledge distillation and pseudo-labeling. Experiments demonstrate significant improvement of our LGFD model over comparison methods.

联邦学习命名实体识别增量学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。