arXiv:2607.28338cs.LGcs.CR2026-07

提出可加密的聚类联邦学习方法,解决隐私与效率的矛盾。

Encryption-Compatible Clustered Federated Learning via Distributed Expectation-Maximization over Metadata

论文配图:Encryption-Compatible Clustered Federated Learning via Distributed Expectation-Maximization over Metadata
图 1 · 摘自论文原文
  • 用分布式期望最大化重构元数据聚类,仅用加法更新服务器
  • 在多个数据集上提升模型效果,通信与计算效率不变
  • 适合需要加密保护的工业级联邦学习场景

聚类联邦学习(CFL)通过将数据分布相似的客户端分组,缓解联邦学习中的数据异构问题。现有方法在隐私保护、通信开销和计算效率之间存在权衡,称为CFL三难困境。主流方法利用元数据(即低维数据集表示)实现高效聚类,但与标准隐私保护机制不兼容。为此,本文提出FLAMECHE,将基于元数据的CFL重构成分布式期望最大化(EM)过程,仅允许服务器进行加法操作,从而保持效率并支持实际安全联邦学习方案。在多种数据集和异构场景下的大量实验表明,FLAMECHE提升了客户端模型性能,实现了加密兼容的元数据聚类,在CFL三难困境中取得更优定位。

原文摘要 · Abstract (English)

Clustered Federated Learning (CFL) addresses data heterogeneity in federated settings by grouping clients with similar data distributions to enable effective training. Existing methods face a trade-off between privacy preservation, communication cost, and computational efficiency. We formalize this as the CFL trilemma, according to which improving two of these dimensions comes at the expense of the third. A prominent paradigm relies on metadata (i.e., low-dimensional representations of client datasets shared with the server) to enable communication- and computation-efficient clustering. However, such approaches are not compatible with standard FL privacy-preserving mechanisms. To address this limitation, we propose FLAMECHE, which reformulates metadata-based CFL as a distributed Expectation-Maximization (EM) procedure, restricting server updates to additive operations while preserving efficiency. This design enables compatibility with practical secure FL schemes. We conducted extensive experiments on multiple datasets under various heterogeneous scenarios. Results show that FLAMECHE improves the effectiveness of client models. It enables encryption-compatible metadata-based clustering, enhancing its positioning within the CFL trilemma.

联邦学习加密计算聚类元数据

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