Physics-Informed Regularization for Domain-Agnostic Dynamical System Modeling
Zijie Huang, Wanjia Zhao, Jingdong Gao, Ziniu Hu, Xiao Luo, Yadi Cao, Yuanzhou Chen, Yizhou Sun, Wei Wang
Abstract
Learning complex physical dynamics purely from data is challenging due to the intrinsic properties of systems to be satisfied. Incorporating physics-informed priors, such as in Hamiltonian Neural Networks (HNNs), achieves high-precision modeling for energy-conservative systems. However, real-world systems often deviate from strict energy conservation and follow different physical priors. To address this, we present a framework that achieves high-precision modeling for a wide range of dynamical systems from the numerical aspect, by enforcing Time-Reversal Symmetry (TRS) via a novel regularization term. It helps preserve energies for conservative systems while serving as a strong inductive bias for non-conservative, reversible systems. While TRS is a domain-specific physical prior, we present the first theoretical proof that TRS loss can universally improve modeling accuracy by minimizing higher-order Taylor terms in ODE integration, which is numerically beneficial to various systems regardless of their properties, even for irreversible systems. By integrating the TRS loss within neural ordinary differential equation models, the proposed model TREAT demonstrates superior performance on diverse physical systems. It achieves a significant 11.5% MSE improvement in a challenging chaotic triple-pendulum scenario, underscoring TREAT's broad applicability and effectiveness.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0092fecd-d8ab-4ca2-b3ab-b5638e98ac02Cited by top-tier papers7
- Flow Field Reconstruction with Sensor Placement Policy LearningRuoyan Li, Guancheng Wan, Zijie Huang, Zixiao Liu et al.NeurIPS 2025 · 4 citations
- Architecture-Aware Learning Curve Extrapolation via Graph Ordinary Differential EquationYanna Ding, Zijie Huang, Xiao Shou, Yihang Guo et al.AAAI 2025 · 4 citations
- Physics-Inspired All-Pair Interaction Learning for 3D Dynamics ModelingKai Yang, Yuqi Huang, Junheng Tao, Wanyu Wang et al.ICLR 2026 · 2 citations
- GeoAda: Efficiently Finetune Geometric Diffusion Models with Equivariant AdaptersWanjia Zhao, Jiaqi Han, Siyi Gu, Mingjian Jiang et al.NeurIPS 2025 · 1 citation
- Towards Generalizable PDE Dynamics Forecasting via Physics-Guided Invariant LearningSiyang Li, Yize Chen, Yan Guo, Ming Huang et al.ICLR 2026 · 1 citation
Builds on18
- Learning to Simulate Complex Physics with Graph NetworksAlvaro Sanchez-Gonzalez, Jonathan Godwin, Tobias Pfaff, Rex Ying et al.ICML 2020 · 1,439 citations
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 1,432 citations
- Learning Mesh-Based Simulation with Graph NetworksTobias Pfaff, Meire Fortunato, Alvaro Sanchez-Gonzalez, Peter W. BattagliaICLR 2021 · 1,175 citations
- GeoDiff: A Geometric Diffusion Model for Molecular Conformation GenerationMinkai Xu, Lantao Yu, Yang Song, Chence Shi et al.ICLR 2022 · 695 citations
- Learning Continuous System Dynamics from Irregularly-Sampled Partial ObservationsZijie Huang, Yizhou Sun, Wei WangNeurIPS 2020 · 103 citations
Related papers
- Time-Reversal Symmetric ODE NetworkIn Huh, Eunho Yang, Sung Ju Hwang, Jinwoo ShinNeurIPS 2020 · 29 citations
- Deconstructing the Inductive Biases of Hamiltonian Neural NetworksNate Gruver, Marc Anton Finzi, Samuel Don Stanton, Andrew Gordon WilsonICLR 2022 · 50 citations
- Sparse Symplectically Integrated Neural NetworksDaniel M. DiPietro, Shiying Xiong, Bo ZhuNeurIPS 2020 · 39 citations
- Weak Form Generalized Hamiltonian LearningKevin Course, Trefor W. Evans, Prasanth B. NairNeurIPS 2020 · 15 citations
- Machine learning structure preserving brackets for forecasting irreversible processesKookjin Lee, Nathaniel Trask, Panos StinisNeurIPS 2021 · 80 citations
