Sam Babichenko

Reading notes / Missing and selective labels

The selective labels problem

Lakkaraju et al., 2017

Context. To ask whether a model would make better bail decisions than judges, one needs outcomes for the defendants the model would release, but outcomes exist only for those the judge released. The labels are selected by the very decisions being compared against, by people who see things the model does not (demeanor, who came with the defendant). Every standard fix, from imputation to doubly robust estimation (Dudík et al., next in this section), assumes no unmeasured confounding and is biased here. Lakkaraju et al. impute nothing and use the variation between decision-makers who see comparable cases. The most lenient one's released set contains an outcome for every case any stricter policy on that caseload would release, provided the stricter policy also refuses what the lenient one refused. So a model can be evaluated by contracting that released set, and the error this leaves depends only on how much the model disagrees with those refusals, which can be computed from the data. Kleinberg et al. (2018, QJE) make the same design a central tool in their study of bail decisions in New York City.

Setting. Data {(xi,ji,ti,yi)}\{(x_i,j_i,t_i,y_i)\}: recorded features xix_i, decision-maker jij_i, decision ti∈{0,1}t_i\in\{0,1\} (1 = release, which produces a label), and outcome yiy_i observed only when ti=1t_i=1. Decision-makers also see unobservables ziz_i that may affect yy. A black-box model BB, trained on separate data, scores risk from xx alone. The acceptance rate is the fraction released, and the failure rate is bad outcomes among those released divided by all cases judged, so the two must be read together as a curve. Contraction needs many decision-makers, cases assigned to them as if at random, and acceptance rates that vary.

Contraction and its error bound. Let qq be the decision-maker with the highest acceptance rate ψ\psi, with caseload DqD_q and released set RqR_q. To evaluate BB at acceptance rate r≤ψr\le\psi, keep everything qq detained detained, also detain the cases in RqR_q that BB scores riskiest until the rate is rr, and report u^(r)=#{failures among the cases still released}∣Dq∣.\hat u(r)=\frac{\#\{\text{failures among the cases still released}\}}{|D_q|}. Let aa be the fraction of qq's detained cases that BB, ranking all of DqD_q on its own, would also detain at rate rr. Then the error relative to BB's true failure rate on DqD_q is at most ∣u^(r)−uB(r)∣≤(1−a) ∣Dq∖Rq∣∣Dq∣.\big|\hat u(r)-u_B(r)\big|\le\frac{(1-a)\,|D_q\setminus R_q|}{|D_q|}. The machine curve r↦u^(r)r\mapsto\hat u(r) is compared with a human curve built by binning decision-makers by acceptance rate.

Step 1

Evaluating on labeled cases alone, or imputing the rest, overstates the model's performance. Released cases are a population the judge has already screened using zz. If a family's presence in court predicts a good outcome for young defendants and judges use it, the labeled data show few failures among the young, and a model trained on them learns that youth is safe and fails on the young defendants the judges would have detained. Imputation from xx assumes y⊥t∣xy\perp t\mid x; when zz drives both, detained cases are riskier than released cases with the same xx, so imputed outcomes understate failures. Contraction counts only observed outcomes, so unobservables can enter only through the swapped cases the bound counts.

Step 2

Random assignment lets one caseload stand in for another, and leniency supplies the labels. The model runs on qq's cases but is compared with judge kk at kk's rate, which is fair only if DqD_q and DkD_k are draws from one distribution of (x,z,y)(x,z,y). Since qq has labeled a superset of what a stricter policy that keeps qq's refusals would release, the model can be evaluated at any rate up to ψ\psi and none above, which would release cases qq detained. One condition is implicit: a released case's outcome must not depend on who released it, or the outcomes in RqR_q are qq's outcomes and not those any policy would see.

Step 3

The error is bounded by the number of swapped cases, which the data determine. BB's own detentions at rate rr include a∣Dq∖Rq∣a|D_q\setminus R_q| of qq's detainees, so it releases the other m=(1−a)∣Dq∖Rq∣m=(1-a)|D_q\setminus R_q| and detains correspondingly more from RqR_q than contraction does. The two released sets therefore differ by exactly mm cases on each side, with unknown outcomes on one side; failures in each group lie between 0 and mm, which gives the bound, and at a=1a=1 contraction is exact. Since aa uses only qq's decisions and BB's scores on DqD_q, the interval's width is known before looking at a label. It is a worst case over the unknown outcomes, conditional on the sample, and says nothing about sampling error in DqD_q.

