Transformers for Modeling Physical Systems

Data set associated with the publication Transformers for Modeling Physical Systems. Transformers are widely used in natural language processing due to their ability to model longer-term dependencies in text. Although these models achieve state-of-the-art performance for many language related tasks, their applicability outside of the natural language processing field has been minimal. In this work, we propose the use of transformer models for the prediction of dynamical systems representative of physical phenomena. 

This data set includes data in HDF5 files for:

Lorenz ODE:

• lorenz_training_rk.tar.gz
• lorenz_valid_rk.tar.gz
• lorenz_test_rk.tar.gz

Flow Around a Cylinder:

• cylinder_training.tar.gz
• cylinder_valid.tar.gz
• cylinder_test.tar.gz

Gray-Scott Reaction-Diffusion:

• grayscott_training.tar.gz
• grayscott_valid.tar.gz
• grayscott_test.tar.gz

Rossler ODE:

• rossler_training.tar.gz
• rossler_valid.tar.gz

As well as several pretrained embedding models for the Google Collab notebooks on Github:

• embedding_lorenz_pretrained.pth
• embedding_cylinder_pretrained.pth
• embedding_rossler_pretrained.pth

See the Github repository for code base: https://github.com/zabaras/transformer-physx/