Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

FasterRisk: Fast and Accurate Interpretable Risk Scores

About

Over the last century, risk scores have been the most popular form of predictive model used in healthcare and criminal justice. Risk scores are sparse linear models with integer coefficients; often these models can be memorized or placed on an index card. Typically, risk scores have been created either without data or by rounding logistic regression coefficients, but these methods do not reliably produce high-quality risk scores. Recent work used mathematical programming, which is computationally slow. We introduce an approach for efficiently producing a collection of high-quality risk scores learned from data. Specifically, our approach produces a pool of almost-optimal sparse continuous solutions, each with a different support set, using a beam-search algorithm. Each of these continuous solutions is transformed into a separate risk score through a "star ray" search, where a range of multipliers are considered before rounding the coefficients sequentially to maintain low logistic loss. Our algorithm returns all of these high-quality risk scores for the user to consider. This method completes within minutes and can be valuable in a broad variety of applications.

Jiachang Liu, Chudi Zhong, Boxuan Li, Margo Seltzer, Cynthia Rudin• 2022

Related benchmarks

TaskDatasetResultRank
Length of Stay (LOS) predictioneICU
AUROC70
12
Atrial Fibrillation (AF) predictionMIMIC
AUROC0.75
12
Chronic Obstructive Pulmonary Disease (COPD) predictionMIMIC
AUROC0.62
12
Lung disease predictionMIMIC
AUROC61
12
Vasopressor (Vaso) use predictioneICU
AUROC0.75
12
Acute Kidney Injury (AKI) predictionMIMIC
AUROC76
12
Cancer PredictionMIMIC
AUROC0.57
12
Heart Failure (HF) predictionMIMIC
AUROC0.73
12
Showing 8 of 8 rows

Other info

Code

Follow for update