高级检索

基于频域增强与因果注意的卫星云图预测算法

Satellite Cloud Image Prediction Algorithm Based on Frequency-Domain Enhancement and Causal Attention

  • 摘要: 针对卫星云图预测中高频细节衰减与长程依赖刻画不足等问题,提出基于频域增强与因果注意的卫星云图预测算法(WA-CLSTM)。该算法基于双尺度时序骨干网络SeqConv,在浅层引入受离散小波分解启发的可学习频率分解模块(Learnable Frequency Decomposition Module, LFDM),对最新时刻特征进行低、高频响应分量解耦与残差回注,从而增强云缘与细窄云带等高频细节信息;在瓶颈尺度引入因果时序聚合模块(Causal Temporal Aggregation Module, CTAM)聚合跨时序上下文,形成稳定趋势锚点;同时在解码端融合时间聚合特征与浅层增强特征,实现细节与语义的同尺度重建。实验结果表明,在FY-4B卫星云图数据集上,以未来4个预测步(T7-T10)的平均结果为统计依据,WA-CLSTM的均方误差(MSE)比F-CLSTM降低了7.9 %,结构相似性(SSIM)提升了4.5 %,峰值信噪比(PSNR)提升了1.71 dB,表现出更优的预测精度与结构保持能力。

     

    Abstract: To address the problems of high-frequency detail degradation and insufficient modeling of long-range dependencies in satellite cloud image prediction, a frequency-domain enhancement and causal-attention based forecasting algorithm, WA-CLSTM, is proposed. Built upon a dual-scale temporal backbone network, SeqConv, the proposed model introduces a Learnable Frequency Decomposition Module (LFDM) at the shallow stage, which decouples low- and high-frequency response components of the most recent features and reinjects them in a residual manner, thereby enhancing high-frequency details such as cloud edges and narrow cloud bands. At the bottleneck scale, a Causal Temporal Aggregation Module (CTAM) is employed to aggregate cross-temporal context and form stable trend anchors. Meanwhile, in the decoding stage, temporally aggregated features and shallow enhanced features are fused to achieve same-scale reconstruction of fine details and semantic structures. Experimental results on the FY-4B satellite cloud image dataset show that, using the average results over four future prediction steps (T7–T10) as the evaluation basis, WA-CLSTM reduces the mean squared error (MSE) by 7.9%, improves the structural similarity index (SSIM) by 4.5%, and increases the peak signal-to-noise ratio (PSNR) by 1.71 dB compared with F-CLSTM, demonstrating superior prediction accuracy and structural preservation capability.

     

/

返回文章
返回