基于四维时空感知的暴雨预测方法、装置、设备及介质

本申请公开了一种基于四维时空感知的暴雨预测方法、装置、设备及介质,方法包括：生成包括相互关联的时间序列、垂直层数、空间网格和气象变量的四维张量；基于四维张量和联合约束函数训练得到的目标3D卷积模型对四维张量进行空间特征提取以及时间建模，得到时空联合的第一特征张量；确定每个空间网格对应的空间位置上的暴雨风险权重，并融合权重与第一特征张量，得到第二特征张量，并对其进行基于时间序列的滚动解码处理，得到暴雨预测结果。本申请通过3D卷积模型充分提取四维张量在大气层的垂直层次以及水平空间的局部特征，并对其进行时间维度的解码得到预测结果，实现对气象变量进行时空感知，提升暴雨预测结果的准确性。