Attention-aided federated learning for dependency-aware collaborative task allocation in edge-assisted smart grid scenarios
Wang, Chenyang; Jia, Bosen; Yu, Hao; Chen, Liandong; Cheng, Kai; Wang, Xiaofei (2022-09-22)
C. Wang, B. Jia, H. Yu, L. Chen, K. Cheng and X. Wang, "Attention-aided Federated Learning for Dependency-Aware Collaborative Task Allocation in Edge-Assisted Smart Grid Scenarios," 2022 IEEE/CIC International Conference on Communications in China (ICCC), 2022, pp. 856-861, doi: 10.1109/ICCC55456.2022.9880777.
© 2022 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
With the significant improvement of the intelligent capabilities of smart devices accompanied by the increasingly high requirements. Edge computing is regarded as an effective solution to achieve rapid response by deploying applications and tasks close to users. However, many studies only consider complete offloading, or offload tasks to edge servers in any proportion when designing the allocation strategies, ignoring the dependencies between subtasks. To deal with the dynamic environment, some learning-based task allocation methods generally adopt a centralized training way, which leads to the excessive network transmission resource consumption, especially in the smart grid scenario. To tackle the aforementioned challenges, we investigate the collaborative task allocation (CTA) problem by jointly considering the difference between the benefit of the tasks execution under a certain allocation strategy and when all tasks are executed locally. In this paper, the objective is to maximize the system gain, and we propose an attention-aided federated learning algorithm to deal with the CTA problem, named AteFL, by learning a shared model and extracting the system context for better representing the network information. The simulation results also show the superiority of the proposed AteFL algorithm.
- Avoin saatavuus