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

• Main Authors: Hanoon, Marwah Sattar, Abdullatif, Alharazi Abdulhadi B., Ahmed, Ali Najah, Razzaq, Arif, Birima, Ahmed H., Ahmed El-Shafie, Ahmed Hussein Kamel
• Format: Article
• Published: Springer Heidelberg 2022
• Subjects: QA75 Electronic computers. Computer science; QE Geology
• Online Access: http://eprints.um.edu.my/33746/

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.