Skip to content
Preprint

From Training to Deployment: Post-Hoc Causal Feature Identification via Sensitivity Ratios

Jul 2026 · 0 citations · 10 references
Computer Science

TL;DR

The Normalised Sensitivity Ratio (NSR) is introduced, a post-hoc, model-agnostic diagnostic for this question under a structured-shift regime: environments differ primarily in the mean of spurious features while the causal mechanism and causal marginals remain stable, as in multi-site clinical data or multi-batch genomics.

Abstract

Given a model that is already trained, which features does it rely on causally versus spuriously? Existing methods require access to the training procedure and cannot answer this post-hoc. We introduce the \textbf{Normalised Sensitivity Ratio~(NSR)}, a post-hoc, model-agnostic diagnostic for this question under a structured-shift regime: environments differ primarily in the mean of spurious features while the causal mechanism and causal marginals remain stable, as in multi-site clinical data or multi-batch genomics. Within this regime, causal features induce constant model sensitivity across environments while spurious features track shift. NSR formalises this as the squared coefficient of variation of per-environment sensitivity. Under a linear structural causal model (SCM) with $K\ge3$ non-degenerate environments, NSR achieves exact identification (Theorem~1). We fully characterise failure: weak shifts ($O(\varepsilon^4)$ collapse), degenerate geometry, and proxy attenuation ($O((1-\alpha)^4)$), giving practitioners quantitative criteria for assessing whether the regime holds. Finite-sample rates are $O_p(n^{-1})$ under the null and $O_p(n^{-1/2})$ under the alternative. Experiments confirm all theoretical predictions on synthetic data (area under the ROC curve [AUROC] $= 1.000$ under conditions satisfying the regime), show consistent rankings across five model families (Kendall $\tau\ge0.529$), and recover six of eight causal features on bike-sharing data (Precision@7 $= 0.75$) without modifying any trained model.

View source

Similar papers

Preprint Jul 2026

CalTwin: Towards Calibrated, Shift-Robust Medical World Models via Fisher-Information Regularisation

The combined objective is derived, which proof steps transfer from the classification setting without modification and which require adaptation, and which require adaptation on the PhysioNet 2019 Sepsis Challenge, treating the two hospital systems as sequential training fragments and the unseen system as an out-of-distribution test.

Behraj Khan, Shabir Ahmad, S. Bukhari et al. · 1 citation
Preprint Jul 2026

Complexity-Budgeted, Interaction-Aware Interpretable Model for Tabular Data

Results support IAIML as a compact, interaction-aware framework appropriate for settings where bounded explanation size and controlled treatment of feature interactions are design requirements.

S. Krishnamoorthy · 0 citations
Review Open access Jul 2026

CausalShift: A Modular, Plugin-Based Framework for Dataset Shift Handling in Machine Learning

CausalShift is proposed, a modular, plugin-based framework for end-to-end dataset shift handling that reduces the in-distribution to out-of-distribution accuracy gap, while remaining competitive on real-world image shift and achieving performance parity with ERM on mild-shift tasks.

Shuang Song, Muhammad Syafiq Mohd Pozi, Nik F. Farid · 0 citations
Preprint Jul 2026

Entangled by Design: Spurious Intra-Variable Signal Routing in Tabular In-Context Learners

It is proved that under ridge ICL, a linear in-context learner, this routing is unavoidable regardless of context size; TabPFN, a state-of-the-art pretrained tabular ICL model, shows qualitatively consistent behaviour empirically, and two lightweight mitigations are introduced.

Athanasios Vlontzos, G. Papanastasiou, Bernhard Kainz et al. · 0 citations
Aug 2026

Post-pretrained lasso statistical inference

This work proposes post-pretrained lasso selective inference (PPL-SI), a novel selective inference method designed to provide statistically valid p values for the pretrained lasso that reliably controls false discoveries and significantly improves the detection of biologically relevant features compared to traditional approaches.

Cao Huyen My, Nguyen Vu Khai Tam, Vo Nguyen Le Duy · 0 citations