Federated Transfer Learning for Distributed Drought Stage Prediction

Abstract

Due to the uncertain nature of drought, it is one of the most menacing natural disasters. Drought modeling (Prediction, Detection, Forecasting, and Stage Prediction) is very essential for efficient policy making. But one of the key problems with drought modeling is the limited availability of centralized datasets. To address this problem, we are a novel proposing federated learning based transfer learning models for the prediction of drought stages. In this study, satellite image dataset was collected from the Tharparkar district (prone to drought) of Pakistan. We trained the dataset using traditional and federated learning approaches, comparing centralized ML models, pre-trained models, and their respective federated learning models (FL-ResNet, FL-DenseNet, FL-MobileNet). The development of these models is the novel aspect of the study specifically for the use case of drought stage prediction. Based on the final evaluation, FL-MobileNet achieved 82% precision while baseline MobileNet scored 68%. The results show the effectiveness of novelty (federated learning), that our proposed framework improves the performance of the drought stage classification task.