Prediction of flash flood peak discharge in hilly areas with ungauged basins based on machine learning

Peak discharge is an essential element of hydrological forecasting. Due to rapid outbreaks of flash floods in hilly areas and the lack of measured data, the fast and accurate estimation of peak discharge is crucial for flash flood hazard management. Three machine learning algorithms were applied to estimate peak discharge; this estimation was compared with the results of hydrological–hydraulic models, and the results were verified with measured watershed data. In this paper, 10 hydrological and geomorphological parameters were selected to predict the flood peak discharge in 103 watersheds in Taiyi Mountain North District. The results show that the particle swarm optimization backpropagation (PSO-BP) neural network model outperforms the BP neural network and random forest regression in prediction performance. PSO-BP has a lower mean absolute error (2.51%), root mean square error (3.74%), and mean absolute percentage error (2.74%) than the other models, which indicates that PSO-BP has high prediction accuracy. Importance analysis revealed that rainfall, early impact rainfall, catchment area, and rain intensity are the key input parameters of PSO-BP. The proposed method was confirmed to be a fast and relatively accurate algorithm for estimating the peak discharge of flash floods in ungauged basins.