IDEAS home Printed from https://ideas.repec.org/a/spr/waterr/v37y2023i13d10.1007_s11269-023-03606-w.html
   My bibliography  Save this article

Assessment of XGBoost to Estimate Total Sediment Loads in Rivers

Author

Listed:
  • Reza Piraei

    (Shiraz University)

  • Seied Hosein Afzali

    (Shiraz University)

  • Majid Niazkar

    (Free University of Bozen-Bolzano)

Abstract

Estimation of total sediment loads is a significant topic in river management as direct measurement is costly and time-consuming. This study aims not only to use the eXtreme Gradient Boosting (XGBoost) model but also to compare its performance with that of other empirical equations and ML models, including Artificial Neural Networks (ANN), AdaBoost, Gradient Boost Regressor, Random Forest Regressor, and Gaussian Process. 543 data points from the United States Geological Survey were used to train and test different models. The results showed that XGBoost outperformed other methods considering six performance metrics. To be more specific, the root mean square errors and determination coefficient were 216 and 0.95, respectively, whereas the corresponding metrics for ANN were 316.23 and 0.87, respectively. To interpret the sediment predictions and delineate the importance of each feature, XGBoost feature importance and SHapley Additive exPlanations (SHAP) were utilized. According to the feature importance analysis, estimations of the XGBoost model was mostly (72%) affected by the water surface width. Moreover, SHAP analysis verified the importance of water surface width on the final predictions. Finally, based on the results achieved in this study, further applications of XGBoost in water resources management are postulated.

