Does Your Neural Network Extrapolate? Feature Engineering as Identifiability Bias for OOD Generalization
Not on the map. Why?
Abstract (arXiv)
Successful deep neural networks discover salient features of data. We show when and why they fail to learn out-of-distribution (OOD)-relevant representations from an in-distribution (ID) training window. This requires decoupling feature learning from data-generating-process (DGP) identifiability. From a single training window, OOD extrapolation is non-identifiable: infinitely many DGPs are varepsilon-observationally equivalent on the training data but diverge arbitrarily outside it, and no in-distribution criterion alone reliably breaks the tie. A structural commitment, the feature map, label map, and model class (varphi, psi, M), dictates the assumed DGP and governs OOD generalization while leaving ID performance essentially unchanged. When architecture, pretraining, augmentation, input formats, or domain knowledge implicitly inject the missing commitment, the model succeeds. When it cannot infer OOD-relevant structure from ID evidence, it fails. Changing only the representation can make the same architecture, at the same in-distribution loss, differ by ~520× out of distribution. When the commitment is correct and identifiable, OOD error vanishes. For example, Fourier coordinates turn periodic extrapolation into interpolation on mathbbS1. The same mechanism predicts outcomes in three natural-science settings (mass-action chemistry; Kepler's-third-law exoplanet prediction, n=2,362; and cross-species coding-DNA detection) and in a 264-run positional-encoding study across Transformer, Mamba, and S4D. Finally, a controlled study shows: correct features are necessary but not sufficient. The model class must express the target, and the transformed training data must cover the relevant representation space.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 10 of its 17 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
10 of 17 references.
References · 10
- 2017Attention Is All You Needcited by 60,361
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2019Invariant Risk Minimizationcited by 1,854
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
- 2023A decoder-only foundation model for time-series forecastingcited by 692
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
- 2020Neural Networks Fail to Learn Periodic Functions and How to Fix Itcited by 100
- 2022Exploring the Whole Rashomon Set of Sparse Decision Treescited by 81
- 2022Modeling the Data-Generating Process is Necessary for Out-of-Distribution Generalizationcited by 34
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2024Beyond Finite Data: Towards Data-free Out-of-distribution Generalization via Extrapolation
- 2022DORA: Exploring Outlier Representations in Deep Neural Networks
- 2024Out-of-distribution generalization via composition: a lens through induction heads in Transformers
- 2025Principled Out-of-Distribution Generalization via Simplicity
- 2026Deep Networks Favor Simple Data
- 2026Biased Generalization in Diffusion Models
- 2023On information captured by neural networks: connections with memorization and generalization
- 2024A Geometric Explanation of the Likelihood OOD Detection Paradox