A short-term flood prediction based on spatial deep learning network: A case study for Xi County, China

Abstract

Floods cause substantial damage across the world every year. Accurate and timely prediction of floods can significantly minimize the loss of life and property. Recently, numerous machine learning models have been used for flood prediction, showing that their performance is preferable to traditional statistical models. However, the existing models neglect the spatial features of floods, which drive flood generation and concentration. In this paper, the area of interest is divided into grids based on longitude and latitude, and the rainfall and discharge collected by stations are combined into tensors according to station coordinates. Different from one-dimensional time series, our input feature is a two-dimensional time series with spatial information. Hence, combining a Convolutional Neural Network (CNN) with a Long Short Term Memory Network (LSTM), we propose the convolution LSTM (ConvLSTM) to extract spatiotemporal features of hydrological information. The methodology is demonstrated using the hydrological data collected at the Xi County stations, located on the Huai River in Henan Province, China. Numerical results indicate that the relative error of arrival time is within 30%, and the relative error of peak discharge is within 20%, satisfying the 2005 Chinese Water Resource Standard on flood prediction permit error. The experiments also show that the ConvLSTM outperforms the recent models in terms of flood arrival time and peak discharge, thereby proving a promising alternative.