通过梯度匹配实现客户端数据蒸馏,一次完成通信节省大量开销
DistDD: Distributed Data Distillation Aggregation through Gradient Matching
- 客户端直接蒸馏数据,仅需一次通信即可生成全局数据集
- 在非独立同分布和错误标签数据下仍保持高效稳定
- 适合需要快速调参与神经架构搜索的联邦学习场景
本文提出 DistDD,一种新型联邦学习框架下的分布式数据蒸馏聚合方法,通过在客户端设备上直接蒸馏数据,大幅减少重复通信需求。与传统联邦学习需多次迭代模型更新不同,DistDD仅通过一次蒸馏过程即可提取全局蒸馏数据集,在保障联邦学习隐私性的同时显著降低通信成本。基于该蒸馏数据集,联邦学习开发者可实现即时参数调优与神经架构搜索(NAS),无需重复整个联邦学习流程。我们提供了 DistDD 算法的详细收敛性证明,强化其数学稳定性与实际应用可靠性。实验表明,DistDD 在非独立同分布(non-i.i.d.)及误标数据场景下依然有效且鲁棒,展现出对复杂现实数据挑战的独特应对能力。此外,我们在神经架构搜索(NAS)用例中评估了其应用效果,验证了其在通信节省方面的显著优势。
原文摘要 · Abstract (English)
In this paper, we introduce DistDD, a novel approach within the federated learning framework that reduces the need for repetitive communication by distilling data directly on clients' devices. Unlike traditional federated learning that requires iterative model updates across nodes, DistDD facilitates a one-time distillation process that extracts a global distilled dataset, maintaining the privacy standards of federated learning while significantly cutting down communication costs. By leveraging the DistDD's distilled dataset, the developers of the FL can achieve just-in-time parameter tuning and neural architecture search over FL without repeating the whole FL process multiple times. We provide a detailed convergence proof of the DistDD algorithm, reinforcing its mathematical stability and reliability for practical applications. Our experiments demonstrate the effectiveness and robustness of DistDD, particularly in non-i.i.d. and mislabeled data scenarios, showcasing its potential to handle complex real-world data challenges distinctively from conventional federated learning methods. We also evaluate DistDD's application in the use case and prove its effectiveness and communication-savings in the NAS use case.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。