arXiv:2604.22580stat.MLcs.LG2026-04

用几何对齐方法提升天气预报模型解释性,解决传统方法模糊问题

Explanation of Dynamic Physical Field Predictions using WassersteinGrad: Application to Autoregressive Weather Forecasting

论文配图:Explanation of Dynamic Physical Field Predictions using WassersteinGrad: Application to Autoregressive Weather Forecasting
图 1 · 摘自论文原文
  • 提出WassersteinGrad,通过熵正则化Wasserstein均值对齐扰动后归因图
  • 在区域气象数据上验证,相比基线方法显著减少解释模糊性
  • 适合需要高可信度解释的气象、气候等动态物理场预测场景

随着人工智能在高风险场景中应用日益增长,解释神经网络预测背后的推理过程已从理论兴趣变为实际要求。本文聚焦自回归神经网络在动态物理场(如天气预报)中的预测解释。梯度类特征归因方法因其可扩展性被广泛使用,尤其在高维输入中。类似地,SmoothGrad等方法通过平均多个带噪声输入的归因图来增强解释鲁棒性。然而,我们在动态物理场中发现:随机输入扰动不会产生平稳的幅值噪声,而是导致归因特征的空间位移,使逐点平均造成空间错位特征的模糊。为此,我们提出WassersteinGrad,通过计算扰动归因图的熵正则化Wasserstein均值,提取几何一致性。在区域气象数据和经气象学家验证的神经模型上,结果表明WassersteinGrad在单步与自回归预测设置下均显著优于传统梯度基方法,展现出更优的可解释性。

原文摘要 · Abstract (English)

As the demand to integrate Artificial Intelligence into high-stakes environments continues to grow, explaining the reasoning behind neural-network predictions has shifted from a theoretical curiosity to a strict operational requirement. Our work is motivated by the explanations of autoregressive neural predictions on dynamic physical fields, as in weather forecasting. Gradient-based feature attribution methods are widely used to explain the predictions on such data, in particular due to their scalability to high-dimensional inputs. It is also interesting to remark that gradient-based techniques such as SmoothGrad are now standard on images to robustify the explanations using pointwise averages of the attribution maps obtained from several noised inputs. Our goal is to efficiently adapt this aggregation strategy to dynamic physical fields. To do so, our first contribution is to identify a fundamental failure mode when averaging perturbed attribution maps on dynamic physical fields: stochastic input perturbations do not induce stationary amplitude noise in attribution maps, but instead cause a geometric displacement of the attributions. Consequently, pointwise averaging blurs these spatially misaligned features. To tackle this issue, we introduce WassersteinGrad, which extracts a geometric consensus of perturbed attribution maps by computing their entropic Wasserstein barycenter. The results, obtained on regional weather data and a meteorologist-validated neural model, demonstrate promising explainability properties of WassersteinGrad over gradient-based baselines across both single-step and autoregressive forecasting settings.

可解释性天气预报归因分析几何对齐

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