Prospective contribution to the MATHMOD 2022 Vienna conference.
We discuss nonlinear model predictive control (NMPC) for multi-body dynamics via physics-informed machine learning methods. Physics-informed neural networks (PINNs) are a promising tool to approximate (partial) differential equations. PINNs are not suited for control tasks in their original form since they are not designed to handle variable control actions or variable initial values. We thus present the idea of enhancing PINNs by adding control actions and initial conditions as additional network inputs. The high-dimensional input space is subsequently reduced via a sampling strategy and a zero-hold assumption. This strategy enables the controller design based on a PINN as an approximation of the underlying system dynamics. The additional benefit is that the sensitivities are easily computed via automatic differentiation, thus leading to efficient gradient-based algorithms. Finally, we present our results using our PINN-based MPC to solve a tracking problem for a complex mechanical system, a multi-link manipulator.
If you use this project for academic work, please consider citing our publication:
Jonas Nicodemus, Jonas Kneifl, Jörg Fehr, Benjamin Unger,
Physics-informed Neural Networks-based Model Predictive Control for Multi-link Manipulators,
IFAC-PapersOnLine, Volume 55, Issue 20, 2022.
To get a local copy up and running follow these simple steps.
A python environment is required, we recommend using a virtual environment.
- Clone the repo
git clone git@github.com:Jonas-Nicodemus/PINNs-based-MPC.git
- Go into the directory
cd PINNs-based-MPC - Install dependencies
pip install -r requirements.txt
There are two executable scripts located in src.
train_pinn.pycan be executed to learn weights and overwrite the already existing ones, which can be found underresources/weights.main.py, then evaluates the PINN first in self-loop prediction mode and subsequently in closed-loop mode connected to the real system (emulated by RK45) for the given reference trajectory.
Distributed under the MIT License. See LICENSE for more information.
Jonas Nicodemus - jonas.nicodemus@simtech.uni-stuttgart.de
Benjamin Unger - benjamin.unger@simtech.uni-stuttgart.de
Jonas Kneifl - jonas.kneifl@itm.uni-stuttgart.de
Jörg Fehr - joerg.fehr@itm.uni-stuttgart.de
Project Link: https://github.com/Jonas-Nicodemus/PINNs-based-MPC