Efficient Distributed Swarm Learning for Edge Computing
Efficient Distributed Swarm Learning for Edge Computing
复制标题
DOI:
10.1109/icc45041.2023.10279508
复制
发表时间:
2023-05
期刊:
影响因子:
--
通讯作者:
Xin Fan;Yue Wang;Yan Huo;Zhi Tian
中科院分区:
文献类型:
--
作者:
Xin Fan;Yue Wang;Yan Huo;Zhi Tian
Federated learning (FL) methods face major challenges including communication bottleneck, data heterogeneity and security concerns in edge IoT scenarios. In this paper, inspired by the success of biological intelligence (BI) of gregarious organisms, we propose a novel edge learning approach for swarm IoT, called communication-efficient and Byzantine-robust distributed swarm learning (CB-DSL), through a holistic integration of AI-enabled stochastic gradient descent and BI-enabled particle swarm optimization. To deal with non-independent and identically distributed (non-i.i.d.) data issues and Byzantine attacks, a very small amount of global data samples are introduced in CB-DSL and shared among IoT workers, which not only alleviates the local data heterogeneity effectively but also enables to fully utilize the exploration-exploitation mechanism of swarm intelligence. Further, we provide convergence analysis to theoretically demonstrate that the proposed CB-DSL is superior to the standard FL with better convergence behavior. In addition, to measure the effectiveness of the introduction of the globally shared dataset, we also evaluate the model divergence by deriving its upper bound. Numerical results verify that the proposed CB-DSL outperforms the existing benchmarks in terms of faster convergence speed, higher convergent accuracy, lower communication cost, and better robustness against non-i.i.d. data and Byzantine attacks11Our code can be found at:https://github.com/fuanxiyin/CB-DSL.git..