Suggested Citation

  • Reza Piraei & Seied Hosein Afzali & Majid Niazkar, 2023. "Assessment of XGBoost to Estimate Total Sediment Loads in Rivers," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 37(13), pages 5289-5306, October.
    • Handle: RePEc:spr:waterr:v:37:y:2023:i:13:d:10.1007_s11269-023-03606-w
    DOI: 10.1007/s11269-023-03606-w
    as

    Download full text from publisher

    File URL: http://link.springer.com/10.1007/s11269-023-03606-w
    File Function: Abstract
    Download Restriction: Access to the full text of the articles in this series is restricted.
    File URL: https://libkey.io/10.1007/s11269-023-03606-w?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    References listed on IDEAS

    as
    1. Javad Zahiri & Zeynab Mollaee & Mohammad Reza Ansari, 2020. "Estimation of Suspended Sediment Concentration by M5 Model Tree Based on Hydrological and Moderate Resolution Imaging Spectroradiometer (MODIS) Data," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 34(12), pages 3725-3737, September.
    2. Elham Ghanbari-Adivi & Mohammad Ehteram & Alireza Farrokhi & Zohreh Sheikh Khozani, 2022. "Combining Radial Basis Function Neural Network Models and Inclusive Multiple Models for Predicting Suspended Sediment Loads," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 36(11), pages 4313-4342, September.
    3. Huiting Zheng & Jiabin Yuan & Long Chen, 2017. "Short-Term Load Forecasting Using EMD-LSTM Neural Networks with a Xgboost Algorithm for Feature Importance Evaluation," Energies, MDPI, vol. 10(8), pages 1-20, August.
    4. Hamid Moeeni & Hossein Bonakdari, 2018. "Impact of Normalization and Input on ARMAX-ANN Model Performance in Suspended Sediment Load Prediction," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 32(3), pages 845-863, February.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Masoud Karbasi & Mohammad Ghasemian & Mehdi Jamei & Anurag Malik & Ozgur Kisi, 2024. "Developing Extended and Unscented Kalman Filter-Based Neural Networks to Predict Cluster-Induced Roughness in Gravel Bed Rivers," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 38(8), pages 3023-3048, June.
    2. Amir Moradinejad, 2024. "Suspended Load Modeling of River Using Soft Computing Techniques," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 38(6), pages 1965-1986, April.
    3. Anurag Barthwal & Mamta Bhatt & Shwetank Avikal & Chandra Prakash, 2025. "Machine learning-based prediction models for electoral outcomes in India: a comparative analysis of exit polls from 2014–2021," Quality & Quantity: International Journal of Methodology, Springer, vol. 59(1), pages 313-338, February.
    4. Congguang Xu & Wei Xiong & Simin Zhang & Hailiang Shi & Shichao Wu & Shanju Bao & Tieqiao Xiao, 2025. "Research on the Nonlinear Relationship Between Carbon Emissions from Residential Land and the Built Environment: A Case Study of Susong County, Anhui Province Using the XGBoost-SHAP Model," Land, MDPI, vol. 14(3), pages 1-21, February.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Elham Ghanbari-Adivi & Mohammad Ehteram & Alireza Farrokhi & Zohreh Sheikh Khozani, 2022. "Combining Radial Basis Function Neural Network Models and Inclusive Multiple Models for Predicting Suspended Sediment Loads," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 36(11), pages 4313-4342, September.
    2. Gebre Gelete, 2023. "Hybrid Extreme Gradient Boosting and Nonlinear Ensemble Models for Suspended Sediment Load Prediction in an Agricultural Catchment," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 37(14), pages 5759-5787, November.
    3. Bibhuti Bhusan Sahoo & Sovan Sankalp & Ozgur Kisi, 2023. "A Novel Smoothing-Based Deep Learning Time-Series Approach for Daily Suspended Sediment Load Prediction," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 37(11), pages 4271-4292, September.
    4. Bingjie Jin & Guihua Zeng & Zhilin Lu & Hongqiao Peng & Shuxin Luo & Xinhe Yang & Haojun Zhu & Mingbo Liu, 2022. "Hybrid LSTM–BPNN-to-BPNN Model Considering Multi-Source Information for Forecasting Medium- and Long-Term Electricity Peak Load," Energies, MDPI, vol. 15(20), pages 1-20, October.
    5. Jiaan Zhang & Chenyu Liu & Leijiao Ge, 2022. "Short-Term Load Forecasting Model of Electric Vehicle Charging Load Based on MCCNN-TCN," Energies, MDPI, vol. 15(7), pages 1-25, April.
    6. Ijaz Ul Haq & Amin Ullah & Samee Ullah Khan & Noman Khan & Mi Young Lee & Seungmin Rho & Sung Wook Baik, 2021. "Sequential Learning-Based Energy Consumption Prediction Model for Residential and Commercial Sectors," Mathematics, MDPI, vol. 9(6), pages 1-17, March.
    7. Umut Ugurlu & Ilkay Oksuz & Oktay Tas, 2018. "Electricity Price Forecasting Using Recurrent Neural Networks," Energies, MDPI, vol. 11(5), pages 1-23, May.
    8. Mohamed Massaoudi & Ines Chihi & Lilia Sidhom & Mohamed Trabelsi & Shady S. Refaat & Fakhreddine S. Oueslati, 2021. "Enhanced Random Forest Model for Robust Short-Term Photovoltaic Power Forecasting Using Weather Measurements," Energies, MDPI, vol. 14(13), pages 1-20, July.
    9. Wei, Nan & Li, Changjun & Peng, Xiaolong & Li, Yang & Zeng, Fanhua, 2019. "Daily natural gas consumption forecasting via the application of a novel hybrid model," Applied Energy, Elsevier, vol. 250(C), pages 358-368.
    10. Gong, Mingju & Zhao, Yin & Sun, Jiawang & Han, Cuitian & Sun, Guannan & Yan, Bo, 2022. "Load forecasting of district heating system based on Informer," Energy, Elsevier, vol. 253(C).
    11. Abdulelah Alkesaiberi & Fouzi Harrou & Ying Sun, 2022. "Efficient Wind Power Prediction Using Machine Learning Methods: A Comparative Study," Energies, MDPI, vol. 15(7), pages 1-24, March.
    12. Alexandros Menelaos Tzortzis & Sotiris Pelekis & Evangelos Spiliotis & Evangelos Karakolis & Spiros Mouzakitis & John Psarras & Dimitris Askounis, 2023. "Transfer Learning for Day-Ahead Load Forecasting: A Case Study on European National Electricity Demand Time Series," Mathematics, MDPI, vol. 12(1), pages 1-24, December.
    13. Luyu Ding & Yang Lv & Ruixiang Jiang & Wenjie Zhao & Qifeng Li & Baozhu Yang & Ligen Yu & Weihong Ma & Ronghua Gao & Qinyang Yu, 2022. "Predicting the Feed Intake of Cattle Based on Jaw Movement Using a Triaxial Accelerometer," Agriculture, MDPI, vol. 12(7), pages 1-18, June.
    14. Zheng Wan & Hui Li, 2023. "Short-Term Power Load Forecasting Based on Feature Filtering and Error Compensation under Imbalanced Samples," Energies, MDPI, vol. 16(10), pages 1-22, May.
    15. Xiaolu Wei & Yubo Tian & Na Li & Huanxin Peng, 2024. "Evaluating ensemble learning techniques for stock index trend prediction: a case of China," Portuguese Economic Journal, Springer;Instituto Superior de Economia e Gestao, vol. 23(3), pages 505-530, September.
    16. Xiaosheng Peng & Kai Cheng & Jianxun Lang & Zuowei Zhang & Tao Cai & Shanxu Duan, 2021. "Short-Term Wind Power Prediction for Wind Farm Clusters Based on SFFS Feature Selection and BLSTM Deep Learning," Energies, MDPI, vol. 14(7), pages 1-18, March.
    17. Wei, Nan & Yin, Chuang & Yin, Lihua & Tan, Jingyi & Liu, Jinyuan & Wang, Shouxi & Qiao, Weibiao & Zeng, Fanhua, 2024. "Short-term load forecasting based on WM algorithm and transfer learning model," Applied Energy, Elsevier, vol. 353(PA).
    18. Tianao Wu & Wei Zhang & Xiyun Jiao & Weihua Guo & Yousef Alhaj Hamoud, 2020. "Comparison of five Boosting-based models for estimating daily reference evapotranspiration with limited meteorological variables," PLOS ONE, Public Library of Science, vol. 15(6), pages 1-28, June.
    19. Federico Divina & Aude Gilson & Francisco Goméz-Vela & Miguel García Torres & José F. Torres, 2018. "Stacking Ensemble Learning for Short-Term Electricity Consumption Forecasting," Energies, MDPI, vol. 11(4), pages 1-31, April.
    20. Mohammed Achite & Saeed Samadianfard & Nehal Elshaboury & Milad Sharafi, 2023. "Modeling and optimization of coagulant dosage in water treatment plants using hybridized random forest model with genetic algorithm optimization," Environment, Development and Sustainability: A Multidisciplinary Approach to the Theory and Practice of Sustainable Development, Springer, vol. 25(10), pages 11189-11207, October.

    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:spr:waterr:v:37:y:2023:i:13:d:10.1007_s11269-023-03606-w. 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.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with 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: Sonal Shukla or Springer Nature Abstracting and Indexing (email available below). General contact details of provider: http://www.springer.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.