arXiv:2503.05803cs.LGcs.AI2025-03被引 1

用损失值替代模型权重共享,降低隐私风险和传输开销。

Federated Learning Framework via Distributed Mutual Learning

  • 客户端定期分享对公共测试集的损失预测,而非模型参数。
  • 通过融合本地损失与跨客户端损失的KL散度优化模型。
  • 在口罩检测任务中精度更高,且更安全、泛化能力更强。

联邦学习通常依赖于共享完整或部分模型权重,这会增加网络带宽负担并引发隐私风险。本文提出一种基于损失的替代方案,采用分布式互学习机制。客户端不传输模型参数,而是定期共享其在公开测试集上的损失预测结果。每个客户端通过结合自身本地损失与其它客户端损失的平均Kullback-Leibler散度来更新模型。该协作方式有效降低通信开销并增强数据隐私保护。在人脸口罩检测任务上的实验表明,该方法优于传统权重共享基线,在未见数据上达到更高准确率,具备更强泛化能力和隐私优势。

原文摘要 · Abstract (English)

Federated Learning often relies on sharing full or partial model weights, which can burden network bandwidth and raise privacy risks. We present a loss-based alternative using distributed mutual learning. Instead of transmitting weights, clients periodically share their loss predictions on a public test set. Each client then refines its model by combining its local loss with the average Kullback-Leibler divergence over losses from other clients. This collaborative approach both reduces transmission overhead and preserves data privacy. Experiments on a face mask detection task demonstrate that our method outperforms weight-sharing baselines, achieving higher accuracy on unseen data while providing stronger generalization and privacy benefits.

联邦学习互学习隐私保护通信优化

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