一种基于transformer架构的降雨预测方法

本发明公开一种基于transformer架构的降雨预测方法，包括构建Transformer降雨模型，对不同时间尺度的特征进行充分提取，Transformer降雨模型模型内部设置有多尺度通道模块、Transformer模块和预测头模块，通过Transformer降雨模型对激光雷达降雨数据进行时序处理和异常值处理，提出了一个数据分布合理的数据集，在提出数据集后提出了一个多尺度通道注意力机制，用于在通道时空间上提取特征，对数据集进行的综合实验验证，完成目标区域的降雨预测；本发明由三个关键部分组成，多尺度通道模块、Transformer模块和预测头模块，针对激光雷达数据的类三维结构，引入多尺度通道模块，提升Transformer的特征提取能力并解决长时间序列中的记忆力衰减问题，通过对双通道分配不同权重来实现特征融合。