A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency Communication
As the frequency of natural disasters increases, the study of emergency communication becomes increasingly important. The use of federated learning (FL) in this scenario can facilitate communication collaboration between devices while protecting privacy, greatly improving system performance. Conside...
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2024-02-01
|
| Series: | Sensors |
| Subjects: | |
| Online Access: | https://www.mdpi.com/1424-8220/24/5/1599 |
| _version_ | 1797263871790022656 |
|---|---|
| author | Chun Zhu Ying Shi Haitao Zhao Keqi Chen Tianyu Zhang Chongyu Bao |
| author_facet | Chun Zhu Ying Shi Haitao Zhao Keqi Chen Tianyu Zhang Chongyu Bao |
| author_sort | Chun Zhu |
| collection | DOAJ |
| description | As the frequency of natural disasters increases, the study of emergency communication becomes increasingly important. The use of federated learning (FL) in this scenario can facilitate communication collaboration between devices while protecting privacy, greatly improving system performance. Considering the complex geographic environment, the flexible mobility and large communication radius of unmanned aerial vehicles (UAVs) make them ideal auxiliary devices for wireless communication. Using the UAV as a mobile base station can better provide stable communication signals. However, the number of ground-based IoT terminals is large and closely distributed, so if all of them transmit data to the UAV, the UAV will not be able to take on all of the computation and communication tasks because of its limited energy. In addition, there is competition for spectrum resources among many terrestrial devices, and all devices transmitting data will bring about an extreme shortage of resources, which will lead to the degradation of model performance. This will bring indelible damage to the rescue of the disaster area and greatly threaten the life safety of the vulnerable and injured. Therefore, we use user scheduling to select some terrestrial devices to participate in the FL process. In order to avoid the resource waste generated by the terrestrial device resource prediction, we use the multi-armed bandit (MAB) algorithm for equipment evaluation. Considering the fairness issue of selection, we try to replace the single criterion with multiple criteria, using model freshness and energy consumption weighting as reward functions. The state of the art of our approach is demonstrated by simulations on the datasets. |
| first_indexed | 2024-04-25T00:19:54Z |
| format | Article |
| id | doaj.art-49bb97d92fe54587b7985bc552dfe579 |
| institution | Directory Open Access Journal |
| issn | 1424-8220 |
| language | English |
| last_indexed | 2024-04-25T00:19:54Z |
| publishDate | 2024-02-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Sensors |
| spelling | doaj.art-49bb97d92fe54587b7985bc552dfe5792024-03-12T16:55:20ZengMDPI AGSensors1424-82202024-02-01245159910.3390/s24051599A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency CommunicationChun Zhu0Ying Shi1Haitao Zhao2Keqi Chen3Tianyu Zhang4Chongyu Bao5College of Science, Nanjing University of Posts and Telecommunications, Nanjing 210003, ChinaSchool of Communication and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing 210003, ChinaCollege of Internet of Things, Nanjing University of Posts and Telecommunications, Nanjing 210003, ChinaSchool of Communication and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing 210003, ChinaSchool of Communication and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing 210003, ChinaPortland Institute, Nanjing University of Posts and Telecommunications, Nanjing 210003, ChinaAs the frequency of natural disasters increases, the study of emergency communication becomes increasingly important. The use of federated learning (FL) in this scenario can facilitate communication collaboration between devices while protecting privacy, greatly improving system performance. Considering the complex geographic environment, the flexible mobility and large communication radius of unmanned aerial vehicles (UAVs) make them ideal auxiliary devices for wireless communication. Using the UAV as a mobile base station can better provide stable communication signals. However, the number of ground-based IoT terminals is large and closely distributed, so if all of them transmit data to the UAV, the UAV will not be able to take on all of the computation and communication tasks because of its limited energy. In addition, there is competition for spectrum resources among many terrestrial devices, and all devices transmitting data will bring about an extreme shortage of resources, which will lead to the degradation of model performance. This will bring indelible damage to the rescue of the disaster area and greatly threaten the life safety of the vulnerable and injured. Therefore, we use user scheduling to select some terrestrial devices to participate in the FL process. In order to avoid the resource waste generated by the terrestrial device resource prediction, we use the multi-armed bandit (MAB) algorithm for equipment evaluation. Considering the fairness issue of selection, we try to replace the single criterion with multiple criteria, using model freshness and energy consumption weighting as reward functions. The state of the art of our approach is demonstrated by simulations on the datasets.https://www.mdpi.com/1424-8220/24/5/1599emergency communicationUAV-assisted communicationfederated learningclient selectionfairness enhancementmulti-armed bandit (MAB) |
| spellingShingle | Chun Zhu Ying Shi Haitao Zhao Keqi Chen Tianyu Zhang Chongyu Bao A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency Communication Sensors emergency communication UAV-assisted communication federated learning client selection fairness enhancement multi-armed bandit (MAB) |
| title | A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency Communication |
| title_full | A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency Communication |
| title_fullStr | A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency Communication |
| title_full_unstemmed | A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency Communication |
| title_short | A Fairness-Enhanced Federated Learning Scheduling Mechanism for UAV-Assisted Emergency Communication |
| title_sort | fairness enhanced federated learning scheduling mechanism for uav assisted emergency communication |
| topic | emergency communication UAV-assisted communication federated learning client selection fairness enhancement multi-armed bandit (MAB) |
| url | https://www.mdpi.com/1424-8220/24/5/1599 |
| work_keys_str_mv | AT chunzhu afairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT yingshi afairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT haitaozhao afairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT keqichen afairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT tianyuzhang afairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT chongyubao afairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT chunzhu fairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT yingshi fairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT haitaozhao fairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT keqichen fairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT tianyuzhang fairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication AT chongyubao fairnessenhancedfederatedlearningschedulingmechanismforuavassistedemergencycommunication |