arXiv:2410.23159cs.CVcs.AI2024-10NeurIPS被引 26

用频域损失提升降水预报清晰度,解决传统方法模糊问题

Fourier Amplitude and Correlation Loss: Beyond Using L2 Loss for Skillful Precipitation Nowcasting

论文配图:Fourier Amplitude and Correlation Loss: Beyond Using L2 Loss for Skillful Precipitation Nowcasting
图 1 · 摘自论文原文
  • 提出频域幅度与相关性损失,替代传统L2损失
  • 在雷达数据上显著提升感知质量与气象评分,模糊度降低
  • 无需调参、通用性强,适合气象预报与图像生成场景

近年来深度学习广泛用于降水临近预报。以往研究多聚焦于新模型架构以提升像素级指标,但常导致预测结果模糊,难以用于实际预报。本文提出一种新的傅里叶幅度与相关性损失(FACL),包含两个新损失项:傅里叶幅度损失(FAL)和傅里叶相关性损失(FCL)。FAL约束模型预测的傅里叶幅度,FCL补充缺失的相位信息。二者协同作用,替代传统的L2损失(如MSE、加权MSE)用于基于信号的数据时空预测。该方法通用、无参数、高效。在1个合成数据集和3个雷达回波数据集上的大量实验表明,该方法显著提升感知质量与气象技能分数,对像素级精度和结构相似性略有牺牲。为进一步改善气象技能分数(如关键成功指数CSI和分数技能得分FSS)的误差,我们提出并采用区域直方图差异(RHD)作为距离度量,考虑局部变换容忍下的图像模式块级相似性。代码已公开于https://github.com/argenycw/FACL。

原文摘要 · Abstract (English)

Deep learning approaches have been widely adopted for precipitation nowcasting in recent years. Previous studies mainly focus on proposing new model architectures to improve pixel-wise metrics. However, they frequently result in blurry predictions which provide limited utility to forecasting operations. In this work, we propose a new Fourier Amplitude and Correlation Loss (FACL) which consists of two novel loss terms: Fourier Amplitude Loss (FAL) and Fourier Correlation Loss (FCL). FAL regularizes the Fourier amplitude of the model prediction and FCL complements the missing phase information. The two loss terms work together to replace the traditional $L_2$ losses such as MSE and weighted MSE for the spatiotemporal prediction problem on signal-based data. Our method is generic, parameter-free and efficient. Extensive experiments using one synthetic dataset and three radar echo datasets demonstrate that our method improves perceptual metrics and meteorology skill scores, with a small trade-off to pixel-wise accuracy and structural similarity. Moreover, to improve the error margin in meteorological skill scores such as Critical Success Index (CSI) and Fractions Skill Score (FSS), we propose and adopt the Regional Histogram Divergence (RHD), a distance metric that considers the patch-wise similarity between signal-based imagery patterns with tolerance to local transforms. Code is available at https://github.com/argenycw/FACL

降水预报图像生成损失函数频域分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。