arXiv · 2609.10778
Counterfactual Marginalisation: Framework for Evaluating Robustness to Nuisance Variables
Abstract
Machine learning models can achieve strong test performance while relying on demographic or acquisition-related shortcuts. We propose counterfactual (CF) marginalisation as a test-time evaluation procedure for assessing robustness of classification models to such variables. Given a CF image generator, we intervene on nuisance parent variables such as age or sex, generate CF versions of each test image, and average predictions over a target intervention distribution. This produces intervention-aware predictions that marginalise demographic effects while preserving patient-specific latent information. We use these predictions to define metrics for CF risk, calibration, stability and worst-case sensitivity. We demonstrate this framework's utility for quantitative robustness evaluation.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Yasin Ibrahim, Hermione Warr, Robin J. Evans, Konstantinos Kamnitsas. 2026-09-09. Counterfactual Marginalisation: Framework for Evaluating Robustness to Nuisance Variables. https://arxiv.org/abs/2609.10778
Cite the original work for its findings. Save a collection to share your selection of sources.