Skip to content

Conversation

JPXKQX
Copy link
Member

@JPXKQX JPXKQX commented Sep 3, 2025

Description

This PR introduces functionality for computing sensitivities within anemoi-inference. Sensitivities help quantify how model inputs influence a perturbed output, enabling better interpretability and diagnostic analysis during different forecast events.
Workflow

The sensitivity computation workflow follows a similar pattern to running a forecast:

  • Select a checkpoint, an initial time, and the forecast length.
  • Define a perturbation applied to the model output.
  • Compute sensitivities, returned as a tensor with the same shape as the input_state.

Currently Supported

  • Perturbations defined as a % of the predicted field for a single variable.
  • Forecast length limited to 1.

Planned Extensions

  • Additional perturbation types (e.g., random noise, location-specific, etc.).
  • Support for forecast lengths greater than 1.

As a contributor to the Anemoi framework, please ensure that your changes include unit tests, updates to any affected dependencies and documentation, and have been tested in a parallel setting (i.e., with multiple GPUs). As a reviewer, you are also responsible for verifying these aspects and requesting changes if they are not adequately addressed. For guidelines about those please refer to https://anemoi.readthedocs.io/en/latest/

By opening this pull request, I affirm that all authors agree to the Contributor License Agreement.

@github-project-automation github-project-automation bot moved this to To be triaged in Anemoi-dev Sep 3, 2025
@github-actions github-actions bot added tests and removed tests labels Sep 3, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
Status: To be triaged
Development

Successfully merging this pull request may close these issues.

4 participants