Learning the Pareto Frontier of Predictive Models under Distribution Shift
Not on the map. Why?
Abstract (arXiv)
Modern machine learning pipelines increasingly rely on reusing pretrained and foundation models across downstream tasks. These pretrained models can differ not only in performance but also in how they can be used: some only provide black-box predictions, while others may permit white-box access to internal representations that can be probed or fine-tuned. When deployed to the target domain in the presence of distribution shift, no single strategy, including zero-shot application, fine-tuning, or directly training a target-specific model, is uniformly the best. In this work, we propose Frontier Learning, a framework that treats a library of candidate models spanning different training histories and access regimes as complementary sources of information rather than mutually exclusive alternatives. Frontier Learning constructs a unified target-domain feature by concatenating internal representations from white-box candidates as well as prediction outputs from black-box candidates, then fits a lightweight, regularized supervised learner on this concatenated representation using labeled target data. Because the resulting hypothesis class contains predictors obtained by zero-shot reuse, fine-tuning, and direct training as special cases, empirical risk minimization over the frontier learner is guaranteed to be no worse, on the training sample, than any individual baseline. We evaluate the framework in simulations spanning varying degrees of source-target compatibility and in two real-world distribution-shift settings: visual domain adaptation on DomainNet/VisDA and clinical mortality prediction across intensive care unit domains using MIMIC-IV-Notes. Across all settings, Frontier Learning matches or outperforms the strongest individual reuse strategy, with the largest gains arising precisely when no single baseline is reliable across the range of shift considered.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 28 references that are in the map. First posted on arXiv 1 Aug 2026.
Builds on
19 of 28 references.
References · 19
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
- 2023Qwen Technical Reportcited by 3,273
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2022Large Language Models Encode Clinical Knowledgecited by 1,807
- 2018Moment Matching for Multi-Source Domain Adaptationcited by 1,327
7 more
- 2019Publicly Available Clinical BERT Embeddingscited by 995
- 2020Do We Really Need to Access the Source Data? Source Hypothesis Transfer for Unsupervised Domain Adaptationcited by 844
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 809
- 2023Generating with Confidence: Uncertainty Quantification for Black-box Large Language Modelscited by 286
- 2022Black-Box Tuning for Language-Model-as-a-Servicecited by 254
- 2023Prediction-Powered Inferencecited by 214
- 2020Transferable Calibration with Lower Bias and Variance in Domain Adaptationcited by 41
Built on by
Not cited yet.