Explanation of Dynamic Physical Field Predictions using WassersteinGrad: Application to Autoregressive Weather Forecasting
Researchers have developed WassersteinGrad, a new method for explaining neural network predictions in dynamic physical fields, particularly for autoregressive weather forecasting. Existing gradient-based methods struggle with these complex data types, as input perturbations can cause geometric displacements in attribution maps, leading to blurred explanations. WassersteinGrad addresses this by computing an entropic Wasserstein barycenter of perturbed attribution maps to achieve a geometric consensus, showing improved explainability over baseline methods on regional weather data. AI
IMPACT Introduces a novel explainability technique for AI models used in critical applications like weather forecasting.