← 最新论文
🤖 machine learning

Rethinking Federated Unlearning via the Lens of Memorization

本文提出联邦记忆剪枝(FedMemPrune),这是一种新颖的联邦遗忘方法,它利用一种新的分组记忆评估指标来区分并移除独特的记忆信息,同时保留重叠模式,从而在不损害模型效用的情况下实现有效的遗忘。

原作者: Jiaheng Wei, Yanjun Zhang, He Zhang, Leo Yu Zhang, Chao Chen, Kok-Leong Ong, Jun Zhang, Yang Xiang

发布于 2026-05-26
📖 1 分钟阅读☕ 轻松阅读

原作者: Jiaheng Wei, Yanjun Zhang, He Zhang, Leo Yu Zhang, Chao Chen, Kok-Leong Ong, Jun Zhang, Yang Xiang

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

以下是论文《通过记忆化视角重新思考联邦遗忘》的解读,将其拆解为简单概念和日常类比。

宏观图景:群聊中的“被遗忘权”

想象一群朋友(客户端)想要共同构建一本超级智能的学习指南(机器学习模型)。由于私人笔记过于私密,他们无法共享,因此只将各自的“顿悟时刻”和总结(模型更新)分享给一位中央导师(服务器)。这就是联邦学习

但如果其中一位朋友,我们称其为“亚历克斯”,决定退出群组,并要求学习指南遗忘亚历克斯贡献的一切,该怎么办?这就是联邦遗忘

法律(如 GDPR)规定亚历克斯拥有“被遗忘权”。最简单的做法是扔掉整本学习指南,让剩下的朋友从头开始。但这耗时极长且消耗大量能量。本文认为,当前的“快速修复”方法实际上正在犯错。

问题所在:因噎废食

作者认为,现有方法过于激进。它们试图抹除与亚历克斯相关的一切,甚至包括亚历克斯学到的那些实际上由其他人共享的内容。

类比:
想象这群朋友正在学习如何烤蛋糕。

  • 共享知识: 大家都同意需要面粉和鸡蛋。这是通用知识。
  • 亚历克斯的独特记忆: 亚历克斯有一个独特的、奇怪的敲鸡蛋技巧,没人使用过。这是“记忆化”。

如果亚历克斯离开,当前的方法会尝试抹除关于鸡蛋的所有知识,包括你需要鸡蛋这一事实。这会伤害剩下的朋友,因为他们仍然需要鸡蛋来烤蛋糕。模型对其他人来说,烤蛋糕的能力会变差。

本文提出:仅仅因为亚历克斯离开了,我们就应该抹除共享知识吗? 答案是否定的。我们应该只移除亚历克斯的独特技巧(记忆化),同时保留共享规则(重叠信息)。

解决方案:“记忆化剪枝”(FedMemPrune)

作者提出了一种名为FedMemPrune的新方法。把它想象成学习指南的智能编辑。

编辑不是擦除整页内容,而是寻找只有亚历克斯写下的特定句子,并删除这些句子。

工作原理(三个步骤):

  1. 寻找“冗余”笔记:
    系统查看剩余朋友的笔记。它会问:“学习指南的哪些部分没有被剩余朋友使用或更新?”

    • 类比: 如果剩余朋友都在写关于“面粉”的内容,但有一段只关于“亚历克斯奇怪的敲鸡蛋技巧”,那么这一段对群组来说是“冗余”的。那里很可能就是亚历克斯独特记忆所在的地方。
  2. 重置记忆:
    系统提取学习指南中那些特定的“冗余”部分并将其彻底清除(重新初始化)。它不仅仅是删除,而是将它们重置为空白状态,就像擦除白板一样。

  3. 微调:
    剩余的朋友快速一起审查指南,确保“面粉”和“鸡蛋”部分依然完美。因为共享知识从未被触碰,所以指南对其他人来说依然卓越。

为何更优

本文将该方法与其他方法进行了测试,发现了三大优势:

  1. 更好的“遗忘”: 它比其他方法更有效地移除了亚历克斯的独特技巧。
  2. 更好的“记忆”: 它保持了群组共享知识的完整性,因此学习指南不会让其他朋友的表现变差。
  3. 公平性: 它平等地对待剩余的朋友。某些方法会无意中伤害那些笔记与亚历克斯最相似的朋友,但此方法保护了她们。
  4. 速度: 它比从头开始(重新训练)快得多,节省了约 50% 的时间。

“分组记忆化”记分卡

作者还意识到,我们需要一种更好的方法来检查遗忘是否生效。不能只看最终成绩;你需要检查具体的问题。

他们创建了一种新的测试,称为分组记忆化评估

  • 他们提取亚历克斯被问到的问题,并根据其“独特性”或“记忆化”程度进行分类。
  • 他们检查新的学习指南是否在独特问题上失败(很好!),但在通用问题上依然通过(也很好!)。
  • 这证明了系统不仅仅是在猜测;它正在具体地遗忘正确的内容。

总结

简而言之,这篇论文指出:当你需要遗忘时,不要蛮干。

如果你想从群策项目中移除某人的贡献,不要烧毁整个项目。相反,识别出只有那个人知道的特定、独特部分,擦除这些部分,然后让团队其余成员去完善共享部分。这样能保持项目强大、公平且快速。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →