arXiv:2602.01186cs.LGcs.AI2026-02中稿 · ICLR被引 2

无需传输模型或数据,仅靠统计信息实现高效联邦学习。

The Gaussian-Head OFL Family: One-Shot Federated Learning from Client Global Statistics

  • 客户端仅上传类别统计量,服务器用高斯头构建分类器。
  • 在强非独立同分布下仍保持高准确率与鲁棒性。
  • 适合隐私敏感、资源受限的分布式场景。

传统联邦学习依赖多轮模型交换与聚合,通信开销大且存在隐私风险。一拍即合联邦学习(OFL)通过单轮通信降低开销,提升实用性。然而现有方法常依赖公开数据集、假设客户端模型同质,或需上传额外数据/模型信息。为此,我们提出高斯头一拍即合联邦学习(GH-OFL)系列方法,假设预训练嵌入服从类条件高斯分布。客户端仅传输每类计数及一阶、二阶矩等充分统计量;服务器通过三部分构建分类头:(i) 由接收统计量直接计算的闭式高斯头(如NB/LDA/QDA);(ii) FisherMix,一种在估计费雪子空间中生成合成样本并训练的余弦边界线性头;(iii) Proto-Hyper,轻量级低秩残差头,通过知识蒸馏在合成样本上优化高斯输出。实验表明,GH-OFL在强非独立同分布偏移下达到顶尖性能,且全程无需传输原始数据。

原文摘要 · Abstract (English)

Classical Federated Learning relies on a multi-round iterative process of model exchange and aggregation between server and clients, with high communication costs and privacy risks from repeated model transmissions. In contrast, one-shot federated learning (OFL) alleviates these limitations by reducing communication to a single round, thereby lowering overhead and enhancing practical deployability. Nevertheless, most existing one-shot approaches remain either impractical or constrained, for example, they often depend on the availability of a public dataset, assume homogeneous client models, or require uploading additional data or model information. To overcome these issues, we introduce the Gaussian-Head OFL (GH-OFL) family, a suite of one-shot federated methods that assume class-conditional Gaussianity of pretrained embeddings. Clients transmit only sufficient statistics (per-class counts and first/second-order moments) and the server builds heads via three components: (i) Closed-form Gaussian heads (NB/LDA/QDA) computed directly from the received statistics; (ii) FisherMix, a linear head with cosine margin trained on synthetic samples drawn in an estimated Fisher subspace; and (iii) Proto-Hyper, a lightweight low-rank residual head that refines Gaussian logits via knowledge distillation on those synthetic samples. In our experiments, GH-OFL methods deliver state-of-the-art robustness and accuracy under strong non-IID skew while remaining strictly data-free.

联邦学习一拍即合高斯头数据隐私

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