本文提出了一種名為STAA的短期降雨多步驟預測模型,該模型基於注意力機制,包括:
時間對齊模塊SATA:採用二維多頭變量自注意力機制,自動學習不同變量之間的關係,提取時間對齊的特徵嵌入。
時空注意力單元STAU:整合各變量的空間特徵,並捕捉時間序列的長期依賴性,提高模型的表達能力。同時,採用大卷積核作為高通濾波器特徵,增強對突發變化和極端事件的擬合能力。
實驗結果顯示,STAA在RMSE、MAE和PCC指標上均優於現有的ConvLSTM、PhyDNet和SimVP模型,RMSE分別降低了42.20%、13.65%和12.61%。在預測極端降雨事件方面,STAA也表現出更出色的能力。
通過對比實驗和案例分析,證明了SATA和STAU模塊的重要性。SATA能夠快速捕捉突發的降雨變化,而STAU則有助於提高模型對長期時間依賴的建模能力。總的來說,STAA在短期降雨預測任務中取得了顯著的性能提升。
翻译成其他语言
从原文生成
arxiv.org
更深入的查询