Federated Learning Incentive Mechanism Setting in UAV-Assisted Space–Terrestrial Integration Networks

Chun Zhu,Mengqi Sui,Haitao Zhao,Keqi Chen,Tianyu Zhang,Chongyu Bao

DOI: https://doi.org/10.3390/electronics13061129

IF: 2.9

2024-03-20

Electronics

Abstract:The UAV-assisted space–terrestrial integrated network provides extensive coverage and high flexibility in communication services. UAVs and ground terminals collaborate to train models and provide services. In order to protect data privacy, federated learning is widely used. However, the participation of UAVs and ground terminals is not gratuitous, and reasonable incentives for federated learning need to be set up to encourage their participation. To address the above issues, this paper proposes a federated reliable incentive mechanism based on hierarchical reinforcement learning. The mechanism allocates inter-round incentives at the upper level to ensure the maximisation of the server's utility, and performs inter-client incentive allocation at the lower level to ensure the minimisation of each round's latency. The reasonable incentive allocation enables the central server to achieve higher model training accuracy under the limited incentive budget, which reduces the cost of model training. At the same time, an attack detection mechanism is implemented to identify malicious clients participating in federated learning, preventing their involvement in aggregation and revoking their incentives. This better ensures the security of model training. Finally, we conducted experiments on Fmnist, and the results indicate that this method effectively improves the accuracy and security of model training.

engineering, electrical & electronic,computer science, information systems,physics, applied

What problem does this paper attempt to address?

The paper attempts to address the issue of how to set up a reasonable federated learning incentive mechanism in a drone-assisted air-ground integrated network to encourage drones and ground terminals to participate in federated learning while ensuring the accuracy and security of model training.

Specifically, the paper focuses on the following aspects:

1. **Data Privacy Protection**: Federated learning allows drones and ground terminals to retain data locally and only upload training parameters to the central server, thereby protecting data privacy.
2. **Resource Consumption and Incentives**: Drones and ground terminals need to consume their own computing and communication resources when participating in federated learning. Therefore, a reasonable incentive mechanism needs to be designed to compensate for these costs and encourage their participation.
3. **Model Training Performance**: Improving the accuracy of model training and reducing training delay under a limited incentive budget.
4. **Malicious Client Detection**: Identifying and excluding malicious clients participating in federated learning to prevent them from affecting the security of model training.

To address the above challenges, the paper proposes a federated learning incentive mechanism based on hierarchical reinforcement learning. This mechanism ensures the efficiency of model training and the security of the system through a multi-level incentive allocation strategy.