REVIEW 4 major objections 6 minor 1 cited by
A compact pretrained model infers ODE vector fields from noisy trajectories in one pass, outpacing a larger symbolic baseline on reconstruction.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · deepseek-v4-flash
2026-08-03 03:11 UTC pith:UM4NGEGV
load-bearing objection FIM-ODE offers a plausible small-model alternative to symbolic amortized ODE inference, but the headline numbers mix denominators and the abstract oversells; it deserves a serious referee, not a desk reject. the 4 major comments →
Foundation Inference Models for Ordinary Differential Equations
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The paper's central claim is that vector-field inference for ODEs can be amortised: a model pretrained on sparse random polynomial vector fields (degree at most three, dimensions one to three) learns a map from noisy trajectory context to a continuous local estimate of the underlying field at any query point. On ODEBench trajectory reconstruction, this estimator achieves variance-weighted R-squared greater than 0.9 in 84.4% of systems at zero noise, compared to 63.1% for a global symbolic baseline, and it remains competitive on generalization to new initial conditions. Pretraining also serves as an initialisation that, after brief finetuning, outperforms per-dataset neural and Gaussian-proce
What carries the argument
The load-bearing component is the transition-based input representation: each consecutive observation pair is encoded as the current state, the displacement, the element-wise squared displacement, and the inter-observation time gap. Together these act as a finite-difference estimate of the local vector field plus a second-moment feature capturing noise and local scale. A transformer encoder forms a permutation-invariant context; a functional attention decoder queries this context at arbitrary spatial locations, producing a continuous neural-operator field estimate. An auxiliary head predicts a per-location heteroscedastic log-variance that weights the Laplace likelihood loss, down-weighting
Load-bearing premise
The vector field in the region visited by the data is determined well enough by local transition statistics—state, displacement, squared displacement, and time gap—that a model trained on polynomial fields transfers to non-polynomial systems in that same region.
What would settle it
Take a conservative system (e.g., an undamped pendulum) and a single noisy trajectory; run FIM-ODE and evaluate the inferred vector field on a grid spanning unvisited regions. If the field error grows sharply away from the trajectory and closed orbits are not preserved, the locality premise fails, as the paper already notes for the pendulum.
If this is right
- In data-rich regimes, a local estimator can beat symbolic regression on trajectory reconstruction even when the system's functional form lies outside the training prior, as shown on a rational reaction-network model.
- Pretraining on simple polynomial ODEs transfers to non-polynomial systems, including trigonometric and rational vector fields, when the relevant behavior is controlled by local features in the visited region.
- A brief finetuning phase turns the pretrained model into a strong per-dataset estimator, outperforming modern neural and GP baselines without requiring derivative estimation or solver-based training.
- Performance on low-data benchmarks is strongly sensitive to the noise realisation, so single-seed evaluations can misrepresent zero-shot capability; the paper reports large variance across 100 seeds.
- The local representation is not universal: for conservative systems such as the frictionless pendulum, the model fails to preserve closed-orbit geometry and generalises poorly to unseen initial conditions.
Where Pith is reading between the lines
- The local representation suggests an evaluation protocol that reports accuracy only on the data-visited region, since global claims about unvisited state-space are exactly where the model fails; the paper's own pendulum analysis supports this boundary.
- The transition features resemble a second-order expansion of the flow; a testable extension is whether adding higher-order or memory-dependent features would capture conservative geometry without a richer prior.
- Because the model can condition on multiple trajectories, it could serve as a fast initialiser for classical system-identification pipelines, proposing a good starting field that a local refinement step then corrects.
- The authors' proposed stationary-GP prior is a natural next test: it may preserve stability while fixing the sink bias that the polynomial prior induces, and its effect on conservative systems is directly measurable with the pendulum benchmark.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper introduces FIM-ODE, a pretrained transformer-based neural operator for amortized inference of low-dimensional ODE vector fields from noisy, irregularly sampled trajectories. It is pretrained on a synthetic corpus of 600K polynomial ODE systems (degree ≤ 3, dimensions 1–3) and evaluated zero-shot on ODEBench against ODEFormer, on low-data VDP/FHN benchmarks against classical neural/GP baselines, and on CMU MoCap human-motion data. The central claims are that FIM-ODE's local representation matches or improves upon the global symbolic ODEFormer despite a simpler prior and much smaller model/data scale, and that pretraining provides a strong initialization for finetuning. The paper includes candid limitations and extensive appendices, but several reporting choices (excluded systems, selected-noise results, and contradictory table/text statements) undermine the headline claims as currently written.
Significance. If the empirical claims are substantiated, the work is significant: it shows that a relatively small (8M-parameter) local neural-operator estimator, pretrained on a simple polynomial prior, can amortize ODE inference competitively against an 86M-parameter symbolic transformer trained on 50M systems. The architecture is plausible, the transition-feature representation and uncertainty weighting are sensible, and the paper includes useful ablations on dataset size, discretization, and noise-seed sensitivity. However, the central comparison is currently not fully reliable because of a possible denominator mismatch in Table 1, and the paper's own tables contradict several textual claims about finetuning and zero-shot performance. These issues are fixable but require substantial revision and re-analysis.
major comments (4)
- [§5.1, Table 1, footnote 2] FIM-ODE is evaluated on 59 ODEBench systems after excluding two systems with dimension > 3, but the ODEformer percentages in Table 1 are not stated to be recomputed on the same 59-system subset. If the ODEformer numbers are taken from d'Ascoli et al. (2024) over all 61 systems, the comparison mixes denominators: 63.1% of 61 is about 38.5 systems, while 84.4% of 59 is about 49.8 systems. Since Table 1 is the main support for the claim of improving upon ODEFormer, the authors must re-evaluate ODEformer on the identical 59-system subset and report per-system results, or clearly state the denominator used for each method.
- [§5.2, Table 4] The text claims that finetuning 'yields the best results among all methods considered,' but Table 4 lists GP-DNF with VDP T1/T2 MSE of 0.03/0.04 while FIM-ODE (Finetuned) has 0.16/0.26. The zero-shot sentence 'FIM-ODE consistently outperforms ODEformer in zero-shot mode' is also contradicted by VDP T1 (FIM-ODE 0.89 vs ODEformer 0.22). Moreover, the 'FIM-ODE (Selected noise)' rows are the best of 100 independent trials (Appendix E.2), not representative zero-shot performance; Table 7 reports mean/median VDP T1 MSE of 0.575/0.340. These claims and rows must be corrected, removed, or replaced with distributional statistics.
- [Abstract and §5.1] The abstract's claim of 'matching and often improving upon ODEFormer ... across a range of regimes' is not supported by the tables. Table 2 shows no consistent advantage in generalization (e.g., at ρ=0.5, σ=0.03, ODEformer is 32.8% vs FIM-ODE 27.0%), and Table 4 shows zero-shot FIM-ODE is substantially worse than ODEformer on VDP T1. The only strong evidence is Table 1 reconstruction, which is subject to the denominator concern above. The claims should be qualified to the reconstruction task and the regimes where FIM-ODE actually wins, with a discussion of observed failures.
- [§5.1, Figure 2, §4.2] The frictionless-pendulum result directly challenges the load-bearing premise of §4.2 that local transition features (state, displacement, squared displacement, time gap) determine the vector field in the data-visited region. For this non-polynomial system, FIM-ODE 'tends to bias the field toward a weak sink near the origin' and 'generalises poorly to unseen initial conditions,' even though it matches the context trajectory. The text nevertheless concludes that 'local estimates can transfer, even when the global functional form is OOD.' This counterexample should be addressed quantitatively, e.g., by reporting ODEBench success rates separately for polynomial and non-polynomial systems, and by tempering the transfer claim.
minor comments (6)
- [Appendix B, References] The reference to 'Hoff, J. H. v. t. Studies in Chemical Dynamics' gives the year 1986; the cited work is from 1884. Also 'V olterra' is misspelled in the text and bibliography.
- [Appendix A.1] The phrase 'any adhering initial value problem' should be 'any admissible initial value problem' or similar; 'adhering' is not standard terminology.
- [Appendix D] The sentence 'After nsteps steps ˆxi' has a redundant word. Also, the description of how initial conditions are synchronized with the observed time grid would benefit from a small diagram or explicit pseudocode.
- [Table 6] The text says FIM-ODE shows a 'clearer edge' at the R2 > 0.8 threshold, but at ρ=0.5, σ=0.03 ODEformer achieves 41.0% vs FIM-ODE 40.2%; the description should acknowledge this exception.
- [Figure 2 caption] The caption says 'clean context trajectory' although the setting uses σ=0.03 noise and ρ=0.5 subsampling. Please rephrase to 'corrupted context trajectory' or similar.
- [General] The notation is sometimes confusing: y is used both for corrupted observations and as a state variable in parts of the appendix. A brief notation table or consistent subscript convention would improve readability.
Circularity Check
No circularity: FIM-ODE's central claims rest on external zero-shot benchmark evaluation, not on self-citations or fitted constants.
full rationale
Walking the derivation chain: the pretraining prior p(f) is a distribution over polynomial ODEs with ground-truth vector fields f(x); trajectories are simulated and corrupted; the model is trained by matching predicted vector fields to these ground-truth values at query locations (Section 4.2, loss Lθ). The ODEBench results in Tables 1 and 2 are zero-shot evaluations against an external baseline (ODEformer) and external per-dataset methods, so the headline quantitative claims are not derived from fitted constants or from the paper's own equations. The self-citations to Seifner et al. (2025a) — 'This architecture is adapted from Seifner et al. (2025a)' and 'We therefore follow Seifner et al. (2025a) and use uncertainty weighting' — are architectural provenance, not load-bearing justification; no uniqueness theorem or ansatz is imported from the authors' prior work to force the result. The GP interpretation in Section 4.1 is explicitly descriptive and does not feed into the loss or evaluation. The paper's own limitation statement in Section 5.1 ('FIM-ODE tends to bias the field toward a weak sink near the origin ... fails to preserve the correct conservative geometry') weakens the generalization claim but is an honest empirical failure mode, not a circular step. The possible ODEBench system-set mismatch flagged by the skeptic (59 vs 61 systems) is an evaluation-validity concern, not a circularity of the derivation; no equation or fitted parameter reduces to the prediction by construction. Hence: no significant circularity.
Axiom & Free-Parameter Ledger
free parameters (7)
- Pretraining prior: max polynomial degree p=3 =
3
- Coefficient scale and global scaling =
cA~N(0,1), s~Uniform[0,2]
- Trajectory rejection threshold =
δ_reject=100
- Corruption ranges =
σ~Uniform[0,0.06], ρ~Uniform[0,0.5]
- Normalization target time-gap =
Δτ_target=0.01
- Low-data zero-shot noise seed =
best of 100 trials (selected row)
- Finetuning hyperparameters =
nsteps=25, nIC=⌊2ℓ/25⌋, σ_i=Δt/5, 200-800 epochs
axioms (6)
- standard math Picard–Lindelöf theorem guarantees unique local solutions for locally Lipschitz polynomial vector fields.
- domain assumption Target observations y are noisy state measurements of an autonomous first-order ODE; multiplicative noise plus Bernoulli subsampling is a faithful corruption model.
- domain assumption The vector field is locally identifiable from transition features (state, Δy/Δτ, Δy², Δτ) in the visited region, and this local estimate transfers OOD.
- domain assumption Reversible instance normalization (per-dimension zero mean/unit variance) and delta-log time centering preserve enough information to predict vector fields; denormalization by the chain rule is exact.
- domain assumption Variance-weighted R² > 0.9 on integrated trajectories is a valid measure of vector-field inference quality.
- domain assumption Neural-ODE finetuning on context trajectories with multiple shooting initial conditions transfers to held-out initial conditions and subjects.
Cite this review
Pith. "Pith review of Foundation Inference Models for Ordinary Differential Equations." pith.science (2026). https://pith.science/paper/UM4NGEGV
@misc{pith2026260208733,
author = {Pith},
title = {Pith review of: Foundation Inference Models for Ordinary Differential Equations},
year = {2026},
howpublished = {\url{https://pith.science/paper/UM4NGEGV}},
note = {Machine review of arXiv:2602.08733}
}
read the original abstract
Ordinary differential equations (ODEs) are central to scientific modelling, but inferring their vector fields from noisy trajectories remains challenging. Current approaches such as symbolic regression, Gaussian process (GP) regression, and Neural ODEs often require complex training pipelines and substantial machine learning expertise, or they depend strongly on system-specific prior knowledge. We propose FIM-ODE, a pretrained Foundation Inference Model that amortises low-dimensional ODE inference by predicting the vector field directly from noisy trajectory data in a single forward pass. We pretrain FIM-ODE on a prior distribution over ODEs with low-degree polynomial vector fields and represent the target field with neural operators. FIM-ODE achieves strong zero-shot performance, matching and often improving upon ODEFormer, a recent pretrained symbolic baseline, across a range of regimes despite using a simpler pretraining prior distribution. Pretraining also provides a strong initialisation for finetuning, enabling fast and stable adaptation that outperforms modern neural and GP baselines without requiring machine learning expertise.
Figures
Forward citations
Cited by 1 Pith paper
-
HyperODE: Zero-Shot Surrogate for Simulation and Inference of Dynamical Systems
A single 13k-parameter hypergraph surrogate trained on simple mass-conserving ODEs predicts calibrated trajectory quantiles and infers parameters across unseen compartmental families, system sizes, interaction orders,...
Reference graph
Works this paper leans on
-
[2]
URL https://openreview.net/forum? id=TzoHLiGVMo. Delahunt, C. B. and Kutz, J. N. A toolkit for data-driven discovery of governing equations in high-noise regimes. IEEE Access, 10:31210–31234, 2022. 9 Foundation Inference Models for ODEs Dooley, S., Khurana, G. S., Mohapatra, C., Naidu, S. V ., and White, C. Forecastpfn: Synthetically-trained zero-shot for...
arXiv 2022
-
[2022]
Newton, I.Philosophiae Naturalis Principia Mathematica
URL https://openreview.net/forum? id=KSugKcbNf9. Newton, I.Philosophiae Naturalis Principia Mathematica. cf. especially Liber de Motu I Sects. II, III, VII. London, 1687. Pal, A., Ma, Y ., Shah, V ., and Rackauckas, C. V . Opening the blackbox: Accelerating neural differential equations by regularizing internal solver heuristics. InInterna- tional Confere...
arXiv 2021
-
[2024]
Berghaus, D., Seifner, P., Cvejoski, K., Ojeda, C., and S´anchez, R
URL https://openreview.net/forum? id=f4v7cmm5sC. Berghaus, D., Seifner, P., Cvejoski, K., Ojeda, C., and S´anchez, R. J. In-context learning of temporal point pro- cesses with foundation inference models.arXiv preprint arXiv:2509.24762, 2025. Bernoulli, J.Extrait de la R ´eponse de M. Bernoulli `a M. Herman, dat´ee de Basle le 7. Octobre 1710.M ´emoires d...
Pith/arXiv arXiv 2025
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.