arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.08147cs.LGcs.DC
收藏

AutoFL:实现异构感知的节能联邦学习

AutoFL: Enabling Heterogeneity-Aware Energy Efficient Federated Learning

Soongsil University(崇实大学);Arizona State University(亚利桑那州立大学);Facebook AI Research(脸书人工智能研究院)

Young Geun Kim, Carole-Jean Wu

摘要:

联邦学习使边缘的分散式移动设备集群能够协作训练共享的机器学习模型,同时将所有原始训练样本保留在设备上。这种分散的训练方法被证明是一种切实可行的解决方案,以减轻隐私泄露的风险。然而,由于非IID训练数据分布、广泛的系统异构性和现场随机变化的运行时效应,在边缘实现高效的FL部署具有挑战性。本文通过考虑边缘执行的随机性,联合优化了最先进FL用例的收敛时间和能效。我们提出AutoFL通过量身定制设计一个强化学习算法,学习和确定哪些K参与设备和每个设备的执行目标,每个FL模型聚合轮随机运行时方差,系统和数据的异构性的存在。通过明智地考虑FL边缘部署的独特特性,AutoFL在K个参与者的集群上分别为本地客户端和全局客户端实现了3.6倍的模型收敛时间和4.7倍和5.2倍的能效。

英文摘要:

Federated learning enables a cluster of decentralized mobile devices at the edge to collaboratively train a shared machine learning model, while keeping all the raw training samples on device. This decentralized training approach is demonstrated as a practical solution to mitigate the risk of privacy leakage. However, enabling efficient FL deployment at the edge is challenging because of non-IID training data distribution, wide system heterogeneity and stochastic-varying runtime effects in the field. This paper jointly optimizes time-to-convergence and energy efficiency of state-of-the-art FL use cases by taking into account the stochastic nature of edge execution. We propose AutoFL by tailor-designing a reinforcement learning algorithm that learns and determines which K participant devices and per-device execution targets for each FL model aggregation round in the presence of stochastic runtime variance, system and data heterogeneity. By considering the unique characteristics of FL edge deployment judiciously, AutoFL achieves 3.6 times faster model convergence time and 4.7 and 5.2 times higher energy efficiency for local clients and globally over the cluster of K participants, respectively.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