本發明涉及基于復頻域深度注意力體素流的短臨降水預測方法,具體涉及一種基于復頻域深度注意力體素流的短臨降水預測方法。
背景技術:
1、短時臨近降雨預報是大氣科學領域的一個重要任務,旨在在未來短時間段內(通常為0-2小時)準確預報當地降雨情況。目前,該技術已經被廣泛應用于氣象災害預警,交通運輸管,城市建設管理,農業生產等方面。其重要性不僅在于方便人們的生活,更在于有助于防災減災,這一直是天氣預報領域的關鍵任務。然而,隨著氣候變化和城市化進程的加速,大氣情況變得越來越復雜,各種氣象現象頻繁發生。氣候變化給人們的生活和工作帶來了許多不利影響,增加了許多不確定性的危險。有效地預測和預防上述氣象現象,將有助于減少損失。
2、近年來,隨著計算機視覺和深度學習技術的快速發展,視頻預測技術逐漸應用于氣象領域。視頻預測技術通過對連續時間序列圖像進行分析和建模,能夠預測未來時間點的圖像變化情況。這種技術在短臨降水預測中具有顯著的優勢,因為降水過程本質上是一個時空演變的動態過程,視頻預測技術能夠更好地捕捉和模擬這一動態變化。然而,目前的視頻預測技術在短臨降水預測中的應用仍面臨一些挑戰。例如,氣象圖像數據的復雜性和多變性要求模型具有較強的適應能力和泛化能力;同時,如何有效融合多源氣象數據,提高預測的準確性和可靠性,也是一個亟待解決的問題。
技術實現思路
1、本發明的目的是針對現有技術存在的不足,提供一種基于復頻域深度注意力體素流的短臨降水預測方法。
2、為實現上述目的,本發明提供了一種基于復頻域深度注意力體素流的短臨降水預測方法,包括:
3、將連續時間序列的氣象圖像數據構建為三維視頻流數據;
4、將所述三維視頻流數據輸入至構建的三維時空體素運動模型中,所述三維時空體素運動模型包括空間路徑、運動路徑和頻域路徑;
5、所述空間路徑用以根據輸入的三維視頻流數據預測未來時間點的三維體素狀態;
6、所述運動路徑用以基于時空屬性隨機游走相似性算法捕獲三維視頻流數據節點間的動態關系;
7、所述頻域路徑用以將輸入的三維視頻流數據反演獲得時域的預測信息;
8、對所述三維體素狀態、節點間的動態關系和時域的預測信息進行融合,以獲得最終的短臨降水預測結果,concat為融合操作。
9、進一步的,所述空間路徑包括體素編碼模塊、雙通道注意力選擇下采樣網絡和體素解碼模塊,所述體素編碼模塊用以對輸入的三維視頻流數據進行特征編碼,所述雙通道注意力選擇下采樣網絡用以在特征提取下采樣過程中避免特征丟失,所述體素解碼模塊用以進行特征解碼,以獲得三維體素狀態,其中,表示第一時間通道,為氣象圖像的高,為氣象圖像的寬。
10、進一步的,所述雙通道注意力選擇下采樣網絡通過分離體素編碼模塊輸出的特征來得到特征和特征,具體如下:
11、
12、其中,為卷積運算操作,為分離操作;
13、所述雙通道注意力選擇下采樣網絡對體素編碼模塊輸出的特征基于全局平均池化操作和全連接操作分別得到特征和特征,并對特征通過歸一化操作分別獲得信道權重和信道權重,所述特征和特征分別為:
14、
15、其中,、為像素索引,為relu激活函數,為批量歸一化;
16、對特征和特征分配信道權重,獲得特征和特征,并在特征和特征方向上選擇重要的特征信道進行融合,具體如下:
17、
18、
19、
20、其中,為所述雙通道注意力選擇下采樣網絡輸出的特征,為特征選擇操作。
21、進一步的,所述運動路徑包括動態圖網絡,所述動態圖網絡捕獲三維視頻流數據節點間的動態關系的方式具體如下:
22、將每一個圖都轉化成一個加權圖,其中,表示頂點子集合,表示邊集合,s、v、p分別為狀態、節點、位置;
23、計算兩個節點之間的相似度為:
24、
25、其中,、為加權圖中節點i和節點j對應的特征向量, p表示屬性向量的維度;
26、計算時間上和空間上的節點?i?與節點?j?之間的相似性度量矩陣為:
27、
28、其中,為空間上的節點?i?與節點?j?之間的相似性度,為時間上的節點?i?與節點?j?之間的相似性度,和分別是空間和時間屬性加權融合系數,為設定的隨機游走步數,為歸一化因子,表示節點?i?與節點?j之間的相似度分數,表示i的鄰居節點集合,表示j的鄰居節點集合;
29、通過引入一個雙向gru模塊,確保動態圖網絡充分學習到時間序列數據中的前后文關系,并通過一個標準化層優化了數據分布,并通過一個多層感知機線性層提煉數據特征,然后通過引入了殘差連接策略避免深層網絡在學習過程中可能出現的性能退化,最終輸出節點間的動態關系,表示第二時間通道。
30、進一步的,所述頻域路徑通過快速傅里葉變換轉換到頻域,然后進行濾波處理,使用單個復值線性層對頻率進行插值,再將插值后的頻率數據映射回時域,以獲得頻域路徑特征,表示第三時間通道。
31、有益效果:本發明提出了視頻預測的方法來預測未來降水的分布,可以及時捕捉到云圖的演變和降水的形成過程,并且提高了空間分辨率;在空間路徑中,避免使用池化層,提出了雙通道注意力選擇下采樣,解決了時空信息丟失,預測精度低的問題;在運動路徑中,捕捉并模擬節點變量之間的動態時空關系,這有助于提高模型對于數據中關鍵特征的捕捉和利用,從而提升預測的準確性;在頻域路徑中,通過傅里葉變換可以提供信號在時間和頻率上的詳細信息,從而進行時頻分析,有利于識別信號中的突發事件、頻率變化或頻率成分隨時間的演變等問題,可以幫助捕獲數據中的動態特性和變化模式,從而提高預測模型的精度。
1.一種基于復頻域深度注意力體素流的短臨降水預測方法,其特征在于,包括:
2.根據權利要求1所述的一種基于復頻域深度注意力體素流的短臨降水預測方法,其特征在于,所述空間路徑包括體素編碼模塊、雙通道注意力選擇下采樣網絡和體素解碼模塊,所述體素編碼模塊用以對輸入的三維視頻流數據進行特征編碼,所述雙通道注意力選擇下采樣網絡用以在特征提取下采樣過程中避免特征丟失,所述體素解碼模塊用以進行特征解碼,以獲得三維體素狀態,其中,表示第一時間通道,為氣象圖像的高,為氣象圖像的寬。
3.根據權利要求2所述的一種基于復頻域深度注意力體素流的短臨降水預測方法,其特征在于,所述雙通道注意力選擇下采樣網絡通過分離體素編碼模塊輸出的特征來得到特征和特征,具體如下:
4.根據權利要求1所述的一種基于復頻域深度注意力體素流的短臨降水預測方法,其特征在于,所述運動路徑包括動態圖網絡,所述動態圖網絡捕獲三維視頻流數據節點間的動態關系的方式具體如下:
5.根據權利要求1所述的一種基于復頻域深度注意力體素流的短臨降水預測方法,其特征在于,所述頻域路徑通過快速傅里葉變換轉換到頻域,然后進行濾波處理,使用單個復值線性層對頻率進行插值,再將插值后的頻率數據映射回時域,以獲得頻域路徑特征,表示第三時間通道。