FIRE: Multi-fidelity Regression with Distribution-conditioned In-context Learning using Tabular Foundation Models
Not on the map. Why?
Abstract (arXiv)
Multi-fidelity (MF) regression often operates in regimes of extreme data imbalance, where the commonly-used Gaussian-process (GP) surrogates struggle with cubic scaling costs and overfit to sparse high-fidelity observations, limiting efficiency and generalization in real-world applications. We introduce FIRE, a training-free MF framework that couples tabular foundation models (TFMs) to perform zero-shot in-context Bayesian inference via a high-fidelity correction model conditioned on the low-fidelity model's posterior predictive distributions. This cross-fidelity information transfer via distributional summaries captures heteroscedastic errors, enabling robust residual learning without model retraining. Across 31 benchmark problems spanning synthetic and real-world tasks (e.g., DrivAerNet, LCBench), FIRE delivers a stronger performance-time trade-off than seven state-of-the-art GP-based or deep learning MF regression methods, ranking highest in accuracy and uncertainty quantification with runtime advantages. Limitations include context window constraints and dependence on the quality of the pre-trained TFM's.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 29 of its 68 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
29 of 68 references.
References · 29
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 643
- 2021Transformers Can Do Bayesian Inferencecited by 338
- 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 267
- 2018When Gaussian Process Meets Big Data: A Review of Scalable GPscited by 205
- 2025TabArena: A Living Benchmark for Machine Learning on Tabular Datacited by 152
- 2025TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Modelscited by 115
- 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
- 2019Practical Multi-fidelity Bayesian Optimization for Hyperparameter Tuningcited by 91
- 2019Distribution Calibration for Regressioncited by 88
- 2016Multi-Information Source Optimizationcited by 83
- 2025Mitra: Mixed Synthetic Priors for Enhancing Tabular Foundation Modelscited by 67
- 2023Statistical Foundations of Prior-Data Fitted Networkscited by 59
17 more
- 2019Deep Gaussian Processes for Multi-fidelity Modelingcited by 48
- 2017Using stacking to average Bayesian predictive distributionscited by 39
- 2025A Closer Look at TabPFN v2: Understanding Its Strengths and Extending Its Capabilitiescited by 38
- 2025Real-TabPFN: Improving Tabular Foundation Models via Continued Pre-training With Real-World Datacited by 36
- 2024DrivAerNet: A Parametric Car Dataset for Data-Driven Aerodynamic Design and Predictioncited by 33
- 2024In-Context Freeze-Thaw Bayesian Optimization for Hyperparameter Optimizationcited by 26
- 2023GP+: A Python Library for Kernel-based learning via Gaussian Processescited by 17
- 2025BlendedNet: A Blended Wing Body Aircraft Dataset and Surrogate Model for Aerodynamic Predictionscited by 16
- 2024Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modelingcited by 13
- 2021Residual Gaussian Process: A Tractable Nonparametric Bayesian Emulator for Multi-fidelity Simulationscited by 8
- 2024Multi-fidelity Gaussian process surrogate modeling for regression problems in physicscited by 4
- 2024Practical multi-fidelity machine learning: fusion of deterministic and Bayesian modelscited by 4
- 2023GAR: Generalized Autoregression for Multi-Fidelity Fusioncited by 3
- 2024Multifidelity linear regression for scientific machine learning from scarce datacited by 3
- 2024Multifidelity Surrogate Models: A New Data Fusion Perspectivecited by 2
- 2025Cost-Sensitive Freeze-thaw Bayesian Optimization for Efficient Hyperparameter Tuningcited by 2
- 2024Warped multifidelity Gaussian processes for data fusion of skewed environmental datacited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.