Conservation-informed Graph Learning for Spatiotemporal Dynamics Prediction

📅 2024-12-30
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
Accurate spatiotemporal dynamics prediction in complex geometric domains under few-shot settings remains challenging due to poor interpretability, physical inconsistency, heavy reliance on large-scale labeled data, and weak generalization of existing deep learning methods. Method: We propose a conservation-law-guided graph neural network (GNN) framework featuring a novel symmetric architecture that explicitly embeds conservation-law priors. It jointly integrates multi-scale graph-structured encoding with implicit latent temporal stepping and enforces physical consistency via symmetry regularization. Contribution/Results: Unlike conventional GNNs, our framework eliminates dependence on extensive labeled data and implicit physics-informed inductive biases, enabling end-to-end interpretable modeling. Extensive experiments on synthetic and real-world datasets demonstrate significant improvements over state-of-the-art baselines in prediction accuracy, cross-domain generalization, and robustness to complex boundaries—achieving both high fidelity and verifiable physical consistency.

Technology Category

Application Category

📝 Abstract
Data-centric methods have shown great potential in understanding and predicting spatiotemporal dynamics, enabling better design and control of the object system. However, pure deep learning models often lack interpretability, fail to obey intrinsic physics, and struggle to cope with the various domains. While geometry-based methods, e.g., graph neural networks (GNNs), have been proposed to further tackle these challenges, they still need to find the implicit physical laws from large datasets and rely excessively on rich labeled data. In this paper, we herein introduce the conservation-informed GNN (CiGNN), an end-to-end explainable learning framework, to learn spatiotemporal dynamics based on limited training data. The network is designed to conform to the general conservation law via symmetry, where conservative and non-conservative information passes over a multiscale space enhanced by a latent temporal marching strategy. The efficacy of our model has been verified in various spatiotemporal systems based on synthetic and real-world datasets, showing superiority over baseline models. Results demonstrate that CiGNN exhibits remarkable accuracy and generalization ability, and is readily applicable to learning for prediction of various spatiotemporal dynamics in a spatial domain with complex geometry.
Problem

Research questions and friction points this paper is trying to address.

Temporal-Spatial Prediction
Deep Learning Limitations
Physics-based Rules
Innovation

Methods, ideas, or system contributions that make the work stand out.

CiGNN
Conservation Laws
Multi-scale Information Propagation
🔎 Similar Papers
No similar papers found.
Y
Yuan Mi
Renmin University of China, Beijing, China
Pu Ren
Pu Ren
Lawrence Berkeley National Lab, Northeastern University
Machine LearningAI for Science
H
Hongteng Xu
Renmin University of China, Beijing, China
H
Hongsheng Liu
Huawei Technologies, Shenzhen, China
Z
Zidong Wang
Huawei Technologies, Shenzhen, China
Y
Yike Guo
HKUST, Hong Kong, China
Ji-Rong Wen
Ji-Rong Wen
Gaoling School of Artificial Intelligence, Renmin University of China
Large Language ModelWeb SearchInformation RetrievalMachine Learning
H
Hao Sun
Renmin University of China, Beijing, China
Y
Yang Liu
University of Chinese Academy of Sciences, Beijing, China