Fed4UL: A Cloud–Edge–End Collaborative Federated Learning Framework for Addressing the Non-IID Data Issue in UAV Logistics

Abstract

Artificial intelligence and the Internet of Things (IoT) have brought great convenience to people’s everyday lives. With the emergence of edge computing, IoT devices such as unmanned aerial vehicles (UAVs) can process data instantly at the point of generation, which significantly decreases the requirement for on-board processing power and minimises the data transfer time to enable real-time applications. Meanwhile, with federated learning (FL), UAVs can enhance their intelligent decision-making capabilities by learning from other UAVs without directly accessing their data. This facilitates rapid model iteration and improvement while safeguarding data privacy. However, in many UAV applications such as UAV logistics, different UAVs may perform different tasks and cover different areas, which can result in heterogeneous data and add to the problem of non-independent and identically distributed (Non-IID) data for model training. To address such a problem, we introduce a novel cloud–edge–end collaborative FL framework, which organises and combines local clients through clustering and aggregation. By employing the cosine similarity, we identified and integrated the most appropriate local model into the global model, which can effectively address the issue of Non-IID data in UAV logistics. The experimental results showed that our approach outperformed traditional FL algorithms on two real-world datasets, CIFAR-10 and MNIST.