Step 4

On synthetic data with a strong unobservable, contraction tracks the truth and imputation reverses the conclusion. With 100 judges of 500 cases each, failure depends on X+Z+0.2WX+Z+0.2W with ZZ seen only by judges, who detain by nearly that index at acceptance rates from 0.1 to 0.9, and the model is a logistic regression trained on the labeled cases of a training half. Contraction follows the true failure-rate curve closely, while labeled-only evaluation and five imputation methods, propensity matching and doubly robust estimation among them, understate the model's failure rate enough to put it below the judges' curve, when in truth the judges, who see ZZ, do better. The imputation error grows with the weight on ZZ, and at weight 1 contraction's mean absolute error is already 6.4 times below the best imputation method's. Its error rises as the lenient rate or the agreement falls, as the bound predicts, and as the lenient caseload shrinks, which is sampling error. The simulation favors contraction, since judges rank by nearly the outcome index, so swapped cases are both high-risk and their failures largely cancel: in my rerun with a model ranking by XX, agreement at r=ψ=0.9r=\psi=0.9 is only about 0.4, yet the actual error is about a tenth of the bound.

Step 5

On three real data sets the model beats the humans, with agreement high enough to trust the estimate. On bail decisions, asthma treatment and insurance underwriting, with gradient-boosted trees trained on the labeled training cases, the contraction curve lies below the human curve at every acceptance rate, and the labeled-only curve lower still, which the paper reads as over-optimism. Agreement rates are 0.891, 0.932 and 0.718, computed at the lenient decision-maker's own rate, where agreement is lowest. An F-test fails to reject random assignment on all three, but it checks balance on recorded features only, and balance on zz is assumed. With no ground truth, these results rest on the assumptions and the bound, whose width the paper does not report.

Consequences and limits. Contraction replaces "no unmeasured confounding" with "as-if random assignment plus varying leniency", the structure of the judge-leniency designs in economics. It reaches only rates up to ψ\psi, with the precision of one caseload, and evaluates a fixed model without saying how to train on selectively labeled data. Payment fraud has the same structure: declined transactions never get an outcome, so scoring a new model on approved traffic alone is the labeled-only evaluation shown here to overstate its performance. If traffic is split at random across policies or review queues that decline at different rates, the most permissive one plays the lenient judge, and the swap bound says how much of a comparison rests on outcomes nobody saw; a small randomly approved slice, which the paper rules out on ethical grounds for bail, is often an acceptable cost there and gives labels directly. The failure labels inside approved traffic are themselves late chargebacks with unreported fraud counted as clean, so every failure rate is an undercount, and contraction compares policies fairly only if that undercount does not differ between the released sets being compared.

Terms

Selective labels
outcomes observed only for cases the decision-maker accepted
Unobservables zz
information the decision-maker uses that the data do not record; breaks y⊥t∣xy\perp t\mid x
Acceptance rate
fraction of a decision-maker's cases accepted (released, approved)
Failure rate
bad outcomes among accepted cases divided by all cases judged
qq, ψ\psi
the most lenient decision-maker and its acceptance rate; the model can be evaluated only at rates up to ψ\psi
Contraction
keep qq's refusals, refuse further cases from RqR_q by model score, count observed failures
Agreement aa
share of qq's refusals the model would also refuse at rate rr; computable from features alone
Swap bound
∣u^−uB∣≤(1−a)∣Dq∖Rq∣/∣Dq∣|\hat u-u_B|\le(1-a)|D_q\setminus R_q|/|D_q|; exact when a=1a=1
Random assignment
caseloads are draws from one distribution, so one caseload stands in for another
Labeled-only evaluation
scoring the model on accepted cases; optimistic because they were screened using zz

Quiz

  1. When is the contraction estimate exactly the model's failure rate on qq's caseload?
  2. Why can contraction not evaluate the model at acceptance rates above ψ\psi?
  3. The paper tests random assignment by regressing a predicted outcome on decision-maker identity. What can this test not detect?
  4. Contraction counts observed failures in the cases a policy releases. If outcomes themselves are under-reported (some failures recorded as clean), when does the comparison of two policies stay fair?

Sources