NeFut Logo NeFut
Admin Login

[CS.AI] Counterfactual Marginalisation: Framework for Evaluating Robustness to Nuisance Variables

Published at: 2026-09-13 22:00 Last updated: 2026-09-15 01:15
#algorithm #AI #Machine Learning

Machine learning models often achieve high test accuracy while relying on demographic or acquisition shortcuts. We introduce counterfactual (CF) marginalisation as a test‑time evaluation procedure to assess classification robustness against such nuisance variables. The method assumes a CF image generator; for each test image we intervene on parent nuisance variables (e.g., age, sex) to produce CF variants, then average the model predictions over a target intervention distribution. This yields predictions that marginalise demographic effects yet preserve patient‑specific latent information. Using these predictions we define metrics for CF risk, calibration, stability and worst‑case sensitivity. Experiments on several datasets demonstrate the framework’s usefulness for quantitative robustness evaluation.

Review

Original Source: https://arxiv.org/abs/2609.10778

[h] Back to Home