P$^2$C$^2$Net: PDE-Preserved Coarse Correction Network for efficient prediction of spatiotemporal dynamics

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Qi, Ren, Pu, Zhou, Hao, Liu, Xin-Yang, Deng, Zhiwen, Zhang, Yi, Chengze, Ruizhi, Liu, Hongsheng, Wang, Zidong, Wang, Jian-Xun, Ji-Rong_Wen, Sun, Hao, Liu, Yang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916464269721600
author Wang, Qi
Ren, Pu
Zhou, Hao
Liu, Xin-Yang
Deng, Zhiwen
Zhang, Yi
Chengze, Ruizhi
Liu, Hongsheng
Wang, Zidong
Wang, Jian-Xun
Ji-Rong_Wen
Sun, Hao
Liu, Yang
author_facet Wang, Qi
Ren, Pu
Zhou, Hao
Liu, Xin-Yang
Deng, Zhiwen
Zhang, Yi
Chengze, Ruizhi
Liu, Hongsheng
Wang, Zidong
Wang, Jian-Xun
Ji-Rong_Wen
Sun, Hao
Liu, Yang
contents When solving partial differential equations (PDEs), classical numerical methods often require fine mesh grids and small time stepping to meet stability, consistency, and convergence conditions, leading to high computational cost. Recently, machine learning has been increasingly utilized to solve PDE problems, but they often encounter challenges related to interpretability, generalizability, and strong dependency on rich labeled data. Hence, we introduce a new PDE-Preserved Coarse Correction Network (P$^2$C$^2$Net) to efficiently solve spatiotemporal PDE problems on coarse mesh grids in small data regimes. The model consists of two synergistic modules: (1) a trainable PDE block that learns to update the coarse solution (i.e., the system state), based on a high-order numerical scheme with boundary condition encoding, and (2) a neural network block that consistently corrects the solution on the fly. In particular, we propose a learnable symmetric Conv filter, with weights shared over the entire model, to accurately estimate the spatial derivatives of PDE based on the neural-corrected system state. The resulting physics-encoded model is capable of handling limited training data (e.g., 3--5 trajectories) and accelerates the prediction of PDE solutions on coarse spatiotemporal grids while maintaining a high accuracy. P$^2$C$^2$Net achieves consistent state-of-the-art performance with over 50\% gain (e.g., in terms of relative prediction error) across four datasets covering complex reaction-diffusion processes and turbulent flows.
format Preprint
id arxiv_https___arxiv_org_abs_2411_00040
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle P$^2$C$^2$Net: PDE-Preserved Coarse Correction Network for efficient prediction of spatiotemporal dynamics
Wang, Qi
Ren, Pu
Zhou, Hao
Liu, Xin-Yang
Deng, Zhiwen
Zhang, Yi
Chengze, Ruizhi
Liu, Hongsheng
Wang, Zidong
Wang, Jian-Xun
Ji-Rong_Wen
Sun, Hao
Liu, Yang
Numerical Analysis
Artificial Intelligence
Machine Learning
When solving partial differential equations (PDEs), classical numerical methods often require fine mesh grids and small time stepping to meet stability, consistency, and convergence conditions, leading to high computational cost. Recently, machine learning has been increasingly utilized to solve PDE problems, but they often encounter challenges related to interpretability, generalizability, and strong dependency on rich labeled data. Hence, we introduce a new PDE-Preserved Coarse Correction Network (P$^2$C$^2$Net) to efficiently solve spatiotemporal PDE problems on coarse mesh grids in small data regimes. The model consists of two synergistic modules: (1) a trainable PDE block that learns to update the coarse solution (i.e., the system state), based on a high-order numerical scheme with boundary condition encoding, and (2) a neural network block that consistently corrects the solution on the fly. In particular, we propose a learnable symmetric Conv filter, with weights shared over the entire model, to accurately estimate the spatial derivatives of PDE based on the neural-corrected system state. The resulting physics-encoded model is capable of handling limited training data (e.g., 3--5 trajectories) and accelerates the prediction of PDE solutions on coarse spatiotemporal grids while maintaining a high accuracy. P$^2$C$^2$Net achieves consistent state-of-the-art performance with over 50\% gain (e.g., in terms of relative prediction error) across four datasets covering complex reaction-diffusion processes and turbulent flows.
title P$^2$C$^2$Net: PDE-Preserved Coarse Correction Network for efficient prediction of spatiotemporal dynamics
topic Numerical Analysis
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2411.00040