Abstract:We describe a framework that can integrate prior physical information, e.g., the presence of kinematic constraints, to support data-driven simulation in multi-body dynamics. Unlike other approaches, e.g., Fully-connected Neural Network (FCNN) or Recurrent Neural Network (RNN)-based methods that are used to model the system states directly, the proposed approach embraces a Neural Ordinary Differential Equation (NODE) paradigm that models the derivatives of the system states. A central part of the proposed methodology is its capacity to learn the multibody system dynamics from prior physical knowledge and constraints combined with data inputs. This learning process is facilitated by a constrained optimization approach, which ensures that physical laws and system constraints are accounted for in the simulation process. The models, data, and code for this work are publicly available as open source at <a class="link-external link-https" href="https://github.com/uwsbel/sbel-reproducibility/tree/master/2024/MNODE-code" rel="external noopener nofollow">this https URL</a>.
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: In the multibody dynamics system (Multibody Dynamics, MBD), how to combine prior physical information (such as motion constraints) and use data - driven methods to model and simulate the dynamic behavior of the system. Specifically, the paper proposes a method based on neural ordinary differential equations (Neural Ordinary Differential Equation, NODE), called MBD - NODE, to learn the dynamics of multibody systems and ensure that physical laws and system constraints are adhered to during the simulation process.
### Specific description of the problem
1. **Limitations of existing methods**:
- Traditional methods such as **fully - connected neural networks (FCNN)** and **recurrent neural networks (RNN)** directly model the system state, but these methods require a large amount of training data and are difficult to handle complex changes in initial conditions.
- **Time - series methods (such as LSTM)** can predict future states, but they are essentially discrete and rely on specific time steps, resulting in outputs being limited to discrete time points.
- Although **Hamiltonian neural networks (HNN)** and **Lagrangian neural networks (LNN)** can consider physical properties such as energy conservation, they face challenges of computational complexity and coordinate transformation when dealing with complex multibody dynamics systems.
2. **Advantages of MBD - NODE**:
- **Continuous - time modeling**: Different from traditional discrete - time models, MBD - NODE adopts continuous - time modeling and can flexibly handle data with different time resolutions.
- **Combining physical constraints**: By introducing constraint optimization methods, MBD - NODE can satisfy physical constraint conditions during both the training and inference stages, thereby improving the accuracy and generalization ability of the model.
- **Efficient learning process**: MBD - NODE utilizes the neural ordinary differential equation framework and combines prior physical knowledge to achieve effective learning of multibody system dynamics.
### Solution
The MBD - NODE method proposed in the paper mainly solves the problem through the following steps:
1. **Define the multibody system dynamics equation**:
\[
\begin{bmatrix}
M & \Phi_q^T \\
\Phi_q & 0
\end{bmatrix}
\begin{bmatrix}
\ddot{q} \\
\lambda
\end{bmatrix}
=
\begin{bmatrix}
F_e \\
\gamma_c
\end{bmatrix}
\]
where \(M\) is the mass matrix, \(\Phi_q\) is the constraint Jacobian matrix, \(q\) is the system state vector (generalized coordinates), \(\ddot{q}\) is the acceleration vector, \(\lambda\) is the Lagrange multiplier, \(F_e\) is the combined vector of generalized external forces and quadratic velocity terms, and \(\gamma_c\) is the right - hand side of the motion constraint equation at the acceleration level.
2. **Construct the MBD - NODE model**:
\[
\frac{dZ(t, \mu)}{dt} = f(Z(t, \mu), u(t), t, \mu; \Theta)
\]
where \(Z(t, \mu)\) is the combination of generalized coordinates and generalized velocities, \(u(t)\) is the external input (such as external force or torque), \(\mu\) is the problem - specific parameter, and \(\Theta\) is the parameter of the neural network.
3. **Loss function and optimization**:
- For the unconstrained case, use the mean - squared error (MSE) as the loss function:
\[
L(\Theta) = \left\| \Phi(Z_0, f, \Delta t) - Z_1 \right\|^2_2
\]
- For the constrained case, handle the constraint conditions by introducing penalty terms or directly solving the constraint equations.
4. **Numerical experiments**: