TENG: Time-Evolving Natural Gradient for Solving PDEs With Deep Neural Nets Toward Machine Precision
Zhuo Chen, Jacob McCarran, Esteban Vizcaino, Marin Soljacic, Di Luo
Abstract
Partial differential equations (PDEs) are instrumental for modeling dynamical systems in science and engineering. The advent of neural networks has initiated a significant shift in tackling these complexities though challenges in accuracy persist, especially for initial value problems. In this paper, we introduce the , generalizing time-dependent variational principles and optimization-based time integration, leveraging natural gradient optimization to obtain high accuracy in neural-network-based PDE solutions. Our comprehensive development includes algorithms like TENG-Euler and its high-order variants, such as TENG-Heun, tailored for enhanced precision and efficiency. TENG's effectiveness is further validated through its performance, surpassing current leading methods and achieving in step-by-step optimizations across a spectrum of PDEs, including the heat equation, Allen-Cahn equation, and Burgers' equation.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers4
- LM: Mutual Information Scaling Law for Long-Context Language ModelingZhuo Chen, Oriol Mayné i Comas, Zhuotao Jin, Di Luo et al.NeurIPS 2025 · 11 citations
- A Dirac-Frenkel-Onsager principle: Instantaneous residual minimization with gauge momentum for nonlinear parametrizations of PDE solutionsMatteo Raviola, Benjamin PeherstorferICML 2026 · 1 citation
- L-CUBE: Isolating Long-Context Capacity from Knowledge with Controllable Mutual Information ScalingZhuo Chen, Oriol Mayné i Comas, Zhuotao Jin, Di Luo et al.ICML 2026
- Metamizer: A Versatile Neural Optimizer for Fast and Accurate Physics SimulationsNils Wandel, Stefan Schulz, Reinhard KleinICLR 2025
Builds on7
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Learning to Simulate Complex Physics with Graph NetworksAlvaro Sanchez-Gonzalez, Jonathan Godwin, Tobias Pfaff, Rex Ying et al.ICML 2020 · 1,439 citations
- Learning Mesh-Based Simulation with Graph NetworksTobias Pfaff, Meire Fortunato, Alvaro Sanchez-Gonzalez, Peter W. BattagliaICLR 2021 · 1,175 citations
- Implicit Neural Spatial Representations for Time-dependent PDEsHonglin Chen, Rundi Wu, Eitan Grinspun, Changxi Zheng et al.ICML 2023 · 54 citations
- Randomized Sparse Neural Galerkin Schemes for Solving Evolution Equations with Deep NetworksJules Berman, Benjamin PeherstorferNeurIPS 2023 · 42 citations
Related papers
- TINNs: Time-Induced Neural Networks for Solving Time-Dependent PDEsChen-Yang Dai, Che-Chia Chang, Te-Sheng Lin, Ming-Chih Lai et al.ICML 2026 · 2 citations
- A Stable and Scalable Method for Solving Initial Value PDEs with Neural NetworksMarc Anton Finzi, Andres Potapczynski, Matthew Choptuik, Andrew Gordon WilsonICLR 2023 · 1 citation
- Mechanistic PDE Networks for Discovery of Governing EquationsAdeel Pervez, Efstratios Gavves, Francesco LocatelloICML 2025
- ANaGRAM: A Natural Gradient Relative to Adapted Model for efficient PINNs learningNilo Schwencke, Cyril FurtlehnerICLR 2025
- Fast training of accurate physics-informed neural networks without gradient descentChinmay Datar, Taniya Kapoor, Abhishek Chandra, Qing Sun et al.ICLR 2026 · 10 citations
