Pith. sign in

REVIEW 31 references

Learning High-dimensional Ionic Model Dynamics Using Fourier Neural Operators

T0 review · reviewed 2026-08-07 · deepseek-v4-flash

Pith's one-line read A single Fourier Neural Operator can learn the full state dynamics of stiff ionic models up to 41 variables with roughly 2% relative L2 test error.

arxiv 2505.14039 v1 pith:FU7MZFBM submitted 2025-05-20 cs.LG cs.NAmath.NAstat.ML

classification cs.LGcs.NAmath.NAstat.ML
keywords dynamicsneuralfouriermodelmodelsdynamicalionicnumber
verification ladder T0 review T1 audit T2 compute T3 formal

The pith

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The reading

The paper studies whether Fourier Neural Operators (FNOs), a neural network architecture popular in scientific machine learning, can learn to simulate ionic models: systems of stiff ordinary differential equations that describe how cell membranes generate electrical signals such as action potentials. The authors train a single FNO to map a simple applied current (a rectangular pulse with two parameters, amplitude and duration) to the full time evolution of every state variable in the model. They test three models of increasing size: the 2-variable FitzHugh-Nagumo model, the 4-variable Hodgkin-Huxley model, and the 41-variable O'Hara-Rudy model of a human heart cell.

The results show held-out test errors of about 0.9% for FitzHugh-Nagumo and 2.2 to 2.7% for the Hodgkin-Huxley and O'Hara-Rudy models, with errors spread relatively evenly across variables. The authors also compare two hyperparameter tuning strategies: unconstrained and constrained to about 500,000 parameters, finding similar accuracy but faster convergence for the unconstrained network. They interpret the results as evidence that FNOs scale to high-dimensional stiff systems.

The main caveats are that the input space is only two-dimensional, the ground truth relies on an unspecified stiff ODE solver, and no code or data is currently available. The comparison to other operator learning methods is also missing, so the paper establishes capability more than superiority.

Extended reading notes

Core claim

The central claim is that FNO parameters and relative errors are bounded independently of the state dimension, and that a single FNO can learn the full solution operator of stiff ionic models including the 41-variable O'Hara-Rudy model with about 2% relative L2 test error. If correct, FNOs are a viable surrogate for high-dimensional ionic dynamics.

Load-bearing premise

The applied current input is restricted to a two-parameter family of piecewise constant functions (amplitude i and stimulus duration Tstim), defined in Section 2. The paper's success on this low-dimensional input family is taken as evidence that the learned operator generalizes to arbitrary applied currents; if FNOs fail on richer input waveforms, the central capability claim is overstated.

Share X Bluesky LinkedIn Reddit HN

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, and a circularity audit.

Assumptions & free parameters 0 free parameters · 4 assumptions · 0 invented entities

The paper makes no theoretical derivation and introduces no scientific free parameters; the network weights and hyperparameters are fitted in the standard ML sense. The central claim rests on the correctness of the reference ODE solver, the faithful implementation of the published ionic models, and the extrapolation of scaling behavior from three test models.

assumptions (4)
  • domain assumption System (1) admits a unique solution for the chosen parameter ranges.
    Stated in Section 2 without proof; the FNO is trained on solutions to these ODEs, so existence and uniqueness are needed.
  • domain assumption The O'Hara-Rudy model equations and parameters from [24] are correctly implemented in the data generation.
    The paper refers to the original paper for the full equations and does not reproduce them; correctness of the surrogate depends on the correctness of the reference implementation.
  • domain assumption The reference stiff ODE solver produces accurate ground truth.
    The dataset is generated with a Runge-Kutta method for stiff ODEs [28]; no tolerances are reported, so the reference solutions are trusted as exact.
  • standard math FNO universality and mesh-independence properties from cited works hold for the present setting.
    The architecture design follows the FNO literature; the paper relies on these theoretical properties to justify the approach.

how reviews work

0 comments
Cite this review

Pith. "Pith review of Learning High-dimensional Ionic Model Dynamics Using Fourier Neural Operators." pith.science (2026). https://pith.science/paper/FU7MZFBM

