Context. In rare-positive problems (clicks, fraud, disease) the information sits mostly in the positives while the compute scales with the total. Epidemiology's answer, case-control sampling, keeps all cases and a few controls, fits logistic regression, and subtracts the log sampling ratio from the intercept (Anderson, 1972; Prentice and Pyke, 1979). The intercept fix is exact only for a correct logistic model, since under misspecification the slopes change too, and the scheme ignores conditional imbalance, keeping an obvious control at the same rate as one that looks like a case. Fithian and Hastie keep each point with probability equal to how surprising its label is under a pilot model, . Every kept label is then close to a fair coin given , the most informative state of a Bernoulli, and an offset undoes the tilt exactly; it is the logit shift of Elkan and Saerens et al., now depending on .
Setting. are i.i.d., includes the intercept, and the logistic model may be misspecified. The target is the population risk minimizer Sampling is accept-reject, . Case-control (CC): depends on only; fit to the subsample and subtract from the intercept. Local case-control (LCC): given a pilot with , accept with , fit an ordinary logistic regression to the subsample, and set . For a larger subsample, accept with probability and weight accepted points by .
Results. (i) Under misspecification the limit of CC depends on the sampling ratio and generally differs from . (ii) Consistency. If , the classes are not separable by a hyperplane, and the pilot in probability, possibly depending on the data, then in probability. (iii) Distribution. If and the pilot is independent of the data with , then where and are the Hessian and score covariance of the logistic loss under the subsampled law, is the cross-derivative of the score under that law in the pilot, and is the acceptance rate. (iv) Efficiency. If the model is correct and the pilot consistent and independent, , with the full-sample MLE's covariance, from a subsample of expected size ; with the factor , the covariance is at most , for a subsample about times larger under strong imbalance.
Label-dependent sampling shifts the log-odds by the log acceptance ratio. Bayes on the accepted points gives For CC the shift is a constant , which the intercept correction removes. For LCC it is , so the subsample's log-odds are close to 0 everywhere when the pilot is good. If the model is correct, both corrections are exact; the methods differ only under misspecification.
Under misspecification, shifting the logit changes what the fit matches. The population score makes the best linear fit get probabilities right on average, so it fits most closely where changes with , near , and nearly ignores regions where . A constant shift changes which regions sit near 1/2, so the CC limit moves with . In the oatmeal example (family history in 10% of people, oatmeal in 50% independently; true log-odds without history and with it, for no oatmeal and oatmeal), the population oatmeal coefficient is 1.4, because only the high-risk column has probabilities that matter. Equal case-control sampling raises every cell by the same amount, the no-history column starts to count, and the coefficient becomes : the sign flips (we reproduced 1.39 and on the four cells). Weighted CC restores consistency at a cost in variance.
With a perfect pilot, the subsample's score is half the population score. Take pilot and subsample parameter 0, where every fitted probability is 1/2. A positive is kept with probability and a negative with probability , so Hence is the LCC target whatever the true : sampling stands in for the weights the full-sample score would have applied. CC is LCC with an intercept-only pilot, which is not under misspecification, hence its inconsistency.
A consistent pilot suffices for consistency, through coupling and convexity. The population version of the LCC risk is minimized at by Step 3, so what remains is convergence when the pilot depends on the data. Drive the accept-reject decisions for every pilot by the same uniforms : decisions under and under its limit differ only when falls between two acceptance probabilities at most apart, and Hoeffding conditional on the data and the pilot controls the few flipped points. Bounded gradients and a finite net make the convergence uniform on compacts, strict convexity keeps the minimizer near , and nonseparability makes each sampled risk coercive.
With an independent pilot, the covariance is a sandwich formula plus a term for the pilot's error. Two errors add. Given the pilot, is an M-estimator on about points, so is asymptotically , where is the limit for a fixed pilot. That limit moves with the pilot, and the implicit function theorem on its score equation gives An independent pilot makes the two pieces independent, which gives (iii). Under correct specification for every pilot, so , and to first order the pilot's error only changes which points are kept. All the experiments use a data-dependent pilot, which is outside this theorem.
The factor two follows from counting Fisher information. With a correct model and pilot , each kept point is a fair coin with Fisher information , and a point at is kept with probability , so the expected information per original point is exactly half the full sample's, from points with . For the same count gives and , hence the bound ; costs 20% extra variance for a subsample about three times larger. A small check of ours agrees (, , independent pilot): about 4,350 kept points gave variance ratios to the full MLE of 2.3, 2.2 and 1.9, against 1.5, 3.4 and 3.6 for 1:1 case-control on about 5,170.
In the experiments LCC beats both case-control variants. In a misspecified simulation at 1% positives, with 2,000 points per method, LCC had squared bias 0.0049 against 0.023 (weighted CC) and 0.15 (CC), and variance 0.025 against 0.16 and 0.043. In a correctly specified one with strong separation, and LCC is roughly 100 times more efficient than uniform subsampling. On web spam (60% spam, so marginally balanced but conditionally imbalanced), the predicted variance ratios to the full-sample fit, about 2 for LCC and 5 for CC, held approximately, slightly optimistic.
Consequences and limits. LCC turns imbalance into compute savings without case-control's misspecification bias, recovering half the full-data information from a sample at most twice the size of the positive class. For payment fraud, the paper's suggestion fits a daily retrain: with yesterday's model as the pilot, keep the frauds it scored as safe and the good transactions it scored as risky, and drop the bulk of obvious good traffic. The usual practice of keeping all fraud and a fraction of legitimate traffic is CC, and its correction (add to the fitted log-odds, Elkan's Theorem 2) restores calibration only if the model is right. Since a misspecified fit is most accurate near the level curve where the sampled probability is 1/2, the sampling ratio picks which level curve of the fit gets right, and the paper raises aiming it at the decline threshold as an open direction. The theory covers unpenalized logistic regression in fixed dimension, needs an independent pilot for the variance result, and gives no finite-sample guidance on pilot quality. A stale pilot after drift always lowers the information per kept point and costs consistency under misspecification, since the limit is then .
Terms
- population logistic risk minimizer; the target even when the model is misspecified
- case-control
- accept by label only; correct the intercept by
- local case-control
- accept with ; refit and add the pilot back,
- log-odds shift
- accepted-point logit
- conditional imbalance
- easy to predict from for most ; CC cannot exploit it, LCC can
- acceptance rate
- large-sample limit for a fixed pilot ; equals at
- Corollary 7
- ; pilot error enters through
- Theorem 8
- correct model, consistent independent pilot: covariance
- factor
- accept , weight ; covariance at most
- weighted CC
- inverse-probability weights; consistent but inefficient; used as the pilot
Quiz
- The consistency theorem allows a pilot that depends on the data. Which device in the proof makes this possible?With shared , the decisions under the data-dependent and its limit differ only when falls between two close acceptance probabilities. Hoeffding conditional on the data then bounds the effect without independence.
- Fraud is 0.5% of 10 million transactions and the pilot is good. About how large is the local case-control subsample?The acceptance rate is about . It is smaller when fraud is concentrated in a region the pilot recognizes, since conditional imbalance lowers further.
- With acceptance and weight at , what does the theory give under strong imbalance?The Hessian scales by and the score variance by at most , so the sandwich is at most . Near the keep probability becomes instead of , so the subsample grows by .
Sources
- William Fithian and Trevor Hastie. Local case-control sampling: Efficient subsampling in imbalanced data sets. Annals of Statistics, 2014.
- J. A. Anderson. Separate sample logistic discrimination. Biometrika, 1972.
- R. L. Prentice and R. Pyke. Logistic disease incidence models and case-control studies. Biometrika, 1979.