MiNO: Cotangent-bundle propagator learning for PDEs
Organizations: Axiom Research Group · Department of Applied Mathematics and Computational Science, The Nelson Mandela African Institution of Science and Technology, 404 Nganana, Kikwe, Arumeru, P.O. Box 447, Arusha, 23311, Tanzania · African Institute for Mathematical Sciences, Research and Innovation Centre, KN 3 Road, Gasharu Cell, Kicukiro Sector, Kicukiro District, Kigali, Rwanda · Graduate School of Engineering, The University of Tokyo, 7-3-1 Hongo, Bunkyo-ku, Tokyo, 113-8656, Japan
Abstract
Scientific machine learning for partial differential equations commonly targets solution fields, as in physics-informed neural networks, or solution maps, as in neural operators. We study a third target: the propagator itself, a phase and amplitude in phase space. The motivation is a gap in regularity. A transported discontinuity is nonsmooth in space and time, yet the rule that moves it can be a polynomial phase carrying unit amplitude, so the object that generates an evolution can be far smoother than the field it generates. The microlocal neural operator (MiNO) learns that object, using the eikonal equation for the phase and the transport equation for the amplitude, and recovers the solution by an oscillatory integral. Sharp fronts and caustics then belong to propagation geometry rather than to a field fitted pointwise. Small residuals certify more than the reconstructed field. They place the learned canonical relation, the geometry that carries singularities, close to the exact one, and they separate trainable error from the frequency-truncation tail. On a matched-budget discontinuous-advection benchmark, MiNO stops improving within 10,000 steps at the accuracy limit of its finite reconstruction window, a limit predicted in closed form, whereas a physics-informed neural network with neural-tangent-kernel loss balancing stays near its initial error. On smooth advection, the mean error is for MiNO and for a supervised Fourier neural operator. Single-branch MiNO is the smallest model compared, and one trained generator serves five unseen initial conditions without retraining.
Figures & tables
| Paradigm | Learned object | Space | Training signal | Output |
| PINN | base manifold | PDE residual | solution field | |
| Neural operator | function-space map | supervised pairs | solution map | |
| MiNO | cotangent bundle | eikonal and transport | propagator reconstruction |
| Budget (thousand steps) | MiNO | PINN | PINN-to-MiNO ratio ( ) |
| Paradigm | Method | Parameters | Time per step (ms) |
| Propagator learning | MiNO (single-branch) | 58,658 | |
| MiNO (two-branch wave) | 117,316 | ||
| Field learning | PINN (NTK-balanced) | 99,649 | |
| Causal PINN | 202,764 | ||
| Operator learning | FNO (supervised) | 329,281 | — |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Meaning |
| Phase space and data | |
| Spatial dimension; domain , , or a chart; time interval | |
| Cotangent bundle ; the phase space of the generator | |
| Base point; initial Fourier (fiber) label of the generator | |
| Current covector | |
| Initial datum; solution of the Cauchy problem | |
| Objective | Benchmark | Section |
| Propagation regimes | ||
| Closed-form propagation calibration | smooth advection | Sec. 5.1.1 |
| Sharp-transport regime comparison | discontinuous advection | Sec. 5.1.2 |
| Two-branch propagation | wave equation | Sec. 5.1.3 |
| Inhomogeneous eikonal construction | stationary-point transport | Sec. 5.1.4 |
| Cross-paradigm variable-coefficient comparison | sinusoidal-speed transport | Sec. 5.1.5 |
PDE Operator Learning
Partial differential equation (PDE) operator learning builds models that map inputs such as coefficients and initial conditions to solution fields.
See all 228 papers