CORTEXA
← Browse
arxivstat.MLcs.LG2026-07-03

Denoised Conformal Alignment for Reliable Selection of Conditional Average Treatment Effect Predictions

Xinyun Lu, Haoang Chi, Zhiheng Zhang

In selective deployment, practitioners act only on a model-chosen subset of individuals based on predicted conditional average treatment effects, but marginal conformal guarantees need not control reliability on that selected subset. We study reliable selection for black-box CATE predictors: selecting candidates whose CATE errors are below a tolerance while controlling the false discovery rate (FDR). Since CATE errors are unobservable, we construct doubly robust proxy errors from pseudo-outcomes; however, naive proxies can lose power under heteroskedasticity because variance overwhelms the reliability signal. We propose Denoised Conformal Alignment, which subtracts an estimated conditional variance component and combines conformal calibration with Benjamini--Hochberg selection. Our analysis shows that validity is governed by stability of proxy/oracle threshold labels, rather than pointwise perfection of the variance estimator. Experiments show substantially improved power while maintaining FDR control across challenging settings.

View free PDFSource page

Related papers

arxivstat.MLcs.LGmath.ST2026-07-02

Prediction Sets for Counterfactual Decisions: Coverage, Optimality, and Conformal Prediction

Yurui Zheng, Ying Jin

Predictions are increasingly used to guide high-stakes decisions, from treatment selection to policy making. To ensure reliability with imperfect predictions, uncertainty quantification methods such as conformal prediction build prediction sets with coverage guarantees. However,…

View free PDFSource page
arxivcs.LGstat.ML2026-07-07

A Quiet Failure in Calibrated Virtual Screening: Marginal Conformal Prediction Under-Covers the Minority Class, and a Class-Conditional Fix Recovers It

Muhammadjon Tursunbadalov, Mustafojon Tursunbadalov

Conformal prediction is being adopted in drug discovery to put an honest number on model reliability: pick an error rate alpha, and the method returns prediction sets containing the true label with probability at least 1 - alpha. We show this guarantee can be dangerous on imbalan…

View free PDFSource page
arxivmath.STcs.LGstat.ML2026-06-30

On Optimal Data Splitting for Split Conformal Prediction

Sayan Das, Bahram Yaghooti, Todd A. Kuffner, Soumendra N. Lahiri

Conformal prediction and its variants, including the split conformal prediction, provide a distribution-free framework for uncertainty quantification by constructing prediction intervals or sets with finite-sample coverage guarantees. The statistical efficiency of these intervals…

View free PDFSource page