stat.MLSep 28, 2026

A Hierarchy of Entropy-Shapley Games for Multivariate Predictive Uncertainty

Authors: Niklas Koenen, Claudia Battistin, Jeriek Van den Abeele, Martin Jullum

Organizations: Leibniz Institute for Prevention Research and Epidemiology – BIPS University of Bremen, Germany · Simula Research Laboratory Oslo, Norway · Telenor Research & Innovation Fornebu, Norway · Norwegian Computing Center Oslo, Norway

Abstract

Modern probabilistic machine learning models increasingly produce multivariate outputs with complex dependence structure, from multi-step time-series forecasts to sample path predictions. Understanding which input features drive the predictive uncertainty is important for risk-aware decisions, model diagnostics, and deciding whether the uncertainty should be mitigated or hedged against. This attribution problem requires a choice of how dependencies between output components are treated. Existing approaches reduce the output to a scalar through aggregation or projection before attribution, thereby obscuring whether features affect marginal uncertainty, dependence structure, or both, while component-wise analyses can miss dependence effects entirely. We close this gap by introducing a hierarchy of three entropy-based Shapley games that make this output-side choice explicit for any ordered multivariate outcome, ranging from per-component marginal entropy to fully joint entropy. The hierarchy isolates a cross-component attribution term that captures how each feature shifts the dependence between output components, a quantity invisible to component-wise methods. We establish a chain-rule decomposition of the joint attribution and characterize the cross-component term through conditional total correlation, providing both closed-form and sample-based estimators. Finally, we demonstrate how the framework captures differences in learned joint structure across probabilistic models from distributional regression to a zero-shot time series foundation model.

Figures & tables

Appendix figures & tables4 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Explaining a probabilistic prediction on the simplex with Shapley compositions

    Aug 2, 2024Paul-Gauthier Noé, Miquel Perelló-Nieto, Jean-François Bonastre +1Shapley ValueMulticlass Classification

  2. Beyond Shapley: Efficient Computation of Asymmetric Shapley Values

    Jun 23, 2026Ezequiel Companeetz, Santiago Cifuentes, Sergio AbriolaShapley ValueEquivalence

  3. RelShap: Relationally Consistent Shapley Explanations

    Aug 11, 2026Seungeun Lee, Joao Fonseca, Julia StoyanovichShapley Additive ExplanationsShapley Value