Search papers, labs, and topics across Lattice.
This paper addresses the problem of selective conformal prediction in interventional settings where exchangeability holds only within subsets of interventions that leave a target variable unaffected. They derive a contamination-robust conformal coverage theorem that explicitly quantifies the impact of misclassifying "unaffected" calibration examples, and introduce a task-driven partial causal learning formulation to estimate only the descendant indicators needed for selective calibration. Experiments on synthetic SEMs and CRISPRi perturbation data validate the derived bound and demonstrate the applicability of the approach to real-world genomic screens.
Selective conformal prediction can maintain 95% coverage under 30% contamination in interventional settings by learning only the necessary partial causal structure, outperforming standard methods that degrade to 87% coverage.
Selective conformal prediction can yield substantially tighter uncertainty sets when we can identify calibration examples that are exchangeable with the test example. In interventional settings, such as perturbation experiments in genomics, exchangeability often holds only within subsets of interventions that leave a target variable "unaffected" (e.g., non-descendants of an intervened node in a causal graph). We study the practical regime where this invariance structure is unknown and must be learned from data. Our contributions are: (i) a contamination-robust conformal coverage theorem that quantifies how misclassification of "unaffected" calibration examples degrades coverage via an explicit function $g(δ,n)$ of the contamination fraction and calibration set size, providing a finite-sample lower bound that holds for arbitrary contaminating distributions; (ii) a task-driven partial causal learning formulation that estimates only the binary descendant indicators $Z_{a,i}=\mathbf{1}\{i\in\mathrm{desc}(a)\}$ needed for selective calibration, rather than the full causal graph; and (iii) algorithms for descendant discovery via perturbation intersection patterns (differentially affected variable set intersections across interventions), and for approximate distance-to-intervention estimation via local invariant causal prediction. We provide recovery conditions under which contamination is controlled. Experiments on synthetic linear structural equation models (SEMs) validate the bound: under controlled contamination up to $δ=0.30$, the corrected procedure maintains $\ge 0.95$ coverage while uncorrected selective CP degrades to $0.867$. A proof-of-concept on Replogle K562 CRISPR interference (CRISPRi) perturbation data demonstrates applicability to real genomic screens.
