Author
Listed:
- Tadele Melese
- Gizachew Assefa
- Baye Terefe
- Tatek Belay
- Getachew Bayable
- Abebe Senamew
Abstract
Accurate drought prediction is essential for proactive water management and agricultural planning, especially in regions like Ethiopia that are highly susceptible to climate variability. This study investigates the classification of the Palmer Drought Severity Index (PDSI) using machine learning models trained on TerraClimate data, incorporating variables such as precipitation, temperature, soil moisture, and vapor pressure deficit. We employed several classifiers Logistic Regression, Support Vector Machine (SVM), k-Nearest Neighbors (KNN), Decision Tree, Random Forest, Gradient Boosting, Naive Bayes, AdaBoost, and XGBoost with Logistic Regression serving as a baseline statistical approach for comparison. To address data imbalance across drought classes, we applied a hybrid resampling method combining manual upsampling and SMOTE. Hyperparameter tuning was conducted using grid search and cross-validation. Random Forest outperformed all models, achieving an accuracy of 71.18%, F1-score of 0.71, and ROC AUC of 0.9000. Gradient Boosting and SVM also performed well with ROC AUC values of 0.8982 and 0.8681, respectively. SHAP analysis revealed that soil moisture, precipitation, and vapor pressure deficit were the most influential features in predicting drought severity. For benchmarking, an ARIMA (3,1,2) time-series model was applied but yielded poor performance (RMSE = 1.789, R² = −0.077), confirming the advantages of non-linear machine learning techniques for complex climate data. The results highlight the utility of ensemble learning in environmental modelling, offering valuable insights for drought early warning systems and climate resilience planning in Ethiopia. Future work should explore integrating localized predictors and real-time data to enhance prediction robustness.
Suggested Citation
Tadele Melese & Gizachew Assefa & Baye Terefe & Tatek Belay & Getachew Bayable & Abebe Senamew, 2025.
"Machine learning-based drought prediction using Palmer Drought Severity Index and TerraClimate data in Ethiopia,"
PLOS ONE, Public Library of Science, vol. 20(6), pages 1-19, June.
Handle:
RePEc:plo:pone00:0326174
DOI: 10.1371/journal.pone.0326174
Download full text from publisher
Corrections
All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0326174. See general information about how to correct material in RePEc.
If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.
We have no bibliographic references for this item. You can help adding them by using this form .
If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.
For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .
Please note that corrections may take a couple of weeks to filter through
the various RePEc services.