Skip to main content
Log in

A comparison of various machine learning approaches performance for prediction suspended sediment load of river systems: a case study in Malaysia

  • Research Article
  • Published:
Earth Science Informatics Aims and scope Submit manuscript

Abstract

Accurate and reliable suspended sediment load (SSL) prediction models are necessary for the planning and management of water resource structures. In this study, four machine learning techniques, namely Gradient boost regression (GBT), Random Forest (RF), Support vector machine (SVM), and Artificial neural network ANN will be developed to predict SSL at the Rantau Panjang station on Johor River basin (JRB), Malaysia. Four evaluation criteria, including the Correlation Coefficient (R), Root Mean Square Error (RMSE), Nash Sutcliffe Efficiency (NSE) and Scatter Index (SI) will utilize to evaluating the performance of the proposed models. The obtained results revealed that all the proposed Machine Learning (ML) models showed superior prediction daily SSL performance. The comparative outcomes among models were carried out using the Taylor diagram. ANN model shows more reliable results than other models with R of 0.989, SI of 0.199, RMSE of 0.011053 and NSE of 0.979. A sensitivity analysis of the models to the input variables revealed that the absence of current day Suspended sediment load data SSLt-1 had the most effect on the SSL. Moreover, to examine validation of most accurate model we proposed divided data to 50% training, 25% testing and 25% validation) sets and ANN provided superior performance. Therefore, the proposed ANN approach is recommended as the most accurate model for SSL prediction.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

References

Download references

Acknowledgements

This research was supported by the Ministry of Education (MOE) through Fundamental Research Grant Scheme (FRGS/1/2020/TK0/UNITEN/02/16). The authors would like to acknowledge the access of data from Department of Irrigation and Drainage Malaysia (JPS).

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Ali Najah Ahmed.

Additional information

Communicated by H. Babaie.

Publisher’s note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Hanoon, M.S., Abdullatif B, A.A., Ahmed, A.N. et al. A comparison of various machine learning approaches performance for prediction suspended sediment load of river systems: a case study in Malaysia. Earth Sci Inform 15, 91–104 (2022). https://doi.org/10.1007/s12145-021-00689-0

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s12145-021-00689-0

Keywords

Navigation