Saved in:
Bibliographic Details
Main Authors: Gregory, Wilson G., Hogg, David W., Blum-Smith, Ben, Arias, Maria Teresa, Wong, Kaze W. K., Villar, Soledad
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2305.12585
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916465173594112
author Gregory, Wilson G.
Hogg, David W.
Blum-Smith, Ben
Arias, Maria Teresa
Wong, Kaze W. K.
Villar, Soledad
author_facet Gregory, Wilson G.
Hogg, David W.
Blum-Smith, Ben
Arias, Maria Teresa
Wong, Kaze W. K.
Villar, Soledad
contents Machine learning methods are increasingly being employed as surrogate models in place of computationally expensive and slow numerical integrators for a bevy of applications in the natural sciences. However, while the laws of physics are relationships between scalars, vectors, and tensors that hold regardless of the frame of reference or chosen coordinate system, surrogate machine learning models are not coordinate-free by default. We enforce coordinate freedom by using geometric convolutions in three model architectures: a ResNet, a Dilated ResNet, and a UNet. In numerical experiments emulating 2D compressible Navier-Stokes, we see better accuracy and improved stability compared to baseline surrogate models in almost all cases. The ease of enforcing coordinate freedom without making major changes to the model architecture provides an exciting recipe for any CNN-based method applied to an appropriate class of problems
format Preprint
id arxiv_https___arxiv_org_abs_2305_12585
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Equivariant geometric convolutions for emulation of dynamical systems
Gregory, Wilson G.
Hogg, David W.
Blum-Smith, Ben
Arias, Maria Teresa
Wong, Kaze W. K.
Villar, Soledad
Machine Learning
Machine learning methods are increasingly being employed as surrogate models in place of computationally expensive and slow numerical integrators for a bevy of applications in the natural sciences. However, while the laws of physics are relationships between scalars, vectors, and tensors that hold regardless of the frame of reference or chosen coordinate system, surrogate machine learning models are not coordinate-free by default. We enforce coordinate freedom by using geometric convolutions in three model architectures: a ResNet, a Dilated ResNet, and a UNet. In numerical experiments emulating 2D compressible Navier-Stokes, we see better accuracy and improved stability compared to baseline surrogate models in almost all cases. The ease of enforcing coordinate freedom without making major changes to the model architecture provides an exciting recipe for any CNN-based method applied to an appropriate class of problems
title Equivariant geometric convolutions for emulation of dynamical systems
topic Machine Learning
url https://arxiv.org/abs/2305.12585