A coupled model integrating dual attention mechanism into BiGRU-RED for multi-step-ahead streamflow forecasting

Chunlin Huang,Ting Zhou,Weide Li,Haijiao Yu,Rongxia Li,Jinjie Fang
DOI: https://doi.org/10.1016/j.jhydrol.2024.132137
IF: 6.4
2024-10-16
Journal of Hydrology
Abstract:The accurate streamflow forecast is of utmost importance in the efficient administration of water resources. In this research, we introduced the DA-BiGRU-RED model, an approach that incorporated the dual attention (DA) mechanism involving both feature and temporal attention into the Bidirectional Gated Recurrent Unit (BiGRU) with a recursive encoder-decoder (RED) structure. The feature attention was derived by allocating weights to the hidden states of the BiGRU in the encoder, enhancing the model's capability to efficiently capture crucial features of the input variables. Concurrently, the temporal attention mechanism was established by jointly weighting the hidden states of the BiGRU in the encoder and decoder, enabling the extraction of temporal message from the input variables. This dual-attention mechanism empowered our model to effectively extract essential information from various kinds and temporal instances of input data, thereby improving the accuracy of multi-step streamflow forecasting. Furthermore, to assess forecasting uncertainty, we employed MC dropout based on Bayesian statistical theory. To gauge the effectiveness of our proposed model, we applied it for 1-day, 3-day, 5-day, and 7-day ahead forecasting in the Heihe River basin in Northwest China. Our model consistently outperformed both the BiGRU-ED and BiGRU models, as evidenced by Nash-Sutcliffe coefficient (NSE) values exceeding 0.69 in nearly all prediction scenarios. Additionally, the uncertainty assessment revealed that the DA-BiGRU-RED model exhibited the highest PUCI values, underscoring its efficacy in extracting key features and temporal information from input variables and providing more accurate and robust forecasting results.
geosciences, multidisciplinary,water resources,engineering, civil
What problem does this paper attempt to address?