Short-term rainfall prediction based on radar echo using an efficient spatio-temporal recurrent unit.

Journal: Scientific reports
Published Date:

Abstract

Accurate short-term precipitation prediction is critical for agricultural production, transportation safety, and water resource management. In this paper, we propose an Efficient Spatio-Temporal Recurrent Unit (ESTRU) for short-term precipitation prediction based on radar echoes. The ability of the model to process spatio-temporal information is enhanced by fusing two ConvGRU units while controlling the complexity. The trajectory tracking structure (TTS) facilitates the capture of rotational and scaling motions and improves the model's adaptability in complex meteorological conditions. The combined effect of the Self-Attention (SA) mechanism and convolution allows the model to focus on both global and local dependencies in spatial information, improving the clarity of the generated images. ESTRU demonstrated the best performance on the radar echo dataset compared to the other nine classical models. Quantitative and qualitative results show that ESTRU can efficiently model complex spatio-temporal relationships in radar echoes.

Authors

  • Dali Wu
    School of Computer and Information Science, Qinghai Institute of Technology, Xining, 810016, China.
  • Shunli Zhang
    Department of General Medicine, The Affiliated Yongchuan Hospital of Chongqing Medical University, Chongqing 402160, China.
  • Guohong Zhao
    School of Information Engineering, Qinghai Vocational and Technical University, Xining, 810003, China.
  • Yongchao Feng
    School of Computer and Information Science, Qinghai Institute of Technology, Xining, 810016, China.
  • Yuan Ma
    School of Public Health, Capital Medical University, Beijing 100069, China; Beijing Municipal Key Laboratory of Clinical Epidemiology, Beijing 100069.
  • Yue Zhang
    Department of Ophthalmology, Beijing Hospital, National Center of Gerontology, Institute of Geriatric Medicine, Chinese Academy of Medical Sciences, Beijing, China.

Keywords

No keywords available for this article.