mlsanityData to 7 Oct 2026

FIRE: Multi-fidelity Regression with Distribution-conditioned In-context Learning using Tabular Foundation Models

arXiv 2601.22371 ↗First posted 29 Jan 2026cs.LGcited by 1

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

  1. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 643
  2. 2021Transformers Can Do Bayesian Inferencecited by 338
  3. 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 267
  4. 2018When Gaussian Process Meets Big Data: A Review of Scalable GPscited by 205
  5. 2025TabArena: A Living Benchmark for Machine Learning on Tabular Datacited by 152
  6. 2025TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Modelscited by 115
  7. 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
  8. 2019Practical Multi-fidelity Bayesian Optimization for Hyperparameter Tuningcited by 91
  9. 2019Distribution Calibration for Regressioncited by 88
  10. 2016Multi-Information Source Optimizationcited by 83
  11. 2025Mitra: Mixed Synthetic Priors for Enhancing Tabular Foundation Modelscited by 67
  12. 2023Statistical Foundations of Prior-Data Fitted Networkscited by 59
17 more
  1. 2019Deep Gaussian Processes for Multi-fidelity Modelingcited by 48
  2. 2017Using stacking to average Bayesian predictive distributionscited by 39
  3. 2025A Closer Look at TabPFN v2: Understanding Its Strengths and Extending Its Capabilitiescited by 38
  4. 2025Real-TabPFN: Improving Tabular Foundation Models via Continued Pre-training With Real-World Datacited by 36
  5. 2024DrivAerNet: A Parametric Car Dataset for Data-Driven Aerodynamic Design and Predictioncited by 33
  6. 2024In-Context Freeze-Thaw Bayesian Optimization for Hyperparameter Optimizationcited by 26
  7. 2023GP+: A Python Library for Kernel-based learning via Gaussian Processescited by 17
  8. 2025BlendedNet: A Blended Wing Body Aircraft Dataset and Surrogate Model for Aerodynamic Predictionscited by 16
  9. 2024Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modelingcited by 13
  10. 2021Residual Gaussian Process: A Tractable Nonparametric Bayesian Emulator for Multi-fidelity Simulationscited by 8
  11. 2024Multi-fidelity Gaussian process surrogate modeling for regression problems in physicscited by 4
  12. 2024Practical multi-fidelity machine learning: fusion of deterministic and Bayesian modelscited by 4
  13. 2023GAR: Generalized Autoregression for Multi-Fidelity Fusioncited by 3
  14. 2024Multifidelity linear regression for scientific machine learning from scarce datacited by 3
  15. 2024Multifidelity Surrogate Models: A New Data Fusion Perspectivecited by 2
  16. 2025Cost-Sensitive Freeze-thaw Bayesian Optimization for Efficient Hyperparameter Tuningcited by 2
  17. 2024Warped multifidelity Gaussian processes for data fusion of skewed environmental datacited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.