arXiv:2605.11815cs.LG2026-05中稿 · the 2nd Internatio…

Fed-BAC通过双层强化学习优化联邦聚类与选参,提升异构数据下的模型性能。

Fed-BAC: Federated Bandit-Guided Additive Clustering in Hierarchical Federated Learning

  • 用上下文与汤普森采样双层强化学习动态分配服务器和选高贡献客户端。
  • 在α=0.1时较基线最高提升35.5个百分点,仅需80%客户端参与,提速4.8倍。
  • 适合大规模异构联邦学习场景,尤其数据分布差异大的系统部署。

分层联邦学习(HFL)利用边缘服务器进行局部聚合,但现有方法缺乏在数据异构下联合优化聚类分配与客户端选择的机制。本文提出Fed-BAC,将加性聚类个性化与两层强化学习框架结合:云端使用上下文贝叶斯多臂老虎机学习服务器到集群的分配,每个边缘服务器采用汤普森采样识别高贡献客户端。加性分解使全局网络共享知识,而各聚类专用网络捕捉分布差异。在三种分类基准(CIFAR-10、SVHN、Fashion-MNIST)上,当Dirichlet非独立同分布参数α=0.5(中度)和α=0.1(严重)时,相比HierFAVG提升最高达+35.5个百分点,比IFCA提升+8.4个百分点;仅需80%客户端参与,收敛速度提升1.5至4.8倍,且改善跨服务器公平性。该效果在CIFAR-10上5倍规模部署中再次验证。异构程度越高,性能优势越显著,说明加性聚类个性化在数据分布差异增大时愈发重要。

原文摘要 · Abstract (English)

Hierarchical federated learning (HFL) leverages edge servers for partial aggregation in edge computing. Yet existing FL methods lack mechanisms for jointly optimizing cluster assignment and client selection under data heterogeneity. This paper proposes Fed-BAC, which integrates additive cluster personalization with a two-level bandit framework: contextual bandits at the cloud learn server-to-cluster assignments, while Thompson Sampling at each edge server identifies high-contributing clients. The additive decomposition enables the sharing of knowledge between groups through a globally aggregated network, while cluster-specific networks capture distribution variations. Across three classification benchmarks (CIFAR-10, SVHN, Fashion-MNIST) under moderate ($α= 0.5$) and severe ($α= 0.1$) Dirichlet non-IID partitioning, Fed-BAC achieves distributed accuracy gains of up to +35.5pp over HierFAVG and +8.4pp over IFCA, while requiring only 80% client participation, converging 1.5 to 4.8$\times$ faster depending on dataset and accuracy target, and improving cross-server fairness. These gains are further validated at 5$\times$ deployment scale on CIFAR-10. The advantage of Fed-BAC increases with heterogeneity severity, confirming that additive cluster personalization becomes increasingly valuable as data distributions diverge.

联邦学习聚类优化强化学习异构数据

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。