Text2PDE: Latent Diffusion Models for Accessible Physics Simulation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhou, Anthony, Li, Zijie, Schneier, Michael, Buchanan Jr, John R, Farimani, Amir Barati
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916597750300672
author Zhou, Anthony
Li, Zijie
Schneier, Michael
Buchanan Jr, John R
Farimani, Amir Barati
author_facet Zhou, Anthony
Li, Zijie
Schneier, Michael
Buchanan Jr, John R
Farimani, Amir Barati
contents Recent advances in deep learning have inspired numerous works on data-driven solutions to partial differential equation (PDE) problems. These neural PDE solvers can often be much faster than their numerical counterparts; however, each presents its unique limitations and generally balances training cost, numerical accuracy, and ease of applicability to different problem setups. To address these limitations, we introduce several methods to apply latent diffusion models to physics simulation. Firstly, we introduce a mesh autoencoder to compress arbitrarily discretized PDE data, allowing for efficient diffusion training across various physics. Furthermore, we investigate full spatio-temporal solution generation to mitigate autoregressive error accumulation. Lastly, we investigate conditioning on initial physical quantities, as well as conditioning solely on a text prompt to introduce text2PDE generation. We show that language can be a compact, interpretable, and accurate modality for generating physics simulations, paving the way for more usable and accessible PDE solvers. Through experiments on both uniform and structured grids, we show that the proposed approach is competitive with current neural PDE solvers in both accuracy and efficiency, with promising scaling behavior up to $\sim$3 billion parameters. By introducing a scalable, accurate, and usable physics simulator, we hope to bring neural PDE solvers closer to practical use.
format Preprint
id arxiv_https___arxiv_org_abs_2410_01153
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Text2PDE: Latent Diffusion Models for Accessible Physics Simulation
Zhou, Anthony
Li, Zijie
Schneier, Michael
Buchanan Jr, John R
Farimani, Amir Barati
Machine Learning
Recent advances in deep learning have inspired numerous works on data-driven solutions to partial differential equation (PDE) problems. These neural PDE solvers can often be much faster than their numerical counterparts; however, each presents its unique limitations and generally balances training cost, numerical accuracy, and ease of applicability to different problem setups. To address these limitations, we introduce several methods to apply latent diffusion models to physics simulation. Firstly, we introduce a mesh autoencoder to compress arbitrarily discretized PDE data, allowing for efficient diffusion training across various physics. Furthermore, we investigate full spatio-temporal solution generation to mitigate autoregressive error accumulation. Lastly, we investigate conditioning on initial physical quantities, as well as conditioning solely on a text prompt to introduce text2PDE generation. We show that language can be a compact, interpretable, and accurate modality for generating physics simulations, paving the way for more usable and accessible PDE solvers. Through experiments on both uniform and structured grids, we show that the proposed approach is competitive with current neural PDE solvers in both accuracy and efficiency, with promising scaling behavior up to $\sim$3 billion parameters. By introducing a scalable, accurate, and usable physics simulator, we hope to bring neural PDE solvers closer to practical use.
title Text2PDE: Latent Diffusion Models for Accessible Physics Simulation
topic Machine Learning
url https://arxiv.org/abs/2410.01153