<p>The evolution of hydrological processes was highly nonlinear due to climate change, human activities, and changes in the watershed subsurface. Typical deep learning (DL) schemes still had limitations in terms of accuracy, stability, and computational complexity. In this study, a novel design based on the “encoding-decoding” architecture was adopted, drawing upon recent advances in the field of artificial intelligence. Incorporating a wavelet feature extractor, the Transformer encoder module was retained to encode the features of the input sequences, while a new module, TimesBlock, was introduced to realize runoff prediction. A new model called WaveTransTimesNet (WTTN) was proposed. The experimental results demonstrated that WTTN exhibited superior prediction performance, excellent generalization ability, and strong robustness, with the Kling-Gupta Efficiency coefficient (KGE) reaching 0.94, 0.95, and 0.96, respectively. Compared to other benchmark models, the proposed model not only identified and utilized periodic trend features in the runoff series more effectively but also predicted runoff more accurately. Additionally, it performed exceptionally well in peak prediction.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

WaveTransTimesNet: an enhanced deep learning monthly runoff prediction model based on wavelet transform and transformer architecture

  • Dong-mei Xu,
  • Zong Li,
  • Wen-chuan Wang,
  • Yang-hao Hong,
  • Miao Gu,
  • Xiao-xue Hu,
  • Jun Wang

摘要

The evolution of hydrological processes was highly nonlinear due to climate change, human activities, and changes in the watershed subsurface. Typical deep learning (DL) schemes still had limitations in terms of accuracy, stability, and computational complexity. In this study, a novel design based on the “encoding-decoding” architecture was adopted, drawing upon recent advances in the field of artificial intelligence. Incorporating a wavelet feature extractor, the Transformer encoder module was retained to encode the features of the input sequences, while a new module, TimesBlock, was introduced to realize runoff prediction. A new model called WaveTransTimesNet (WTTN) was proposed. The experimental results demonstrated that WTTN exhibited superior prediction performance, excellent generalization ability, and strong robustness, with the Kling-Gupta Efficiency coefficient (KGE) reaching 0.94, 0.95, and 0.96, respectively. Compared to other benchmark models, the proposed model not only identified and utilized periodic trend features in the runoff series more effectively but also predicted runoff more accurately. Additionally, it performed exceptionally well in peak prediction.