Federated transfer learning for distributed drought stage prediction
Discover Artificial Intelligence, vol.5, no.1, 2025 (Scopus)
- Publication Type: Article / Article
- Volume: 5 Issue: 1
- Publication Date: 2025
- Doi Number: 10.1007/s44163-025-00288-8
- Journal Name: Discover Artificial Intelligence
- Journal Indexes: Scopus
- Keywords: Distributed learning, Federated learning, Remote sensing, Satellite images, Transfer learning
- Open Archive Collection: AVESIS Open Access Collection
- Yıldız Technical University Affiliated: Yes
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.