OmniPred: Language Models as Universal Regressors
Authors: name: Xingyou Song; name: Oscar Li; name: Chansoo Lee; name: Bangding; name: Yang; name: Daiyi Peng; name: Sagi Perel; name: Yutian Chen
Published: 2024-02-22 13:36:53

Abstract: Over the broad landscape of experimental design, regression has been a
powerful tool to accurately predict the outcome metrics of a system or model
given a set of parameters, but has been traditionally restricted to methods
which are only applicable to a specific task. In this paper, we propose
OmniPred, a framework for training language models as universal end-to-end
regressors over (x,y) evaluation data from diverse real world experiments.
Using data sourced from Google Vizier, one of the largest blackbox optimization
databases in the world, our extensive experiments demonstrate that through only
textual representations of mathematical parameters and values, language models
are capable of very precise numerical regression, and if given the opportunity
to train over multiple tasks, can significantly outperform traditional
regression models.