Black-Box Assisted Regression: Phase Transitions and Minimax Optimality
Yan Zhou
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on11
- FixMatch: Simplifying Semi-Supervised Learning with Consistency and ConfidenceKihyuk Sohn, David Berthelot, Nicholas Carlini, Zizhao Zhang et al.NeurIPS 2020 · 5,129 citations
- Tent: Fully Test-Time Adaptation by Entropy MinimizationDequan Wang, Evan Shelhamer, Shaoteng Liu, Bruno A. Olshausen et al.ICLR 2021 · 1,731 citations
- Do We Really Need to Access the Source Data? Source Hypothesis Transfer for Unsupervised Domain AdaptationJian Liang, Dapeng Hu, Jiashi FengICML 2020 · 1,624 citations
- Continual Test-Time Domain AdaptationQin Wang, Olga Fink, Luc Van Gool, Dengxin DaiCVPR 2022 · 383 citations
- A Theory of Label Propagation for Subpopulation ShiftTianle Cai, Ruiqi Gao, Jason D. Lee, Qi LeiICML 2021 · 55 citations
Related papers
- Robust Nonparametric Regression under Poisoning AttackPuning Zhao, Zhiguo WanAAAI 2024 · 13 citations
- FIRE: Multi-fidelity Regression with Distribution-conditioned In-context Learning using Tabular Foundation ModelsRosen Yu, Nicholas Sung, Faez AhmedICML 2026 · 1 citation
- Residual Feature Integration is Sufficient to Prevent Negative TransferYichen Xu, Ryumei Nakada, Linjun Zhang, Lexin LiICLR 2026 · 1 citation
- Conformal Prediction using Conditional HistogramsMatteo Sesia, Yaniv RomanoNeurIPS 2021 · 106 citations
- Minimax Lower Bounds for Transfer Learning with Linear and One-hidden Layer Neural NetworksSeyed Mohammadreza Mousavi Kalan, Zalan Fabian, Salman Avestimehr, Mahdi SoltanolkotabiNeurIPS 2020 · 37 citations
