What went wrong and when? Instance-wise feature importance for time-series black-box models
Sana Tonekaboni, Shalmali Joshi, Kieran Campbell, David Duvenaud, Anna Goldenberg
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. ⇤ Equal author † Work done while at the Vector Institute 34th Conference on Neural Information Processing Systems (NeurIPS 2020), Vancouver, Canada.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext bedc50d6-2f48-4220-9ce4-4116e72307ecCited by top-tier papers28
- Unsupervised Representation Learning for Time Series with Temporal Neighborhood CodingSana Tonekaboni, Danny Eytan, Anna GoldenbergICLR 2021 · 386 citations
- Generative Time Series Forecasting with Diffusion, Denoise, and DisentanglementYan Li, Xinjiang Lu, Yaqing Wang, Dejing DouNeurIPS 2022 · 203 citations
- Encoding Time-Series Explanations through Self-Supervised Model Behavior ConsistencyOwen Queen, Tom Hartvigsen, Teddy Koker, Huan He et al.NeurIPS 2023 · 55 citations
- TimeX++: Learning Time-Series Explanations with Information BottleneckZichuan Liu, Tianchun Wang, Jimeng Shi, Xu Zheng et al.ICML 2024 · 33 citations
- Listen to Interpret: Post-hoc Interpretability for Audio Networks with NMFJayneel Parekh, Sanjeel Parekh, Pavlo Mozharovskyi, Florence d'Alché-Buc et al.NeurIPS 2022 · 32 citations
Related papers
- Explaining Time Series Predictions with Dynamic MasksJonathan Crabbé, Mihaela van der SchaarICML 2021 · 115 citations
- Contimask: Explaining Irregular Time Series via Perturbations in Continuous TimeMax Möbus, Björn Braun, Christian HolzNeurIPS 2025 · 2 citations
- Temporal Dependencies in Feature Importance for Time Series PredictionKin Kwan Leung, Clayton Rooke, Jonathan Smith, Saba Zuberi et al.ICLR 2023 · 7 citations
- TimeSAE: Causal Sparse Decoding for Faithful Explanations of Black-Box Time Series ModelsKhalid Oublal, Quentin Bouniot, Qi Gan, Stephan Clemencon et al.ICML 2026 · 2 citations
- TRACER: A Framework for Facilitating Accurate and Interpretable Analytics for High Stakes ApplicationsKaiping Zheng, Shaofeng Cai, Horng Ruey Chua, Wei Wang et al.SIGMOD 2020 · 22 citations
