Calibration drift in regression and machine learning models for acute kidney injury.

Davis SE, Lasko TA, Chen G, Siew ED, Matheny ME
J Am Med Inform Assoc. 2017 24 (6): 1052-1061

PMID: 28379439 · PMCID: PMC6080675 · DOI:10.1093/jamia/ocx030

Objective - Predictive analytics create opportunities to incorporate personalized risk estimates into clinical decision support. Models must be well calibrated to support decision-making, yet calibration deteriorates over time. This study explored the influence of modeling methods on performance drift and connected observed drift with data shifts in the patient population.

Materials and Methods - Using 2003 admissions to Department of Veterans Affairs hospitals nationwide, we developed 7 parallel models for hospital-acquired acute kidney injury using common regression and machine learning methods, validating each over 9 subsequent years.

Results - Discrimination was maintained for all models. Calibration declined as all models increasingly overpredicted risk. However, the random forest and neural network models maintained calibration across ranges of probability, capturing more admissions than did the regression models. The magnitude of overprediction increased over time for the regression models while remaining stable and small for the machine learning models. Changes in the rate of acute kidney injury were strongly linked to increasing overprediction, while changes in predictor-outcome associations corresponded with diverging patterns of calibration drift across methods.

Conclusions - Efficient and effective updating protocols will be essential for maintaining accuracy of, user confidence in, and safety of personalized risk predictions to support decision-making. Model updating protocols should be tailored to account for variations in calibration drift across methods and respond to periods of rapid performance drift rather than be limited to regularly scheduled annual or biannual intervals.

© The Author 2017. Published by Oxford University Press on behalf of the American Medical Informatics Association. All rights reserved. For Permissions, please email: journals.permissions@oup.com

MeSH Terms (13)

Acute Kidney Injury Aged Bayes Theorem Clinical Decision-Making Decision Support Techniques Female Hospitals, Veterans Humans Logistic Models Machine Learning Male Middle Aged United States

Connections (1)

This publication is referenced by other Labnodes entities:

Links