@misc{pith2026250514039,
  author       = {Pith},
  title        = {Pith review of: Learning High-dimensional Ionic Model Dynamics Using Fourier Neural Operators},
  year         = {2026},
  howpublished = {\url{https://pith.science/paper/FU7MZFBM}},
  note         = {Machine review of arXiv:2505.14039}
}
read the original abstract

Ionic models, described by systems of stiff ordinary differential equations, are fundamental tools for simulating the complex dynamics of excitable cells in both Computational Neuroscience and Cardiology. Approximating these models using Artificial Neural Networks poses significant challenges due to their inherent stiffness, multiscale nonlinearities, and the wide range of dynamical behaviors they exhibit, including multiple equilibrium points, limit cycles, and intricate interactions. While in previous studies the dynamics of the transmembrane potential has been predicted in low dimensionality settings, in the present study we extend these results by investigating whether Fourier Neural Operators can effectively learn the evolution of all the state variables within these dynamical systems in higher dimensions. We demonstrate the effectiveness of this approach by accurately learning the dynamics of three well-established ionic models with increasing dimensionality: the two-variable FitzHugh-Nagumo model, the four-variable Hodgkin-Huxley model, and the forty-one-variable O'Hara-Rudy model. To ensure the selection of near-optimal configurations for the Fourier Neural Operator, we conducted automatic hyperparameter tuning under two scenarios: an unconstrained setting, where the number of trainable parameters is not limited, and a constrained case with a fixed number of trainable parameters. Both constrained and unconstrained architectures achieve comparable results in terms of accuracy across all the models considered. However, the unconstrained architecture required approximately half the number of training epochs to achieve similar error levels, as evidenced by the loss function values recorded during training. These results underline the capabilities of Fourier Neural Operators to accurately capture complex multiscale dynamics, even in high-dimensional dynamical systems.

Figures

Figures reproduced from arXiv: 2505.14039 by the authors.

Figure 1
Figure 1. Visual representation of a Fourier Neural Operator. [PITH_FULL_IMAGE:figures/full_fig_p004_1.png] view at source ↗
Figure 2
Figure 2. FHN model: FNO training relative L 2 (blue) loss and relative test L 1 (green), L 2 (orange), and H1 (red) loss functions. (a) Unconstrained FNO model, (b) constrained FNO model. (a) Unconstrained FNO bar plot. (b) Constrained FNO bar plot. (c) Box plot for unconstrained and constrained FNO [PITH_FULL_IMAGE:figures/full_fig_p010_2.png] view at source ↗
Figure 3
Figure 3. FHN model: FNO performance comparison. Figure [PITH_FULL_IMAGE:figures/full_fig_p010_3.png] view at source ↗
Figures from the paper (9 more)
Figure 5
Figure 5. Figure 5: FHN model: examples of FNO performance. Each column represents a single example from a [PITH_FULL_IMAGE:figures/full_fig_p011_5.png]
Figure 6
Figure 6. Figure 6: HH model: FNO training relative L 2 (blue) loss and test relative L 1 (green), L 2 (orange), and H1 (red) loss functions. (a) Unconstrained FNO model, (b) constrained FNO model. (a) Unconstrained FNO bar plot. (b) Constrained FNO bar plot. 13 [PITH_FULL_IMAGE:figures/…
Figure 7
Figure 7. Figure 7: HH model: FNO performance comparison. Fig. [PITH_FULL_IMAGE:figures/full_fig_p014_7.png]
Figure 8
Figure 8. Figure 8: HH model: examples of FNO performance. Each column represents a single example from a [PITH_FULL_IMAGE:figures/full_fig_p014_8.png]
Figure 9
Figure 9. Figure 9: ORd model: FNO training relative L 2 (blue) loss and test relative L 1 (green), L 2 (orange), and H1 (red) loss functions. (a) Unconstrained FNO model, (b) constrained FNO model. (a) Unconstrained FNO bar plot. (b) Constrained FNO bar plot. (c) Box plot for the unconst…
Figure 10
Figure 10. Figure 10: ORd model: FNO performance comparison. Fig. [PITH_FULL_IMAGE:figures/full_fig_p016_10.png]
Figure 11
Figure 11. Figure 11: ORd model: examples of FNO performance. Each figure contains five plots, arranged in order [PITH_FULL_IMAGE:figures/full_fig_p017_11.png]
Figure 12
Figure 12. Figure 12: Dimensionality dependence of FNO parameters (from the summary Tables [PITH_FULL_IMAGE:figures/full_fig_p018_12.png]
Figure 13
Figure 13. Figure 13: Component-wise box plot for the unconstrained FNO relative errors on the test set. [PITH_FULL_IMAGE:figures/full_fig_p019_13.png]

Discussion (0). Sign in to comment.

Reference graph

Works this paper leans on

31 extracted references · 24 canonical work pages

  1. [1]

    Neural operators for accelerating scientific simulations and design, Nature Reviews Physics 6, no

    Kamyar Azizzadenesheli, Nikola , Zongyi Li, Miguel Liu-Schiaffini, Jean Kossaifi, Anima Anandkumar. Neural operators for accelerating scientific simulations and design, Nature Reviews Physics 6, no. 5 : 320-328, 2024

  2. [2]

    Francesca Bartolucci, Emmanuel de B´ ezenac, Bogdan Raoni´ c, Roberto Molinaro, Siddhartha Mishra, Rima Alaifari,Are neural operators really neural operators? frame theory meets operator learning, SAM Research Report, volume 2023, 2023

  3. [3]

    James Bergstra, R´ emi Bardenet, Yoshua Bengio, Bal´ azs K´ egl,Algorithms for hyper-parameter optimization, Advances in Neural Information Processing Systems, volume 24, 2011

  4. [4]

    Kovachki, Matthew E

    Edoardo Calvello, Nikola B. Kovachki, Matthew E. Levine, Andrew M. Stuart,Continuum attention for neural operators, arXiv preprint arXiv:2406.06486, 2024

  5. [5]

    Daniel Cebr ´ ıan-Lacasa, Pedro Parra-Rivas, Daniel Ruiz-Reyn´ es, Lendert Gelens,Six decades of the FitzHugh-Nagumo model: A guide through its spatio-temporal dynamics and influence across disciplines, arXiv preprint arXiv:2404.11403, 2024

  6. [6]

    Pavarino,Learning the Hodgkin–Huxley model with operator learning techniques, Computer Methods in Applied Mechanics and Engineering, volume 432, page 117381, 2024

    Edoardo Centofanti, Massimiliano Ghiotto, Luca F. Pavarino,Learning the Hodgkin–Huxley model with operator learning techniques, Computer Methods in Applied Mechanics and Engineering, volume 432, page 117381, 2024

  7. [7]

    CINECA Supercomputing Centre, SuperComputing Applications and Innovation Department, LEONARDO: A Pan-European Pre-Exascale Supercomputer for HPC and AI applications, Journal of Large-Scale Research Facilities, 8, A186, 2024

  8. [8]

    Salvatore Cuomo, Vincenzo S. Di Cola, Fabio Giampaolo, Gianluigi Rozza, Maziar Raissi, Francesco Piccialli,Scientific machine learning through physics–informed neural networks: Where we are and what’s next, Journal of Scientific Computing, 92, 3, 88, 2022

Show all 31 references
  1. [9]

    Richard FitzHugh,Impulses and physiological states in theoretical models of nerve membrane, Bio- physical Journal, 1, 6, 445–466, 1961. 20

  2. [10]

    Franzone, Luca F

    Piero C. Franzone, Luca F. Pavarino, Simone Scacchi,Mathematical Cardiac Electrophysiology, 13, Springer, 2014

  3. [11]

    Massimiliano Ghiotto,HyperNOs: Automated and Parallel Library for Neural Operators Research, arXiv preprint arXiv:2503.18087, 2025

  4. [12]

    Karniadakis,Physics-informed deep neural operator networks, Machine Learning in Modeling and Simulation: Methods and Applications, 219–254, 2023

    Somdatta Goswami, Aniruddha Bora, Yue Yu, George E. Karniadakis,Physics-informed deep neural operator networks, Machine Learning in Modeling and Simulation: Methods and Applications, 219–254, 2023

  5. [13]

    Hodgkin, Andrew F

    Alan L. Hodgkin, Andrew F. Huxley,A quantitative description of membrane current and its application to conduction and excitation in nerve, The Journal of Physiology, 117, 4, 500, 1952

  6. [14]

    Izhikevich,Dynamical Systems in Neuroscience, MIT press, 2007

    Eugene M. Izhikevich,Dynamical Systems in Neuroscience, MIT press, 2007

  7. [15]

    James Keener, James Sneyd,Mathematical Physiology: II: Systems physiology, Springer, 2009

  8. [16]

    Nikola Kovachki, Zongyi Li, Burigede Liu, Kamyar Azizzadenesheli, Kaushik Bhattacharya, Andrew Stuart, Anima Anandkumar,Neural operator: Learning maps between function spaces with applications to pdes, Journal of Machine Learning Research, 24, 89, 1–97, 2023

  9. [17]

    Stuart,Nonlocality and nonlinearity implies universality in operator learning, arXiv preprint arXiv:2304.13221, 2023

    Samuel Lanthaler, Zongyi Li, Andrew M. Stuart,Nonlocality and nonlinearity implies universality in operator learning, arXiv preprint arXiv:2304.13221, 2023

  10. [18]

    Liam Li, Kevin Jamieson, Afshin Rostamizadeh, Ekaterina Gonina, Jonathan Ben-Tzur, Moritz Hardt, Benjamin Recht, and Ameet Talwalkar.A system for massively parallel hyperparameter tuning, Proceedings of machine learning and systems 2: 230-246 2020

  11. [19]

    Farimani,Transformer for partial differential equations’ operator learning, arXiv preprint arXiv:2205.13671, 2022

    Zijie Li, Kazem Meidani, Amir B. Farimani,Transformer for partial differential equations’ operator learning, arXiv preprint arXiv:2205.13671, 2022

  12. [20]

    Stuart, Anima Anandkumar,Fourier neural operator for parametric partial differential equations, arXiv preprint arXiv:2010.08895, 2020

    Zongyi Li, Nikola Kovachki, Kamyar Azizzadenesheli, Burigede Liu, Kaushik Bhattacharya, Andrew M. Stuart, Anima Anandkumar,Fourier neural operator for parametric partial differential equations, arXiv preprint arXiv:2010.08895, 2020

  13. [21]

    Gonzalez, Ion Stoica,Tune: A Research Platform for Distributed Model Selection and Training, arXiv preprint arXiv:1807.05118, 2018

    Richard Liaw, Eric Liang, Robert Nishihara, Philipp Moritz, Joseph E. Gonzalez, Ion Stoica,Tune: A Research Platform for Distributed Model Selection and Training, arXiv preprint arXiv:1807.05118, 2018

  14. [22]

    Ilya Loshchilov, Frank Hutter,Decoupled weight decay regularization, arXiv preprint arXiv:1711.05101, 2017

  15. [23]

    Karniadakis,Learning nonlinear operators via DeepONet based on the universal approximation theorem of operators, Nature Machine Intelligence, 3, 3, 218–229, 2021

    Lu Lu, Pengzhan Jin, Guofei Pang, Zhongqiang Zhang, George E. Karniadakis,Learning nonlinear operators via DeepONet based on the universal approximation theorem of operators, Nature Machine Intelligence, 3, 3, 218–229, 2021

  16. [24]

    Thomas O’Hara, L´ aszl´ o Vir´ ag, Andr´ as Varr´ o, Yoram Rudy,Simulation of the undiseased human cardiac ventricular action potential: model formulation and experimental validation, PLoS Computational Biology, 7, 5, e1002061, 2011

  17. [25]

    Maziar Raissi, Paris Perdikaris, George E. Karniadakis,Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations,Journal of Computational Physics, 378, 686-707, 2019

  18. [26]

    Bogdan Raonic, Roberto Molinaro, Tobias Rohner, Siddhartha Mishra, Emmanuel de Bezenac, Convolutional neural operators, ICLR 2023 Workshop on Physics for Machine Learning, 2023

  19. [27]

    Francesco Regazzoni, Stefano Pagani, Matteo Salvador, Luca Dede’, Alfio Quarteroni,Learning the intrinsic dynamics of spatio-temporal processes through Latent Dynamics Networks, Nature Communications, 15, 1, 1834, 2024. 21

  20. [28]

    Shampine, Mark W

    Lawrence F. Shampine, Mark W. Reichelt, Jacek A. Kierzenka,Solving index-1 DAEs in MATLAB and Simulink, SIAM Review, 41, 3, 538–552, 1999

  21. [29]

    Simin Shekarpaz, Fanhai Zeng, George E. Karniadakis,Splitting Physics-Informed Neural Networks for Inferring the Dynamics of Integer-and Fractional-Order Neuron Models, Communications in Computational Physics, 35, 1, 1–37, 2024

  22. [30]

    Tapas Tripura, Souvik Chakraborty,Wavelet neural operator for solving parametric partial differential equations in computational mechanics problems, Computer Methods in Applied Mechanics and Engineering, 404, 115783, 2023

  23. [31]

    Haixu Wu, Huakun Luo, Haowen Wang, Jianmin Wang, Mingsheng Long,Transolver: A fast transformer solver for PDEs on general geometries, arXiv preprint arXiv:2402.02366, 2024. 22

Pith tools

Reviewed August 7, 2026 · model on record in the stance chip above.