FedFiOT 用 Fisher 加权最优传输改进单轮联邦学习模型合并

论文AI 评分 62/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

arXiv 论文提出 FedFiOT,面向单轮与少轮联邦学习。这类设置通信轮次少,但客户端模型容易漂移,学到相似特征的神经元在不同客户端处于不同位置,加上本地数据异质性使参数重要性不一,直接平均会把不匹配的神经元合并、并对重要性不同的参数等权处理。FedFiOT 用 Fisher 信息衡量参数重要性,引导神经元对齐与模型聚合,通过交替最小化一个共享的 Fisher 加权目标,联合优化神经元对应关系与全局模型参数。理论部分给出逐层合并目标单调下降的结论,以及刻画客户端异质性与 Fisher 非均匀性影响的平稳性界;多个数据集实验显示,在单轮与少轮设置下均优于对比基线。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

少轮联邦学习的两个痛点

单轮和少轮联邦学习把全局模型的构建压缩到一次或少数几次通信,通信开销因此下降,但效果通常不如常规多轮方案。论文把原因归到两点:缺少频繁同步时,各客户端模型会彼此偏离,学到相似特征的神经元在不同客户端落在不同位置上;本地数据分布不一致,又让对应参数对各自预测的重要程度出现差别。若直接做平均,等于把位置对不上的神经元强行拼在一起,同时对重要性不同的参数一视同仁地赋权,合并结果自然偏离。这一分析构成了 FedFiOT 的设计出发点。

Fisher 信息引导的对齐与聚合

FedFiOT 是一个 Fisher 加权最优传输框架。它借助 Fisher 信息衡量每个参数对模型预测的重要程度,再把这种重要性带进神经元对齐与模型聚合两个环节,而不是只按参数数值做匹配。具体做法是对一个共享的 Fisher 加权目标做交替最小化:一边更新神经元之间的对应关系,一边更新全局模型参数,两者在同一目标下联合推进。这样,位置错位的神经元先被对齐,聚合时也按重要性区别对待,而不是等权相加。

理论保证与实验表现

论文给出两方面理论结果:其优化过程会让逐层合并目标单调下降;同时建立一个平稳性界,用来刻画客户端异质性和 Fisher 权重非均匀分布对最终结果的影响程度。实验覆盖多个数据集,在单轮与少轮两种联邦学习设置下,FedFiOT 相对参与比较的基线都取得了一致的提升。材料未披露具体数据集名称、参数量与提升幅度,也未说明代码是否公开。

为什么值得看

少轮联邦学习省通信但精度掉,这篇给出对齐加加权聚合的具体做法与理论界,值得关注联邦学习合并方向的人一看。

量化对齐论文数据集

信源1 家

  1. [1]arXiv cs.AI一手信源FedFiOT: Fisher-Weighted Optimal Transport for Model Merging in One-Shot and Few-Round Federated Learning

关键事实

  • FedFiOT 针对单轮与少轮联邦学习,目标是减少通信轮次带来的性能损失[1]

  • 方法用 Fisher 信息量化参数重要性,并据此引导神经元对齐与模型聚合[1]

  • FedFiOT 通过交替最小化共享的 Fisher 加权目标,联合优化神经元对应关系与全局模型参数[1]

  • 论文证明该优化过程使逐层合并目标单调下降,并给出刻画客户端异质性与 Fisher 非均匀性影响的平稳性界[1]

  • 多个数据集上的实验显示,FedFiOT 在单轮与少轮设置下均稳定优于对比基线[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。