Local Discrimination of Binary Classifiers via Localized AUC Functional
Authors: Wenye Ren, Anna Győrffy-Kerekes, Tancredi Dorsi, Olga Demler
Organizations: Department of Computer Science, ETH Zurich, Switzerland · Department of Computer Science, Tsinghua University, China · Center for Learning Systems, Max Planck Institute, Tübingen, Germany · Department of Mathematics, ETH Zurich, Switzerland · Mass General Brigham Hospital, Harvard Medical School, Boston, MA
Risk prediction models are frequently evaluated with AUC, which means that during evaluation we are blind to performance heterogeneity, which could lead to bad decision-making in clinics. In this work, we aim to change that by identifying patient cohorts for whom the risk prediction model might not be an accurate predictor. To separate these cohorts, we use a version of the weighted AUC, which we call Localized AUC or LAUC. Using this metric, we build a method that we coin LOCATOR. This method is able to identify cohorts of patients for whom the risk prediction model is inaccurate. We test this method both in real-world data and in simulations. We also build a statistical background for the use of Localized AUC for our purposes.
Figures & tables
Figure 1: The comparison of predicted \logit(\lauc(xi)) values between model gM ( Model M in legend ) and model gN ( Model N in legend ) in a simulated data set. According to the simulation design in section 4 , model fM should give more precise prediction when x3<0.5 than model fN , while model fN should give more precise prediction when x3≥0.5 . gM and gN are XGBoost models trained based on the weighted \logit(\lauc(xi)) of fM and fN respectively. From the figure it is clear that model gM has larger \logit(\lauc(xi)) than model gN when x3<0.5 while model gN has larger \logit(\lauc(xi)) than model gM when x3≥0.5 , which aligns with our expectation.
Figure 2: ROC curves comparison of low- and high-performance subgroups from the negative-control simulation 4.2 . The XGBoost model used to identify the high- and low-performance subgroups is based on pure noise. The confidence bands of the two ROC curves overlap, as expected.
Figure 3: ROC curves comparison of low- and high-performance subgroups of the simulation as in section 4.3 . The test set is split into low- and high-performance subgroups according to the median of the predictions of the trained XGBoost model. The confidence level is set to be 95%. The high-performance subgroup has significantly higher ROC curve than low-performance subgroup, and the confidence bands are separable. The standard (unweighted) AUC value for low-performance subgroup is AUC=0.69, with CI=0.66-0.71, which is lower than the standard AUC value for high-performance subgroup AUC=0.88, with CI=0.86-0.90. The figure gives another validation for our method in identifying performance subgroups of the certain given model.
Figure 4: The distribution of x1 of low- and high-performance subgroups, identified by the XGBoost model trained in the simulation. The low-performance subgroup is in the middle of two high-performance subgroups, which aligns with the simulation set up x1∼N(A⋅y,1) with two normal distributions overlapping around x1=2A=0.595116 . So at the middle point it is hard to decide whether the data point comes from the event data y=1 or non-event data y=0 .
Figure 5: The three plots show the capability of different XGBoost models with maximum delta (AUC i difference), 70th percentile delta, and median delta. The two ROC curves in a high- and a low-performance subgroup were identified by splitting above and below the median of XGBoost predictions of \logit(\lauc(xi)) . Subgroup 0%−50% corresponds to low-performance subgroup and Subgroup 50%−100% corresponds to high-performance subgroup. The two ROC curves show separation, with AUC values shown in the legend significantly different as expected. With confidence level of 95% , the confidence bands are shown in the figure and the confidence intervals of AUC values are shown in the legend.
Figure 6: AUC with respect to the fraction assigned to the low-and high-performance subgroup in the NHANES data.
Subgroup
Total Number
CVD death number
mean classical AUC value
Low perforamance
2663
190
0.780
High performance
2663
37
0.880
Table 1: Total sizes, CVD death event numbers and mean AUC values of high and low performance subgroups from test data set
Figure 7: External Validaton in UK Biobank. The two ROC curves of a high- and a low-performance subgroup were identified by splitting above and below the median of XGBoost predictions of \logit(\lauc(xi)) Subgroup 0%−50% corresponds to low-performance subgroup and Subgroup 50%−100% corresponds to high-performance subgroup. The two ROC curves show separation, with AUC values shown in the legend significantly different as expected. With confidence level of 95% , the confidence bands are shown in the figure and the confidence intervals of AUC values are shown in the legend.
Figure 8: SHAP summary plot. the most critical variable is age, followed by sex, bp_tx(now taking prescribed medicine), smoking and statin medication use.
Figure 9: Attrition of NHANES data. Initially there are 62,160 respondents from the NHANES 1999–2010 cycles (A–F). We restricted the sample to adults aged 30–79 years (the age range for the PREVENT equations), leaving 23,920 individuals. After excluding pregnant women ( n=466 ) and individuals with self-reported prior cardiovascular disease ( n=2,272 ), 21,182 remained. We further excluded participants with missing data on any of the 11 PREVENT predictor variables ( n=3,151 ), resulting in 18,031 individuals with complete data. Of these, 259 were excluded because the preventr scoring algorithm flagged out‑of‑range input values, leaving 17,772 valid risk scores. Finally, we restricted to participants eligible for mortality linkage, yielding the final analytical cohort of 17,748 individuals.
Figure 10: Absence of UMAP clusters in NHANES datset. UMAP visualization of the distribution of age, sex (male=1, female=0), education (Less Than 9th Grade=1, 9-11th Grade (Includes 12th grade with no diploma)=2, High School Grad/GED or Equivalent=3, Some College or AA degree=4, College Graduate or above=5, Refused=7, Don’t Know=9), race/ethnicity(Mexican American=1, Other Hispanic=2, Non-Hispanic White=3, Non-Hispanic Black=4, Other Race - Including Multi-Racial=5) and the ratio of poverty to income in training data set. From the figure it can be seen that there isn’t obvious cluster of subgroups.
Figure 11: Three ROC curves of three models split using 33% and 67% percentiles of XGBoost predictions \logit(\lauc(xi)) . Subgroup 0%−33% corresponds to low-performance subgroup, Subgroup 33%−67% corresponds to medium-performance subgroup, and Subgroup 67%−100% corresponds to high-performance subgroup. The ROC curves of the three subgroups shows increasing performance when climbing from low- to high-performance subgroup. The AUC values shown in the legend also validate this. With confidence level of 95% , the confidence band is shown in the figure and the confidence intervals of AUC values are shown in the legend.
Figure 12: Harmonization summary of NHANES and UK Biobank variable definitions
Figure 13: The histogram of standard AUC values of low- and high-performance subgroups in the test set of NHANES data with 100 seeds (0-99). The whole method is implemented: train the XGBoost model in the training set, then identify low- and high-performance subgroups in the test set. Then calculate the unweighted AUC in low- and high-performance subgroups respectively. Repeat the procedure for 100 seeds from 0 to 99 and plot the histogram using all AUC values. The figure shows that low-performance subgroup has lower AUC than high-performance subgroup. The mean AUC of low-performance subgroup is 0.744, with standard variation 0.0157. The mean AUC of high-performance subgroup is 0.822, with standard vatiation 0.0394.
Figure 14: UMAP visualization of the distribution of age, sex (male=1, female=0), education (Less Than 9th Grade=1, 9-11th Grade (Includes 12th grade with no diploma)=2, High School Grad/GED or Equivalent=3, Some College or AA degree=4, College Graduate or above=5, Refused=7, Don’t Know=9), race/ethnicity(Mexican American=1, Other Hispanic=2, Non-Hispanic White=3, Non-Hispanic Black=4, Other Race - Including Multi-Racial=5) and the ratio of poverty to income in testing data set.
Figure 15: The histogram of the calculated weighted AUC in test set. The chosen seed is 64, corresponding to the maximum AUC-delta parsimonious model. Mean AUC: 0.8260, Median AUC: 0.8282
Figure 16: The histogram of the calculated weighted AUC in train set. The chosen seed is 64, corresponding to the maximum AUC-delta parsimonious model. Mean AUC: 0.8321, Median AUC: 0.8359
Figure 17: The histogram of the Gower distance in test set. The chosen seed is 64, corresponding to the maximum AUC-delta parsimonious model. The histograms show that the distribution of Gower distance is close to Poisson or Gaussian distribution, which helps show that both the training set and testing set are non-biased and are not having preference towards specific subgroups which share certain similarities.
Grouped by cvd_death
Missing
Overall
False
True
P-Value
n
17748
16992
756
Age (years), mean (SD)
0
51.5 (13.5)
50.9 (13.2)
65.3 (11.0)
< 0.001
Sex, n (%)
female
8989 (50.6)
8671 (51.0)
318 (42.1)
< 0.001
male
8759 (49.4)
8321 (49.0)
438 (57.9)
Race / Ethnicity, n (%)
Mexican American
3682 (20.7)
3557 (20.9)
125 (16.5)
< 0.001
Table 2: Baseline Demographic and Clinical Characteristics
Grouped by Outcome
Missing
Overall
High-Performance Subgroup
Low-Performance Subgroup
P-Value
n
3550
1775
1775
Hypertension Meds, n (%)
False
2698 (76.0)
1608 (90.6)
1090 (61.4)
< 0.001
True
852 (24.0)
167 (9.4)
685 (38.6)
Statin Use, n (%)
False
3099 (87.3)
1692 (95.3)
1407 (79.3)
< 0.001
True
451 (12.7)
83 (4.7)
368 (20.7)
Table 3: Baseline Demographic and Clinical Characteristics
Binary classification assigns examples to one of two classes while balancing accuracy, calibration, class imbalance, and the costs of different errors.
In high-stakes risk prediction, interval-valued predictions provide a natural way to represent predictive uncertainty. However, standard evaluation tools such as the receiver operating characteristic (ROC) curve and the area under the curve (AUC) are designed for point-valued predictions and do not capture how uncertainty affects discrimination performance. To address this gap, we develop an uncertainty-aware ROC framework and corresponding interval-based AUC (iAUC) metrics for evaluating the performance of interval-valued risk predictions. The framework constructs two ROC-style curves with associated areas AUCL and AUCU, and shows that these quantities induce a three-region decomposition of positive-negative rankings into confidently correct, ambiguous, and confidently incorrect orderings. Under valid class-conditional coverage, AUCL and AUCU, together with the pairwise miscoverage rate, provide theoretical bounds on the optimal AUC, linking interval coverage to achievable discrimination. The framework is compatible with a range of methods for quantifying epistemic uncertainty in risk prediction, including Bayesian, bootstrap, and ensemble-based procedures, and provides a practical way to compare predictive models and interval-generation methods. Experiments on clinical benchmark data and MIMIC-IV mortality prediction show that models with similar point AUC can have different uncertainty-aware ranking profiles, and that the proposed decomposition can support risk-aware model selection and threshold-based abstention.
Yuqi Li, Matthew M. Engelhard
Department of Biostatistics & Bioinformatics, Duke University
Accurate diagnostic and risk-prediction models are important for supporting clinical decision-making during infectious disease outbreaks. However, privacy and governance requirements may restrict patient-level data sharing across healthcare institutions, and data distributions often vary. Moreover, AUC is widely used to evaluate discriminative performance, motivating its direct optimization in model development. We propose geographically regularized AUC-maximizing personalized federated learning (GrAUC-PFL), which directly optimizes a smooth pairwise AUC surrogate to learn personalized models while keeping patient-level data local and accounting for institutional heterogeneity. Graph-based regularization encourages geographically neighboring institutions to have similar coefficient vectors while retaining a personalized models. Simulations and a real-data application suggest improved discriminative performance, particularly when geographically neighboring institutions have similar data-generating characteristics.
Mayu Hiraishi, Kensuke Tanioka, Toshio Shimokawa
Department of Medicine · Wakayama Medical University · Wakayama, Japan +3
Modern machine learning models can be accurate on average yet still make mistakes that dominate deployment cost. We introduce Locus, a distribution-free wrapper that produces a per-input loss-scale reliability score for a fixed prediction function. Rather than quantifying uncertainty about the label, Locus models the realized loss of the prediction function using any engine that outputs a predictive distribution for the loss given an input. A simple split-calibration step turns this function into a distribution-free interpretable score that is comparable across inputs and can be read as an upper loss level. The score is useful on its own for ranking, and it can optionally be thresholded to obtain a transparent flagging rule with distribution-free control of large-loss events. Experiments across 13 regression benchmarks show that Locus yields effective risk ranking and reduces large-loss frequency compared to standard heuristics.
Matheus Barreto, Mário de Castro, Thiago R. Ramos +2
Department of Statistics, Federal University of S˜ao Carlos, S˜ao Carlos, S˜ao Paulo, Brazil · Institute of Mathematics and Computer Science, University of S˜ao Paulo, S˜ao Carlos, S˜ao Paulo, Brazil · School of Forest, Fisheries, and Geomatics Sciences, University of Florida, Gainesville, Florida, United States of America