arXiv:2412.03950cs.LGcs.DC2024-12被引 1

平衡移动边缘物联网中设备能耗,提升模型精度与系统稳定性。

BEFL: Balancing Energy Consumption in Federated Learning for Mobile Edge IoT

  • 联合优化模型精度、总能耗和设备间能耗差异。
  • 能耗方差降低72.7%,总能耗减少28.2%,精度提升1.6%。
  • 适合资源受限的移动端边缘物联网场景使用。

联邦学习(FL)是一种保护隐私的分布式学习范式,旨在构建高精度全局模型。在移动边缘物联网(MEIoT)中,训练与通信过程会显著消耗设备有限的电池资源。现有研究多聚焦于降低总体能耗,但可能引发能耗不均,导致能量敏感设备提前离线。为此,我们提出BEFL,一个联合优化框架,兼顾提升全局模型精度、最小化总能耗、降低设备间能耗差异三项目标。首先,考虑MEIoT的通信约束与设备异构性,采用序列最小二乘规划(SLSQP)算法合理分配通信资源;在此基础上,引入一种结合聚类划分与效用驱动的启发式客户端选择算法,缓解整体能耗及能耗差异。此外,将该算法作为离线模仿学习模板用于预训练,同时在线采用基于排序的强化学习方法进一步提升训练效率。实验表明,相比现有方法,BEFL使全局模型精度提升1.6%,能耗方差降低72.7%,总能耗减少28.2%。相关代码见https://github.com/juzehao/BEFL。

原文摘要 · Abstract (English)

Federated Learning (FL) is a privacy-preserving distributed learning paradigm designed to build a highly accurate global model. In Mobile Edge IoT (MEIoT), the training and communication processes can significantly deplete the limited battery resources of devices. Existing research primarily focuses on reducing overall energy consumption, but this may inadvertently create energy consumption imbalances, leading to the premature dropout of energy-sensitive devices.To address these challenges, we propose BEFL, a joint optimization framework aimed at balancing three objectives: enhancing global model accuracy, minimizing total energy consumption, and reducing energy usage disparities among devices. First, taking into account the communication constraints of MEIoT and the heterogeneity of devices, we employed the Sequential Least Squares Programming (SLSQP) algorithm for the rational allocation of communication resources. Based on this, we introduce a heuristic client selection algorithm that combines cluster partitioning with utility-driven approaches to alleviate both the total energy consumption of all devices and the discrepancies in energy usage.Furthermore, we utilize the proposed heuristic client selection algorithm as a template for offline imitation learning during pre-training, while adopting a ranking-based reinforcement learning approach online to further boost training efficiency. Our experiments reveal that BEFL improves global model accuracy by 1.6\%, reduces energy consumption variance by 72.7\%, and lowers total energy consumption by 28.2\% compared to existing methods. The relevant code can be found at \href{URL}{https://github.com/juzehao/BEFL}.

联邦学习能耗平衡边缘计算设备异构

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