Short-Term Flood Prediction Model Based on Pre-Training Enhancement

Yang Xia,Jiamin Lu
DOI: https://doi.org/10.3390/electronics13112203
IF: 2.9
2024-06-05
Electronics
Abstract:With the rapid advancement of deep learning techniques, deep learning-based flood prediction models have drawn significant attention. However, for short-term prediction in small- and medium-sized river basins, models typically rely on hydrological data spanning from the past several hours to one day, utilizing a fixed-length input window. Such input limits the models' adaptability to diverse rainfall events and restricts their capability to comprehend historical temporal patterns. To address the underutilization of historical information by existing models, we introduce a Pre-training Enhanced Short-term Flood Prediction Method (PE-SFPM) to enrich the model's temporal understanding without necessitating changes to the input window size. In the pre-training stage, the model uses a random masking and prediction strategy to learn segment features, capturing a more comprehensive evolutionary trend of historical floods. In the flow forecasting stage, temporal features and spatial features are captured and fused using the temporal attention, spatial attention, and gated fusion. Features are further enhanced by integrating segment features using a feed-forward network. Experimental results demonstrate that the proposed PE-SFPM model achieves excellent performance in short-term flood prediction tasks.
engineering, electrical & electronic,physics, applied,computer science, information systems
What problem does this paper attempt to address?
This paper presents a new approach to solve the problem of short-term flood prediction, especially for small to medium-sized watersheds. Current models typically rely on hydrological data from the past few hours to one day and use fixed-length input windows, which limit their adaptability to different rainfall events and their ability to understand historical temporal patterns. The paper introduces a pre-training enhanced short-term flood prediction method (PE-SFPM) that learns fragment features of historical data during the pre-training phase, enhancing the model's understanding of time without changing the input window size. During the flow prediction phase, the model captures and combines time and spatial features through time attention, spatial attention, and gate fusion. Experimental results show that the proposed PE-SFPM model performs well in the task of short-term flood prediction.