Search papers, labs, and topics across Lattice.
The paper introduces the Inverse Neural Operator (INO), a two-stage framework leveraging a Conditional Fourier Neural Operator (C-FNO) for trajectory reconstruction and an Amortized Drifting Model (ADM) for parameter estimation in ODEs. INO avoids Jacobian instabilities associated with gradient-based methods by learning a kernel-weighted velocity field in parameter space, guiding parameter updates without backpropagation through the surrogate model. Results on stiff atmospheric chemistry and gene regulatory network benchmarks demonstrate INO's superior parameter recovery accuracy and a 487x speedup compared to iterative gradient descent.
Forget slow, unstable gradient descent for ODE parameter fitting: a neural operator learns to directly "drift" parameters towards the ground truth with a 487x speedup.
We propose the Inverse Neural Operator (INO), a two-stage framework for recovering hidden ODE parameters from sparse, partial observations. In Stage 1, a Conditional Fourier Neural Operator (C-FNO) with cross-attention learns a differentiable surrogate that reconstructs full ODE trajectories from arbitrary sparse inputs, suppressing high-frequency artifacts via spectral regularization. In Stage 2, an Amortized Drifting Model (ADM) learns a kernel-weighted velocity field in parameter space, transporting random parameter initializations toward the ground truth without backpropagating through the surrogate, avoiding the Jacobian instabilities that afflict gradient-based inversion in stiff regimes. Experiments on a real-world stiff atmospheric chemistry benchmark (POLLU, 25 parameters) and a synthetic Gene Regulatory Network (GRN, 40 parameters) show that INO outperforms gradient-based and amortized baselines in parameter recovery accuracy while requiring only 0.23s inference time, a 487x speedup over iterative gradient descent.
