DA-UNet:基于多模态融合的降雨预测

DA-UNET: RAINFALL PREDICTION BASED ON MULTIMODAL FUSION

  • 摘要: 为提高降雨预测的准确率,同时针对网络计算复杂度高的问题,提出一种基于卷积注意力机制和深度可分离卷积的多模态融合预测降雨的方法——DA-UNet。DA-UNet利用多模态信息的互补性,通过注意力机制捕捉图像各部分之间的特征长依赖关系来提升降水预报的准确率,通过改变特征提取过程来降低网络参数。在MeteoNet数据集上进行对比实验及消融实验,结果表明,与其他算法相比,多模态融合的DA-UNet全面提高降雨的预报性能,同时将参数量减少四分之三。

     

    Abstract: To improve the accuracy of rainfall prediction and to solve the problem of high computational complexity, a DA-UNet with multimodal fusion for rainfall prediction based on convolutional block attention mechanism and depthwise separable convolution is proposed. The DA-UNet integrated multimodal data, captured the feature dependence between each part of the image through the attention mechanism to improve the accuracy of precipitation forecast, and reduced the network parameters by varying the feature extraction process. Comparative experiments and ablation experiments were carried out on MeteoNet dataset. The results show that the multimodal fusion of DA-UNet improves the overall rainfall forecasting performance compared with other algorithms, while reducing the parameter volume by three-quarters.

     

/

返回文章
返回