/PgDMM

Physics-guided Deep Markov Models

Primary LanguagePythonMIT LicenseMIT

Physics-guided Deep Markov Models (PgDMM)

This repository contains codes and data for the following publication:

PgDMM is a hybrid modeling framework for unsupervised learning and identification of nonlinear dynamical systems. The proposed framework takes advantage of the expressive power of deep learning, while retaining the driving physics of the dynamical system by imposing physics-driven restrictions on the side of the latent space.

Framework

PgDMM_framework

Repository Overview

  • silverbox_pgdmm - Codes for implementing PgDMM on Silverbox benchmark.
    • SNLS80mV.mat - Silverbox benchmark data.
    • data_loader.py - Functions for loading and processing raw data.
    • loss.py - Analytical loss functions.
    • models.py - PgDMM models.
    • training.py - Manages training and evaluation of models.
    • silverbox_main.py - Configuration for training PgDMM models on Silverbox benchmark.
  • silverbox_dmm - Codes for implementing DMM on Silverbox benchmark.
    • SNLS80mV.mat - Silverbox benchmark data.
    • data_loader.py - Functions for loading and processing raw data.
    • loss.py - Analytical loss functions.
    • models.py - DMM models.
    • training.py - Manages training and evaluation of models.
    • silverbox_main.py - Configuration for training DMM models on Silverbox benchmark.

Citation

Please cite the following paper if you find the work relevant and useful in your research:

@article{liu2022physics,
  title={Physics-guided deep Markov models for learning nonlinear dynamical systems with uncertainty},
  author={Liu, Wei and Lai, Zhilu and Bacsa, Kiran and Chatzi, Eleni},
  journal={Mechanical Systems and Signal Processing},
  volume={178},
  pages={109276},
  year={2022},
  publisher={Elsevier}
}