arXiv:2409.11442cs.LGcs.AI2024-09被引 3

用灰狼优化算法选参与设备,让无线联邦学习更省电、更快收敛。

A Green Multi-Attribute Client Selection for Over-The-Air Federated Learning: A Grey-Wolf-Optimizer Approach

  • 用灰狼优化器综合考虑精度、能耗、延迟等多指标选设备
  • 模型损失降低,收敛速度加快,能源效率提升显著
  • 适合资源受限场景,兼顾公平性与可靠性

联邦学习(FL)因其无需集中敏感数据即可训练模型而受到广泛关注。尽管该方法在隐私保护和降低通信开销方面具有优势,但在异构环境或资源受限场景中仍面临部署复杂性和互操作性挑战。为应对这些问题,无线联邦学习(OTA-FL)通过无线广播模型更新,避免了设备间直接连接或依赖中心服务器。然而,OTA-FL也带来了更高的能耗和网络延迟。本文提出一种基于灰狼优化器(GWO)的多属性客户端选择框架,通过动态控制每轮参与设备数量,在考虑精度、能耗、延迟、可靠性和公平性约束的前提下,优化整个OTA-FL流程。实验表明,所提方法在模型损失最小化、收敛时间缩短和能效提升方面均优于现有基准。具体而言,模型损失下降明显,收敛速度加快,同时保持高公平性与可靠性水平。

原文摘要 · Abstract (English)

Federated Learning (FL) has gained attention across various industries for its capability to train machine learning models without centralizing sensitive data. While this approach offers significant benefits such as privacy preservation and decreased communication overhead, it presents several challenges, including deployment complexity and interoperability issues, particularly in heterogeneous scenarios or resource-constrained environments. Over-the-air (OTA) FL was introduced to tackle these challenges by disseminating model updates without necessitating direct device-to-device connections or centralized servers. However, OTA-FL brought forth limitations associated with heightened energy consumption and network latency. In this paper, we propose a multi-attribute client selection framework employing the grey wolf optimizer (GWO) to strategically control the number of participants in each round and optimize the OTA-FL process while considering accuracy, energy, delay, reliability, and fairness constraints of participating devices. We evaluate the performance of our multi-attribute client selection approach in terms of model loss minimization, convergence time reduction, and energy efficiency. In our experimental evaluation, we assessed and compared the performance of our approach against the existing state-of-the-art methods. Our results demonstrate that the proposed GWO-based client selection outperforms these baselines across various metrics. Specifically, our approach achieves a notable reduction in model loss, accelerates convergence time, and enhances energy efficiency while maintaining high fairness and reliability indicators.

联邦学习无线通信优化算法能效

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