Into the danger zone: stable extrapolation in high-dimensional function and operator learning
Not on the map. Why?
Abstract (arXiv)
Out-of-distribution (OOD) generalization is a central challenge in scientific machine learning. We study regression problems in which the test distribution differs from the training distribution and ask: under what assumptions on the target function or operator is stable extrapolation possible, and how far beyond the training domain can one extrapolate? Existing theory controls the test error through additive penalties measuring the discrepancy between the training and test distributions. Such guarantees show robustness to small distribution shifts, but can be very pessimistic in comparison to OOD performance observed empirically. We identify classes of holomorphic functions and operators for which the OOD generalization error converges at algebraic rates even in the presence of large distribution shifts. This phenomenon stems from the increasing smoothness of higher-index coordinates, leading to what we term a `blessing of high dimensionality'. For learning with either polynomials, deep neural networks or deep neural operators, we derive explicit rates for arbitrary test measures supported on suitable domains and quantify how the admissible domain depends on the underlying regularity of the function or operator. Our extrapolation guarantees are independent of the test distribution, depending only on its support. We also present a series of numerical experiments across a range of functions and operators that support the main theoretical findings.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 202 references that are in the map. First posted on arXiv 29 Sep 2026.
Builds on
19 of 202 references.
References · 19
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2021Out of Distribution Generalization in Machine Learningcited by 60
- 2023The Parametric Complexity of Operator Learningcited by 28
- 2025Defining Foundation Models for Computational Science: A Call for Clarity and Rigorcited by 17
- 2024The Sample Complexity of Learning Lipschitz Operators with respect to Gaussian Measurescited by 11
- 2024Bridging the Gap Between Approximation and Learning via Optimal Approximation by ReLU MLPs of Maximal Regularitycited by 10
- 2026Test-time Generalization for Physics through Neural Operator Splittingcited by 10
- 2023Understanding and Mitigating Extrapolation Failures in Physics-Informed Neural Networkscited by 9
- 2025SIMSHIFT: A Benchmark for Adapting Neural Surrogates to Distribution Shiftscited by 7
- 2026Performance of Neural and Polynomial Operator Surrogatescited by 7
- 2026Diagnosing Failure Modes of Neural Operators Across Diverse PDE Familiescited by 6
- 2025Optimal Recovery Meets Minimax Estimationcited by 4
7 more
- 2026A short tour of operator learning theory: Convergence rates, statistical limits, and open questionscited by 4
- 2025Learning where to learn: Training data distribution optimization for scientific machine learningcited by 3
- 2026Out-of-distribution generalization of deep-learning surrogates for 2D PDE-generated dynamics in the small-data regimecited by 3
- 2026Generalization Guarantees for Multi-Input Neural Operator Learning in Sobolev Spacescited by 3
- 2026Do Physics Foundation Models Learn Generalizable Physics? A Bias-Aware Benchmark Across Physical Regimes and Distribution Shiftscited by 1
- 2026Is Zero-Shot Super-Resolution Possible in Operator Learning?cited by 1
- 2026Algorithmic Foundations of Deep Learning: Complexity-Theoretic Rates and a Characterization of Universal Approximationcited by 1
Built on by
Not cited yet.