Personalized Additive Modeling for Multi-level Federated Learning
本文提出了联邦多级加性建模(FeMAM),这是一个动态且高效的框架,通过构建由全局、子组和客户端特定模型组成的加性组合,并在训练过程中进行自适应增长与剪枝,来解决联邦学习中复杂的非独立同分布(non-IID)数据异构问题。
原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明
想象一下,你正在尝试教 50 个不同的人如何烹饪。
在标准课堂(传统 AI)中,老师给每个人提供完全相同的食谱。如果学生们的口味或食材非常不同,这个食谱对任何人都不会很奏效。
在“联邦学习”(Federated Learning)的情景中,学生们都在自己的厨房里。由于隐私规则,他们不能分享自己的秘密食材(数据)。他们只能将他们的“想法”传回给老师,以更新主食谱。
问题在于,这 50 名学生各不相同。有些人是意大利人,有些是素食主义者,有些是专业厨师,而有些则是初学者。单一的食谱太简单了,而给每个人一个完全独特的食谱又会过于混乱。
问题所在:“一刀切”的困境
现有的方法试图通过以下两种方式解决这个问题:
- 分组: 将所有人分为“意大利人组”和“素食者组”。但如果一个意大利学生同时也是个初学者呢?这些分组过于僵化了。
- 个性化每一个人: 给每个人一个独特的食谱。但这忽略了他们可能拥有共同的知识(比如如何切洋葱)。
论文指出,现实生活更加复杂。相似性存在于多个层面:
- 第 1 层(全局): 每个人都知道如何烧开水。
- 第 2 层(子群): “意大利人”组的每个人都知道如何做意面酱。
- 第 3 层(个人): 你 特别知道如何添加一种秘密的家族香料。
目前大多数 AI 系统一次只能处理这两个层级中的一个。当结构变得杂乱且多层化时,它们就会陷入困境。
解决方案:FeMAM(“乐高塔”法)
作者提出了一种名为 FeMAM(联邦多层加性建模)的新方法。
把 FeMAM 想象成在搭建一座 乐高塔,每一层都增加了更多的细节:
- 底层: 每个人都从同一个基础模块开始(全局模型)。这涵盖了“烧开水”的部分。
- 中间层: 随着学生的训练,系统会自动识别出谁需要额外的帮助。它为意大利组增加了一个“意面”模块,并为其他群体增加了一个“素食”模块。这些模块是在子群之间共享的。
- 顶层: 最后,它为你添加了一个微小的、定制的“秘密香料”模块。
神奇之处在于:
FeMAM 并不是为每个学生从头开始建造一座全新的巨大塔楼,而是堆叠这些模块。
- 你拿取全局模块。
- 在上面加上组别模块。
- 再在上面加上你的个人模块。
- 结果: 你的最终塔楼是所有这些部分的总和。
如何保持高效(“生长与修剪”策略)
你可能会想:“等等,如果每个人都有 5 个不同的模块,那往返传输的数据量不是太多了吗?”
论文说并非如此,这要归功于两条聪明的规则:
- 循序渐进(Boosting/提升法): 系统不会一次性构建整个塔楼。它先构建基础层,等待其稳定,然后 再添加下一层。它只将新的这一层传回给老师。这降低了通信成本,就像寄一张明信片而不是寄一本书一样。
- “停止”按钮(Pruning/修剪法): 并不是每个人都需要 5 层。
- 如果一个学生只是个简单的初学者,系统会意识到:“嘿,基础层对你来说已经足够了!”并停止添加更多模块。
- 如果一个学生是复杂的专家,系统会不断添加层级直到达到完美。
- 这意味着系统会适应学生。它不会把复杂的塔楼强加给简单的用户,从而节省了他们的时间和存储空间。
他们的发现
研究人员在图像数据集(如识别猫和狗)和大型语言模型(如 AI 聊天机器人)上测试了该方法。
- 多功能性: 无论学生是按简单的类别分组,还是存在杂乱复杂的差异,FeMAM 的表现都优于那些试图将所有人强制固定在分组中或给予每个人完全独特模型的方案。
- 智能适应: 在实验中,系统自动决定某些用户需要 1 层,某些需要 2 层,而某些需要 5 层。它不是靠猜测,而是学习了每个人的最佳需求。
- 成本: 尽管它使用了多“层”模型,但运行成本并不比标准方法高多少,因为它一次只通信一层。
核心结论
FeMAM 就像一位聪明的、灵活的老师,她意识到学习是分层进行的。她从大家共有的知识开始,加入特定群体共有的知识,最后再加上让你与众不同的特质。她循序渐进地构建这些知识,只添加必要的层级,这使得它非常适合这样一个每个人都各不相同但又相互联系的世界。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。