中文

基于扩散模型的联邦知识图谱遗忘学习

机器学习 2024-03-14 v1 人工智能

摘要

联邦学习(FL)通过实现模型共享与协作,同时保护数据隐私,促进了人工智能技术的发展与应用。知识图谱(KG)嵌入表示通过将实体和关系映射到向量空间,为知识推理和应用提供了基础。联邦知识图谱嵌入能够在保护本地数据隐私的同时,利用来自多样化客户端来源的知识。然而,由于隐私保护以及适应动态数据变化等需求,引发了对机器遗忘学习(MU)的研究。但是,在遗忘特定数据对模型影响的同时,保持知识图谱嵌入模型的性能是具有挑战性的。在本文中,我们提出了 FedDM,一个专为联邦知识图谱中的机器遗忘学习定制的新型框架。利用扩散模型,我们生成噪声数据,以合理地减轻特定知识对联邦学习模型的影响,同时保持模型在其余数据上的整体性能。我们在基准数据集上进行了实验评估,以检验所提出模型的有效性。大量实验表明,FedDM 在知识遗忘方面取得了令人鼓舞的结果。

关键词

引用

@article{arxiv.2403.08554,
  title  = {Federated Knowledge Graph Unlearning via Diffusion Model},
  author = {Bingchen Liu and Yuanyuan Fang},
  journal= {arXiv preprint arXiv:2403.08554},
  year   = {2024}
}