提出管道循环调度框架,让低能设备在充电时也能持续参与联邦学习。
Computation-aware Energy-harvesting Federated Learning with Pipelined Cyclic Scheduling
- 将客户端分组形成流水线,组间模型传递实现训练与充电重叠。
- 在非独立同分布数据下,能耗降低超基准方法,极端标签偏斜时仍保持准确。
- 适合电池能量波动大的边缘设备,尤其适用于能源收集场景。
联邦学习(FL)是分布式学习的强大范式,但模型复杂度提升导致客户端本地训练计算能耗显著增加。这一问题在能量收集型联邦学习(EHFL)系统中尤为关键,因设备参与受限于能量波动。为此,我们提出 PipeCycle 框架,将客户端组织为流水线循环分组:当一组完成组内聚合后,其模型直接作为新组本地训练的参考,实现多组并行、重叠充电与训练过程。我们在支持跨多个时间片的本地训练的现实能量消耗模型下,提供收敛性分析,证明管道循环结构引入有限时域过时性约束,避免了异步联邦学习中常见的指数级增长因素。数值实验在独立同分布(IID)与非独立同分布(non-IID)数据及不同电池充电概率下均表明,PipeCycle 在达到目标精度时,累计能耗显著低于现有基准方法,尤其在严重标签偏斜情况下,其他循环方案性能退化至近随机水平,而本方法仍保持有效。
原文摘要 · Abstract (English)
Federated learning (FL) is a powerful paradigm for distributed learning, but increasing model complexity leads to significant energy consumption from client-side computations for local training. This challenge is critical in energy-harvesting FL (EHFL) systems, where the participation availability of each device fluctuates because of limited energy. To address this, we propose PipeCycle, a battery-aware distributed learning framework that organizes clients into pipelined cyclic groups. When a group completes its intra-group aggregation, its aggregated model is relayed directly to a newly formed group as a reference for local training, allowing multiple groups to coexist in the pipeline while overlapping client recharging periods with active training in other pipeline stages. We provide a convergence analysis of PipeCycle under a realistic energy consumption model in which local training spans multiple time slots, and show that the cyclic structure of the pipeline imposes a finite-horizon staleness bound that avoids the exponential factors typical of asynchronous FL analyses. Numerical experiments across both IID and non-IID data and various battery charging probabilities show that PipeCycle reaches a target accuracy with substantially lower cumulative energy than existing FL baselines, particularly under severe label skew where competing cyclic schemes collapse to near-chance accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。