Federated Learning and Autonomous UAVs for Hazardous Zone Detection and AQI Prediction in IoT Environment

Abstract

Air pollution monitoring, finding the hazardous zone, and future air quality predictions have recently become a significant issue for many researchers. With the adverse effect of low air quality on human health, it has become necessary for predicting the air quality index (AQI) accurately and on time. The unmanned aerial vehicle (UAV) can collect air quality data with high spatial and temporal resolutions. Using a fleet of UAVs could be considered a good option. In the proposed work, we implement a distributed federated learning (FL) algorithm within a UAV swarm that collects air quality data using built-in sensors. A scheme for finding the area with the highest AQI value is proposed using swarm intelligence. The collected data are then fed to a CNN-LSTM model to predict the AQI. The trained local model is sent to the central server, and the server aggregates the received models from UAVs in the swarm. A global model is created and is transmitted to the UAV swarm again in the next iteration. The proposed architecture is compared with other time-series models. The results show that the proposed model predicts AQI daily with a minimal error rate on a real-time data set from Delhi.