- Open Access
Physics-informed time series analysis with Kolmogorov-Arnold networks under Ehrenfest constraints
Phys. Rev. Research 8, 023018 – Published 6 April, 2026
DOI: https://doi.org/10.1103/84vm-tvzm
Abstract
The prediction of quantum dynamical responses lies at the heart of modern physics. Yet, modeling these time-dependent behaviors remains a formidable challenge because quantum systems evolve in high-dimensional Hilbert spaces, often rendering traditional numerical methods computationally prohibitive. While large language models have achieved remarkable success in sequential prediction, quantum dynamics presents a fundamentally different challenge: forecasting the entire temporal evolution of quantum systems rather than merely the next element in a sequence. Existing neural architectures such as recurrent and convolutional networks often require vast training datasets and suffer from spurious oscillations that compromise physical interpretability. In this work, we introduce an approach: Kolmogorov-Arnold networks (KANs) augmented with physics-informed loss functions that enforce the Ehrenfest theorems. Our method achieves superior accuracy with significantly less training data: It requires only of the samples (200) compared to temporal convolution networks (3700). We further introduce the chain of KANs, an architecture that embeds temporal causality directly into the model design, making it particularly well suited for time series modeling. Our results demonstrate that physics-informed KANs offer a compelling advantage over conventional black-box models, maintaining both mathematical rigor and physical consistency while dramatically reducing data requirements.
Physics Subject Headings (PhySH)
Article Text
References (53)
- D. I. Bondar, R. Cabrera, R. R. Lompay, M. Y. Ivanov, and H. A. Rabitz, Operational dynamic modeling transcending quantum and classical mechanics, Phys. Rev. Lett. 109, 190403 (2012).
- P. Ehrenfest, Bemerkung über die angenäherte Gültigkeit der klassischen Mechanik innerhalb der Quantenmechanik, Z. Phys. 45, 455 (1927).
- L. E. Ballentine, Y. Yang, and J. P. Zibin, Inadequacy of Ehrenfest's theorem to characterize the classical regime, Phys. Rev. A 50, 2854 (1994).
- A. G. Campos, D. I. Bondar, R. Cabrera, and H. A. Rabitz, How to make distinct dynamical systems appear spectrally identical, Phys. Rev. Lett. 118, 083201 (2017).
- G. McCaul, C. Orthodoxou, K. Jacobs, G. H. Booth, and D. I. Bondar, Driven imposters: Controlling expectations in many-body systems, Phys. Rev. Lett. 124, 183201 (2020).
- R. Cabrera, A. G. Campos, D. I. Bondar, and H. A. Rabitz, Dirac open-quantum-system dynamics: Formulations and simulations, Phys. Rev. A 94, 052111 (2016).
- R. Cabrera, A. G. Campos, H. A. Rabitz, and D. I. Bondar, Operational dynamical modeling of spin 1/2 relativistic particles: The Dirac equation and its classical limit, Eur. Phys. J. Spec. Top. 227, 2195 (2019).
- D. I. Bondar, R. Cabrera, A. Campos, S. Mukamel, and H. A. Rabitz, Wigner–Lindblad equations for quantum friction, J. Phys. Chem. Lett. 7, 1632 (2016).
- D. V. Zhdanov, D. I. Bondar, and T. Seideman, No thermalization without correlations, Phys. Rev. Lett. 119, 170402 (2017).
- S. L. Vuglar, D. V. Zhdanov, R. Cabrera, T. Seideman, C. Jarzynski, and D. I. Bondar, Nonconservative forces via quantum reservoir engineering, Phys. Rev. Lett. 120, 230404 (2018).
- A. J. Schimmoller, G. McCaul, H. Abele, and D. I. Bondar, Decoherence-free entropic gravity: Model and experimental tests, Phys. Rev. Res. 3, 033065 (2021).
- E. J. Sung, A. G. Campos, H. Abele, and D. I. Bondar, Decoherence-free entropic gravity for a Dirac fermion, Phys. Rev. D 108, 104036 (2023).
- F. Krausz and M. Ivanov, Attosecond physics, Rev. Mod. Phys. 81, 163 (2009).
- S. Ghimire, A. D. DiChiara, E. Sistrunk, P. Agostini, L. F. DiMauro, and D. A. Reis, Observation of high-order harmonic generation in a bulk crystal, Nat. Phys. 7, 138 (2011).
- T. T. Luu, M. Garg, S. Y. Kruchinin, A. Moulet, M. T. Hassan, and E. Goulielmakis, Extreme ultraviolet high-harmonic spectroscopy of solids, Nature (London) 521, 498 (2015).
- G. McCaul, K. Jacobs, and D. I. Bondar, Towards single atom computing via high harmonic generation, Eur. Phys. J. Plus 138, 123 (2023).
- A. B. Magann, G. McCaul, H. A. Rabitz, and D. I. Bondar, Sequential optical response suppression for chemical mixture characterization, Quantum 6, 626 (2022).
- S. Takayoshi, Y. Murakami, and P. Werner, High-harmonic generation in quantum spin systems, Phys. Rev. B 99, 184303 (2019).
- T. N. Ikeda and M. Sato, High-harmonic generation by electric polarization, spin current, and magnetization, Phys. Rev. B 100, 214424 (2019).
- R. K. Malla, D. Kim, D. E. Kim, A. Chacón, and W. J. M. Kort-Kamp, Ultrafast nonequilibrium dynamics and high-harmonic generation in two-dimensional quantum spin Hall materials, Phys. Rev. Mater. 7, L051201 (2023).
- M. Lytova, M. Spanner, and I. Tamblyn, Deep learning and high harmonic generation, Can. J. Phys. 101, 132 (2023).
- J. Serrano, J. M. Pablos-Marín, and C. Hernández-García, Machine-learning applied to the simulation of high harmonic generation driven by structured laser beams, EPJ Web Conf. 287, 13018 (2023).
- Y. Shen, E. Liao, T. Tai, Y. Liao, C. Sun, C. Lee, S. See, and H. Chen, Deep learning‐based photodamage reduction on harmonic generation microscope at low‐level optical power, J. Biophotonics 17, e202300285 (2023).
- N. D. Klimkin, A. Jiménez-Galán, R. E. F. Silva, and M. Ivanov, Symmetry-aware deep neural networks for high harmonic spectroscopy in solids, Opt. Express 31, 20559 (2023).
- A. Mihailescu, A new approach to theoretical investigations of high harmonics generation by means of fs laser interaction with overdense plasma layers. Combining particle-in-cell simulations with machine learning, J. Instrum. 11, C12004 (2016).
- J.-Z. Yan, S.-S. Zhao, W.-D. Lan, S.-Y. Li, S.-S. Zhou, J.-G. Chen, J.-Y. Zhang, and Y.-J. Yang, Calculation of high-order harmonic generation of atoms and molecules by combining time series prediction and neural networks, Opt. Express 30, 35444 (2022).
- J. M. Pablos-Marín, J. Serrano, and C. Hernández-García, Simulating macroscopic high-order harmonic generation driven by structured laser beams using artificial intelligence, Comput. Phys. Commun. 291, 108823 (2023).
- T. Zahavy, A. Dikopoltsev, D. Moss, G. I. Haham, O. Cohen, S. Mannor, and M. Segev, Deep learning reconstruction of ultrashort pulses, Optica 5, 666 (2018).
- C. Brunner, A. Duensing, C. Schröder, M. Mittermair, V. Golkov, M. Pollanka, D. Cremers, and R. Kienberger, Deep learning in attosecond metrology, Opt. Express 30, 15669 (2022).
- A. Sanchez-Gonzalez et al., Accurate prediction of X-ray pulse properties from a free-electron laser using machine learning, Nat. Commun. 8, 15461 (2017).
- S. Kumar Giri, U. Saalmann, and J. M. Rost, Purifying electron spectra from noisy pulses with machine learning using synthetic Hamilton matrices, Phys. Rev. Lett. 124, 113201 (2020).
- A. Sen, B. K. Parida, K. Jacobs, and D. I. Bondar, Input-output optics as a causal time series mapping: A generative machine learning solution, Phys. Rev. Res. 7, 023015 (2025).
- Z. Liu, Y. Wang, S. Vaidya, F. Ruehle, J. Halverson, M. Soljačić, T. Y. Hou, and M. Tegmark, KAN: Kolmogorov-Arnold networks, in Proceedings of the 13th International Conference on Learning Representations (ICLR) (OpenReview.net, Singapore, 2025), pp. 1–13.
- M. Raissi, P. Perdikaris, and G. Karniadakis, Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations, J. Comput. Phys. 378, 686 (2019).
- G. Pang, L. Lu, and G. E. Karniadakis, fPINNs: Fractional physics-informed neural networks, SIAM J. Sci. Comput. 41, A2603 (2019).
- L. Yuan, Y.-Q. Ni, X.-Y. Deng, and S. Hao, A-PINN: Auxiliary physics informed neural networks for forward and inverse problems of nonlinear integro-differential equations, J. Comput. Phys. 462, 111260 (2022).
- J. O’Leary, J. A. Paulson, and A. Mesbah, Stochastic physics-informed neural ordinary differential equations, J. Comput. Phys. 468, 111466 (2022).
- S. Cuomo, V. S. Di Cola, F. Giampaolo, G. Rozza, M. Raissi, and F. Piccialli, Scientific machine learning through physics–informed neural networks: Where we are and what's next, J. Sci. Comput. 92, 88 (2022).
- S. Patra, S. Panda, B. K. Parida, M. Arya, K. Jacobs, D. I. Bondar, and A. Sen, Physics informed Kolmogorov-Arnold neural networks for dynamical analysis via efficent-KAN and WAV-KAN, J. Mach. Learn. Res. 26, 1 (2025).
- S. Wang, S. Sankaran, H. Wang, and P. Perdikaris, An expert's guide to training physics-informed neural networks, arXiv:2308.08468.
- A. Krishnapriyan, A. Gholami, S. Zhe, R. Kirby, and M. W. Mahoney, Characterizing possible failure modes in physics-informed neural networks, Adv. Neural Inf. Process. Syst. 34, 26548 (2021).
- Z. K. Lawal, H. Yassin, D. T. C. Lai, and A. C. Idris, Physics-informed neural network (PINN) evolution and beyond: A systematic literature review and bibliometric analysis, Big Data Cognit. Comput. 6, 140 (2022).
- W. Ji, W. Qiu, Z. Shi, S. Pan, and S. Deng, Stiff-PINN: Physics-informed neural network for stiff chemical kinetics, J. Phys. Chem. A 125, 8098 (2021).
- O. Fuks and H. A. Tchelepi, Limitations of physics informed machine learning for nonlinear two-phase transport in porous media, J. Mach. Learn. Model. Comput. 1, 19 (2020).
- V. Dwivedi, N. Parashar, and B. Srinivasan, Distributed learning machines for solving forward and inverse problems in partial differential equations, Neurocomputing 420, 299 (2021).
- V. I. Arnold, On the representation of functions of several variables as a superposition of functions of a smaller number of variables, in Collected Works, edited by A. B. Givental, B. A. Khesin, J. E. Marsden, A. N. Varchenko, V. A. Vassiliev, O. Y. Viro, and V. M. Zakalyukin (Springer, Berlin, 2009), pp. 25–46.
- G. Vidal, Classical simulation of infinite-size quantum lattice systems in one spatial dimension, Phys. Rev. Lett. 98, 070201 (2007).
- A. De Gaetano, PINNs and Sobolev spaces, in 2025 IEEE 29th International Conference on Intelligent Engineering Systems (INES) (IEEE, Piscataway, NJ, 2025), pp. 000011–000012.
- I. Lukin and A. Sen, Physics-informed time series analysis with Kolmogorov-Arnold networks under Ehrenfest constraints, GitHub (2025), https://github.com/illyaLukin/KAN_TS.
- S. A. Morris, Hilbert 13: Are there any genuine continuous multivariate real-valued functions? Bull. Am. Math. Soc. 58, 107 (2021).
- S. Somvanshi, S. A. Javed, M. M. Islam, D. Pandit, and S. Das, A survey on Kolmogorov-Arnold network, ACM Comput. Surv. 58, 1 (2025).
- Blealtan, Efficient KAN: A memory-efficient KAN implementation, GitHub (2024), https://github.com/Blealtan/efficient-kan.
- Z. Bozorgasl and H. Chen, Wav-KAN: Wavelet Kolmogorov-Arnold networks, arXiv:2405.12832.