Tilted Mondrian: Triage and Mitigation of Fine-grained Disparities in Conformal Prediction
Organizations: McGill University, Montreal, Canada · TD Insurance, Montreal, Canada · Layer 6 AI, Toronto, Canada
Abstract
Fairness in conformal prediction is often pursued through Mondrian calibration, which provides finite-sample coverage guarantees separately across groups. These guarantees, however, are coarse: substantial disparities can remain within finer strata. This coarse-to-fine fairness issue raises two questions: how severe can such disparities be, and how can they be mitigated? Aggregate disparities can depend on how groups are distributed across strata. For severity assessment, we introduce propensity standardization to examine the effect of population composition, and show how group composition can induce disagreement between aggregate and finer-stratum comparisons. For mitigation, we develop Tilted Mondrian to improve fairness compared to group-Mondrian at a finer level without additional model training. Tilted Mondrian retains groupwise coverage guarantees and mitigates coverage and prediction set size disparities within finer strata. Finally, we derive a method-agnostic bound that quantifies how much reduction in finer-stratum coverage disparity can be achieved.
Figures & tables
| Bias in Bios | |||
|---|---|---|---|
| Score | Reversals | Conflict strength | ESS |
| SIMPLE | |||
| APS | |||
| RAPS | |||
| SAPS | |||
| Experiment | Score | Harm | |||
|---|---|---|---|---|---|
| Bias in Bios | SIMPLE | (surgeon) | |||
| Folktables occupation | SIMPLE | (Less than high school) | |||
| Folktables income | SIMPLE | (Bachelor) | |||
| ATUS | SIMPLE | (employed) |
| Experiment | Score | Harm | |||
|---|---|---|---|---|---|
| Bias in Bios | SIMPLE | (psychologist) | |||
| Folktables occupation | SIMPLE | (High school or GED) | |||
| Folktables income | SIMPLE | (Graduate or professional) | |||
| ATUS | SIMPLE | (not in labor force) |
Appendix figures & tables25 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Meaning |
|---|---|
| Input, label, label space, and number of classes | |
| Nonconformity score and split-conformal threshold | |
| Delivered prediction set and set size | |
| Grid of nominal miscoverage levels used for curve reporting | |
| Selected information efficiency anchor | |
| Nonempty set-size version used in Proposition 7 |
| Setting | Year(s) | Train | Cal. | Prop./Tilt | Test | Total |
|---|---|---|---|---|---|---|
| Bias in Bios | all | 127,023 | 79,390 | 79,390 | 31,756 | 317,559 |
| Folktables occupation | 2014 | 163,527 | 102,204 | 102,205 | 40,882 | 408,818 |
| Folktables occupation | 2015 | 162,130 | 101,331 | 101,332 | 40,533 | 405,326 |
| Folktables occupation | 2016–2018 each | 164,195 | 102,622 | 102,622 | 41,049 | 410,488 |
| Folktables income quintile | 2014–2018 each | 164,195 | 102,622 | 102,622 | 41,049 | 410,488 |
| ATUS | all | 71,356 | 44,683 | 44,655 | 17,783 | 178,477 |
| Setting | Threshold policy | Scores | grid | Support rule |
|---|---|---|---|---|
| Bias in Bios | Gender-conditional thresholds | SIMPLE; APS; SAPS( ); RAPS( , ) | 0.020 to 0.100; step 0.005 | At least 200 test observations per Gender group within each occupation stratum; the same supported grid is used for raw and standardized summaries. |
| Folktables occupation | Pooled threshold within year | SIMPLE | 0.010 to 0.100; step 0.001 | At least 200 test observations per group within each education stratum. |
| Folktables income quintile | Pooled threshold within year | SIMPLE | 0.010 to 0.100; step 0.001 | Same test education stratum support rule as Folktables occupation. |
| ATUS | Pooled threshold | SIMPLE primary; APS, RAPS( , ), and SAPS( ) checks | 0.020 to 0.250; step 0.010 | Prespecified employment stratum support rule. |
| Setting | Predictor and score | Profile | Propensity model |
|---|---|---|---|
| Bias in Bios | DistilBERT top-10 occupation classifier; SIMPLE, APS, SAPS, and RAPS. | 37-dimensional frozen probability profile. | Feed-forward neural-network propensity model with hidden layers and ReLU activations; fitted on the propensity split. |
| Folktables occupation | Histogram gradient boosting on ACS person records, one model per year; SIMPLE. | 37-dimensional frozen softmax profile; Gender and education are excluded from . | Logistic propensity model with balanced class weights; fitted separately by year on the propensity split. |
| Folktables income quintile | Histogram gradient boosting for income quintiles, one model per year; SIMPLE. | 22-dimensional frozen softmax profile; race and education are excluded from . | Logistic propensity model with balanced class weights; fitted separately by year on the propensity split. |
| ATUS | Regularized histogram gradient boosting on diary, time, calendar, demographic, region, and household features; SIMPLE with APS, RAPS, and SAPS checks. | Four probability features: , , margin, and entropy. | Logistic propensity model fitted on the propensity split. |
| Dataset | Alignment loss | Squared composition residual | Stratum-gap variance |
|---|---|---|---|
| Bias in Bios | – | – | – |
| Folktables occupation | – | – | – |
| Folktables income | – | – | – |
| ATUS | – | – | – |
| Measure | Value |
|---|---|
| Effective sample size fraction | |
| Profile mean standardized mean difference | |
| Profile maximum standardized mean difference |
| Measure | Raw | Standardized |
|---|---|---|
| Reversals, summed | ||
| High conflict strength reversals | ||
| Conflict strength | ||
| Years with reversal count drop | ||
| Years with strength drop | ||
| Median cell total variation | ||
| Measure | Raw | Standardized |
|---|---|---|
| Reversals, summed | ||
| High conflict strength reversals | ||
| Conflict strength | ||
| Years with reversal count drop | ||
| Years with strength drop | ||
| Median cell total variation | ||
| Measure | Raw | Standardized |
|---|---|---|
| SIMPLE reversals | ||
| SIMPLE conflict strength | ||
| Maximum profile standardized mean difference | ||
| Employment cell total variation | ||
| Effective sample size fraction | ||
| Report | Target population | Weight behavior | Interpretation |
|---|---|---|---|
| Raw group curve | Group-specific | No reweighting | Observed-mixture burden delivered to each group. |
| Marginal-profile curve | Marginal distribution | and | Full-population standardized contrast; sensitive to weak support. |
| Standardized curve | Common standardized distribution | Bounded factors and | Propensity standardized contrast; bounded and stable. |
| Uniform-cell curve | Selected reporting cells equally weighted | Cell weights chosen by design | Cell-equal sensitivity summary; can overweight rare or unstable cells. |
| Experiment | Group and intervention stratum |
|---|---|
| Bias in Bios | Gender and occupation, with |
| Folktables occupation | Gender and education bucket |
| Folktables income | Race group and education bucket |
| ATUS | Gender and employment status |
| Experiment | Score | Harm | ||||
|---|---|---|---|---|---|---|
| Bias in Bios | APS | (nurse) | ||||
| Bias in Bios ( ) | RAPS | (nurse) | ||||
| Bias in Bios | SAPS | (surgeon) | ||||
| Folktables occupation | APS | (Bachelor) | ||||
| Folktables occupation | RAPS | (Less than high school) | ||||
| Folktables occupation | SAPS | (Bachelor) |
| Dataset | Score | Theorem 6 bound | Active term | Tilt split | Test |
|---|---|---|---|---|---|
| Bias in Bios | SIMPLE | 0.156 | gap | 0.125 | 0.071 |
| APS | 0.028 | -budget term | 0.017 | 0.008 | |
| RAPS ( ) | 0.041 | gap | 0.040 | 0.006 | |
| SAPS | 0.166 | gap | 0.164 | 0.173 | |
| Folktables occupation | SIMPLE | 0.037 | gap | 0.022 | 0.024 |
| APS | 0.039 | -budget term | 0.021 | 0.019 |
| Score | |||||
|---|---|---|---|---|---|
| Bias in Bios | |||||
| SIMPLE | |||||
| APS | |||||
| RAPS ( ) | |||||
| SAPS | |||||
| Folktables occupation | |||||
| Score | Method | Bias in Bios | Folktables occupation | Folktables income | ATUS |
|---|---|---|---|---|---|
| SIMPLE | Tilted | ||||
| Cellwise | |||||
| APS | Tilted | ||||
| Cellwise | |||||
| RAPS | Tilted | ||||
| Cellwise |
| Score | Method | Bias in Bios | Folktables occupation | Folktables income | ATUS |
|---|---|---|---|---|---|
| SIMPLE | Tilted | ||||
| Cellwise | |||||
| APS | Tilted | ||||
| Cellwise | |||||
| RAPS | Tilted | ||||
| Cellwise |
| Method | Tilt split worst-gap reduction | Test worst-gap reduction | Tilt split maximum harm | Test mean set size change | Null interventions |
|---|---|---|---|---|---|
| Bias in Bios | |||||
| TM (20) | |||||
| BTM (10) | |||||
| Folktables occupation | |||||
| TM (20) | |||||
| BTM (10) | |||||
| Tilt split final mean gap | Test final mean gap | |||
|---|---|---|---|---|
| Dataset | Tilted Mondrian | Boosted Tilted Mondrian | Tilted Mondrian | Boosted Tilted Mondrian |
| Bias in Bios | ||||
| Folktables occupation | ||||
| Folktables income | ||||
| American Time Use Survey | ||||
| Setting | Raw standardized strength | Relative reduction median [P5, P95] (%) | Reversal D/T/I | Test ESS |
|---|---|---|---|---|
| Bias in Bios | ||||
| SIMPLE | ||||
| APS | ||||
| SAPS | ||||
| RAPS | ||||
| Folktables occupation | ||||
| Score | Harm | P/T/N | |||
|---|---|---|---|---|---|
| Bias in Bios | |||||
| SIMPLE | |||||
| APS | |||||
| RAPS | |||||
| SAPS | |||||
| Folktables occupation | |||||
| Score | Harm | (pp) | P/T/N | ||
|---|---|---|---|---|---|
| Bias in Bios | |||||
| SIMPLE | |||||
| APS | |||||
| RAPS | |||||
| SAPS | |||||
| Folktables occupation | |||||
| Coverage-targeted mitigation | Set-size-targeted mitigation | |||
| Score | (pp) | (pp) | ||
| Bias in Bios | ||||
| SIMPLE | ||||
| APS | ||||
| RAPS | ||||
| SAPS | ||||
| Coverage-targeted mitigation | Set-size-targeted mitigation | |||
| Score | median cell | max cell | median cell | max cell |
| Bias in Bios | ||||
| SIMPLE | ||||
| APS | ||||
| RAPS | ||||
| SAPS | ||||