DynSHAP: Towards Explainable Dynamic Survival Analysis
Nastasya Anokhina, Jonas Jürß, Pietro Liò
Abstract
Deep learning models for dynamic survival analysis (DSA) achieve strong predictive performance by incorporating longitudinal patient data, but their black box nature limits clinical trust and adoption. Existing explainability methods cannot handle longitudinal, irregular inputs and functional survival outputs simultaneously, which limits their usability in DSA. We propose DynSHAP, a SHAP framework suited specifically for dynamic survival analysis. It extends common marginal SHAP estimators to this setting by treating time--feature pairs as players in the Shapley game. We further introduce Temporal DynSHAP, which learns linear dependencies in features over time and uses conditional sampling to address them in explanations. When applied to synthetic data with known ground-truth attributions, Temporal DynSHAP recovers temporally dependent features more accurately than marginal estimators for a given state-of-the-art model. Applied to two real-world clinical datasets and two DSA architectures, DynSHAP produces attributions faithful to model learning, allowing medical experts to see which patient information drove the prediction and when.
Create a lesson
Related papers
MAxBench: A Multinomial Concept Recovery Benchmark
Divya Appapogu, Freya Behrens, Yonatan Belinkov et al.
CanvasAnneal: Curriculum Reinforcement Learning for Diffusion Language Models
Blake Olson, Yuhang Song, Emmett McQuinn et al.
Benign Loss Landscapes Can Coexist with Worst-Case Hardness
Zach Furman, Stephan Wäldchen, Yangda Bei et al.
A Unified and Constrained View of Regularization-Based Robust Reinforcement Learning
Amine Andam, Jamal Bentahar, Mustapha Hedabou
MCRL2: Multi-resource Cross-attention-based Representation Learning-augmented Reinforcement Learning for Cloud Microservice Scheduling
Tiangang Li, Shi Ying, Xiangbo Tian et al.
Robust Policy Optimization via Adversarial Importance Sampling
Amine Andam, Jamal Bentahar, Mustapha Hedabou