Chain-of-Influence: Tracing Interdependencies Across Time and Features in Clinical Predictive Modelings
By: Yubo Li, Rema Padman
Potential Business Impact:
Shows how sickness spreads in a patient's body.
Modeling clinical time-series data is hampered by the challenge of capturing latent, time-varying dependencies among features. State-of-the-art approaches often rely on black-box mechanisms or simple aggregation, failing to explicitly model how the influence of one clinical variable propagates through others over time. We propose $\textbf{Chain-of-Influence (CoI)}$, an interpretable deep learning framework that constructs an explicit, time-unfolded graph of feature interactions. CoI leverages a multi-level attention architecture: first, a temporal attention layer identifies critical time points in a patient's record; second, a cross-feature attention layer models the directed influence from features at these time points to subsequent features. This design enables the tracing of influence pathways, providing a granular audit trail that shows how any feature at any time contributes to the final prediction, both directly and through its influence on other variables. We evaluate CoI on mortality and disease progression tasks using the MIMIC-IV dataset and a private chronic kidney disease cohort. Our framework significantly outperforms existing methods in predictive accuracy. More importantly, through case studies, we show that CoI can uncover clinically meaningful, patient-specific patterns of disease progression that are opaque to other models, offering unprecedented transparency into the temporal and cross-feature dependencies that inform clinical decision-making.
Similar Papers
Community Detection on Model Explanation Graphs for Explainable AI
Social and Information Networks
Finds groups of clues that help computers decide.
CoC: Chain-of-Cancer based on Cross-Modal Autoregressive Traction for Survival Prediction
Machine Learning (CS)
Helps doctors guess how long cancer patients will live.
Towards Identifiability of Hierarchical Temporal Causal Representation Learning
Machine Learning (CS)
Lets computers understand complex patterns over time.