arXiv:2502.15677cs.CLcs.AI2025-02被引 3

FLEKE让多个机构在保护隐私前提下高效协同更新大模型知识。

FLEKE: Federated Locate-then-Edit Knowledge Editing

  • 分两阶段进行:本地计算知识向量,再通过相似度检索复用。
  • 性能保留96%以上,计算开销比基线降低约一半。
  • 适合医疗、金融等需隐私保护的多机构协作场景。

定位-编辑知识编辑(LEKE)是无需全量重训练即可更新大语言模型的关键技术。然而现有方法假设单用户场景,在真实多客户端环境中效率低下:分散机构独立更新重叠知识,导致冗余的中介知识向量(MKV)计算并引发隐私问题。为此,我们提出联邦定位-编辑知识编辑(FLEKE),支持多客户端在保护隐私的前提下协同执行LEKE,显著降低计算开销。我们设计了两阶段框架FedEdit:第一阶段,各客户端本地执行LEKE并上传计算出的MKVs;第二阶段,不依赖服务器集中共享,而是允许客户端基于余弦相似度检索相关MKVs,实现知识复用与再编辑,减少重复计算。在两个基准数据集上的实验表明,FedEdit保持了非联邦LEKE超过96%的性能,相比基于FedAvg的基线提升约两倍。此外,我们发现,在我们的FedEdit框架下,MEMIT比PMET表现更稳定。代码已开源:https://github.com/zongkaiz/FLEKE。

原文摘要 · Abstract (English)

Locate-then-Edit Knowledge Editing (LEKE) is a key technique for updating large language models (LLMs) without full retraining. However, existing methods assume a single-user setting and become inefficient in real-world multi-client scenarios, where decentralized organizations (e.g., hospitals, financial institutions) independently update overlapping knowledge, leading to redundant mediator knowledge vector (MKV) computations and privacy concerns. To address these challenges, we introduce Federated Locate-then-Edit Knowledge Editing (FLEKE), a novel task that enables multiple clients to collaboratively perform LEKE while preserving privacy and reducing computational overhead. To achieve this, we propose FedEdit, a two-stage framework that optimizes MKV selection and reuse. In the first stage, clients locally apply LEKE and upload the computed MKVs. In the second stage, rather than relying solely on server-based MKV sharing, FLEKE allows clients retrieve relevant MKVs based on cosine similarity, enabling knowledge re-edit and minimizing redundant computations. Experimental results on two benchmark datasets demonstrate that FedEdit retains over 96% of the performance of non-federated LEKE while significantly outperforming a FedAvg-based baseline by approximately twofold. Besides, we find that MEMIT performs more consistently than PMET in the FLEKE task with our FedEdit framework. Our code is available at https://github.com/zongkaiz/FLEKE.

知识编辑联邦学习大模型隐私保护

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