FedFiOT 用 Fisher 加权最优传输改进单轮联邦学习模型合并
本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。
全文梳理
AI 摘要依据下方信源原文自动整理,非原文转载少轮联邦学习的两个痛点
单轮和少轮联邦学习把全局模型的构建压缩到一次或少数几次通信,通信开销因此下降,但效果通常不如常规多轮方案。论文把原因归到两点:缺少频繁同步时,各客户端模型会彼此偏离,学到相似特征的神经元在不同客户端落在不同位置上;本地数据分布不一致,又让对应参数对各自预测的重要程度出现差别。若直接做平均,等于把位置对不上的神经元强行拼在一起,同时对重要性不同的参数一视同仁地赋权,合并结果自然偏离。这一分析构成了 FedFiOT 的设计出发点。
Fisher 信息引导的对齐与聚合
FedFiOT 是一个 Fisher 加权最优传输框架。它借助 Fisher 信息衡量每个参数对模型预测的重要程度,再把这种重要性带进神经元对齐与模型聚合两个环节,而不是只按参数数值做匹配。具体做法是对一个共享的 Fisher 加权目标做交替最小化:一边更新神经元之间的对应关系,一边更新全局模型参数,两者在同一目标下联合推进。这样,位置错位的神经元先被对齐,聚合时也按重要性区别对待,而不是等权相加。
理论保证与实验表现
论文给出两方面理论结果:其优化过程会让逐层合并目标单调下降;同时建立一个平稳性界,用来刻画客户端异质性和 Fisher 权重非均匀分布对最终结果的影响程度。实验覆盖多个数据集,在单轮与少轮两种联邦学习设置下,FedFiOT 相对参与比较的基线都取得了一致的提升。材料未披露具体数据集名称、参数量与提升幅度,也未说明代码是否公开。
为什么值得看
少轮联邦学习省通信但精度掉,这篇给出对齐加加权聚合的具体做法与理论界,值得关注联邦学习合并方向的人一看。
信源1 家
关键事实
相关 · 论文
本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。
