English

What went wrong and when? Instance-wise Feature Importance for Time-series Models

Machine Learning 2020-10-29 v3 Machine Learning

Abstract

Explanations of time series models are useful for high stakes applications like healthcare but have received little attention in machine learning literature. We propose FIT, a framework that evaluates the importance of observations for a multivariate time-series black-box model by quantifying the shift in the predictive distribution over time. FIT defines the importance of an observation based on its contribution to the distributional shift under a KL-divergence that contrasts the predictive distribution against a counterfactual where the rest of the features are unobserved. We also demonstrate the need to control for time-dependent distribution shifts. We compare with state-of-the-art baselines on simulated and real-world clinical data and demonstrate that our approach is superior in identifying important time points and observations throughout the time series.

Keywords

Cite

@article{arxiv.2003.02821,
  title  = {What went wrong and when? Instance-wise Feature Importance for Time-series Models},
  author = {Sana Tonekaboni and Shalmali Joshi and Kieran Campbell and David Duvenaud and Anna Goldenberg},
  journal= {arXiv preprint arXiv:2003.02821},
  year   = {2020}
}
R2 v1 2026-06-23T14:05:33.104Z