ICML2026
Black-Box Assisted Regression: Phase Transitions and Minimax Optimality
Yan Zhou
摘要
Foundation models are often used as fixed black-box predictors for downstream tasks with limited labeled data, but their predictions may be biased and unsafe to trust blindly. We study this setting through black-box assisted nonparametric regression: a learner observes labeled samples and can query a fixed predictor , while the target is close to in up to an unknown radius . We give a finite-sample minimax characterization showing a phase transition at , with leading risk . We then analyze a Safe Residual Estimator: it learns a correction around , initializes the residual head at zero so the initial predictor equals , and uses holdout selection to revert to when the learned correction is not supported by validation data. Here, ``safe'' means avoiding negative transfer, i.e., performing worse than the black-box predictor alone. The estimator matches the leading minimax term up to an additive validation-selection cost. Synthetic regression experiments verify the predicted phase transition, while CIFAR-100 with CLIP and AG News with Qwen3-8B provide practice-facing evidence that the same residual-correction tradeoff is useful beyond the formal squared-loss regression setting.