The Label Complexity of Class-Conditional Coverage under Distribution Shift

2026-07-20Machine Learning

Machine LearningComputer Vision and Pattern Recognition
AI summary

The authors studied how well recognition systems handle changes between training and test data, a problem called distribution shift. They found that while overall accuracy might look good, accuracy for individual categories can be much worse without warning. They show that fixing this problem needs some labeled examples from the new data, with more labels needed for better accuracy across classes. Their experiments on skeleton action recognition and image corruption confirm these findings, showing limits to how much unlabeled data can help. Overall, they highlight the challenges and conditions to improve fairness across categories under distribution shifts.

distribution shiftsplit conformal predictionmarginal coverageper-class coveragecalibrationskeleton action recognitionlabel efficiencycovariatespseudo-labelstarget domain adaptation
Authors
Weijia Han, Lisha Qu
Abstract
Standard evaluation of many recognition systems contains distribution shift by construction, since benchmarks place disjoint conditions in the training and test splits. Under such a shift, split conformal prediction keeps marginal coverage near the nominal level while per-class coverage fails silently: on a real cross-subject skeleton benchmark, marginal coverage stays near ninety percent, the worst action class is covered about seventy percent of the time, and ten of the sixty classes fall below eighty percent coverage. We characterize the cost of restoring per-class validity. First, an impossibility: once the shift acts jointly on the covariates and the labels, the target class-conditional score law is unidentified from source labels and an unlabeled target sample, so no label-free method attains per-class coverage that is at once valid and efficient. Second, we make the cost precise: per-class validity alone needs only a handful of target labels per class, while the label count necessary and sufficient for validity together with per-class efficiency grows as the inverse square of the efficiency tolerance and the logarithm of the number of classes, with matching upper and lower bounds. Third, within the evaluated prediction-powered inference family, even the most favorable use of the classifier's own pseudo-labels on an unbounded unlabeled target pool improves efficiency by at most a small constant factor where coverage collapses. Skeleton action recognition is our real-data case study. A per-class calibration using source labels alone recovers a substantial share of the per-class gap while the shift preserves marginal coverage, and stops helping exactly when marginal coverage itself breaks. Three real shifts of increasing severity trace this boundary, and the same collapse and recovery appears on a natural-image corruption benchmark, beyond any single modality.