arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2405.16472cs.LG

多级联邦学习中的个性化加法建模

Personalized Additive Modeling for Multi-level Federated Learning

University of Technology Sydney(悉尼科技大学);The Hong Kong Polytechnic University(香港理工大学)

Shutong Chen, Guodong Long, Tianyi Zhou, Jie Ma, Jing Jiang, Chengqi Zhang

AI总结:

本文提出FeMAM,通过多级共享模型和加法组合构建个性化预测器,有效处理复杂非独立同分布数据,优于传统联邦学习基线。

摘要:

现代AI面临在通用性与用户特定个性化之间的平衡挑战。在联邦学习(FL)中,这一挑战因高度异质的客户端数据和复杂的非IID模式而加剧。现有FL方法受限于固定的异质性设置,难以应对复杂结构。本文从多级非IID视角出发,通过全局、子组和客户端特定组件捕捉客户端相似性,提出FeMAM,通过动态增长和剪枝模型,以低成本实现多级共享模型和个性化预测器,实验显示其在复杂非IID结构中表现优异。

英文摘要:

Contemporary AI faces the challenge of balancing generality with user-specific personalization. In federated learning (FL), this challenge is amplified by highly heterogeneous client data with complex non-IID patterns beyond standard IID assumptions. Many existing FL methods are designed for relatively restricted heterogeneity settings (e.g., a fixed number of clusters or a fixed form of personalization), limiting their robustness under complex structures. In this work, we study FL from a \emph{multi-level non-IID} perspective, where client similarity is captured by multiple granularities of shared knowledge: global, subgroup, and client-specific components. This view captures coarse-to-fine relationships while requiring less prior knowledge of task boundaries. Building on this insight, we propose \emph{Federated Multi-level Additive Modeling} (FeMAM), which learns multiple levels of shareable models and constructs personalized predictors via additive composition across levels. To move beyond a fixed structure, FeMAM allows models to grow and be pruned dynamically during training, adapting to diverse federated scenarios. Despite employing multiple models, FeMAM remains cost-friendly by unlocking only a small subset (one level) of models for training at a time. Extensive experiments show that FeMAM effectively approximates diverse complex non-IID structures and consistently outperforms representative clustered and personalized FL baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