נעה דגן

אקדמי בכיר

Addressing bias in prediction models by improving subpopulation calibration

Noam Barda, Gal Yona, Guy N. Rothblum, Philip Greenland, Morton Leibowitz, Ran Balicer,Eitan Bachmat,Noa Dagan

Objective: To illustrate the problem of subpopulation miscalibration, to adapt an algorithm for recalibration of the predictions, and to validate its performance. Materials and Methods: In this retrospective cohort study, we evaluated the calibration of predictions based on the Pooled Cohort Equations (PCE) and the fracture risk assessment tool (FRAX) in the overall population and in subpopulations defined by the intersection of age, sex, ethnicity, socioeconomic status, and immigration history. We next applied the recalibration algorithm and assessed the change in calibration metrics, including calibration-in-The-large. Results: 1 021 041 patients were included in the PCE population, and 1 116 324 patients were included in the FRAX population. Baseline overall model calibration of the 2 tested models was good, but calibration in a substantial portion of the subpopulations was poor. After applying the algorithm, subpopulation calibration statistics were greatly improved, with the variance of the calibration-in-The-large values across all subpopulations reduced by 98.8% and 94.3% in the PCE and FRAX models, respectively. Discussion: Prediction models in medicine are increasingly common. Calibration, the agreement between predicted and observed risks, is commonly poor for subpopulations that were underrepresented in the development set of the models, resulting in bias and reduced performance for these subpopulations. In this work, we empirically evaluated an adapted version of the fairness algorithm designed by Hebert-Johnson et al. (2017) and demonstrated its use in improving subpopulation miscalibration. Conclusion: A postprocessing and model-independent fairness algorithm for recalibration of predictive models greatly decreases the bias of subpopulation miscalibration and thus increases fairness and equality.

שפת פרסום אנגלית
דפים 549-558
כתב עת Journal of the American Medical Informatics Association : JAMIA
כרך 28
נושא מספר 3
סטטוס פרסום פורסם - 01.03.2021

Keywords

Predictive
models, algorithmic fairness, calibration, model bias, cardiovascular disease, osteoporosis

ASJC Scopus subject areas

Health Informatics

Sustainable Development Goals

SDG 3 - Good Health and Well-being
גישה למסמך
10.1093/jamia/ocaa283
קבצים וקישורים אחרים
Link to publication in Scopus