pyprnn 0.0.2

Physically Recurrent Neural Networks

Physically Recurrent Neural Networks

Intact constitutive models embedded in an encoder-decoder MLP architecture.

If you have accurate material models at the microscale and would like to perform computational homogenization, those same models can be directly embedded into a hybrid architecture to make macroscale predictions.

Because the models in the architecture are the exact same as in the micromodel, a number of features can be directly inherited and therefore not learned from data:

• Path dependency (loading/unloading/reloading) without training for it
• Strain rate dependency while training with only a single rate
• Consistent step size dependence (independent for inviscid models; correct dependence for viscous models)
• Between $10\times$ and $100\times$ less training data than RNNs for comparable performance

Journal papers and preprints

• MA Maia, IBCM Rocha, P Kerfriden, FP van der Meer (2023), PRNNs for 2D composites, elastoplastic

• MA Maia, IBCM Rocha, FP van der Meer (2024), PRNNs for 3D composites, finite-strain thermoviscoplasticity, creep and fatigue

• N Kovacs, MA Maia, IBCM Rocha, C Furtado, PP Camanho, FP van der Meer (2024), PRNNs for micromodels including distributed cohesive damage

• MA Maia, IBCM Rocha, D Kovacevic, FP van der Meer (2024), Reproducing creep and fatigue experiments in thermoplastics using PRNNs -- COMING SOON

In this package

The code in this package contains a standalone demonstration of PRNNs for a 2D micromodel with $J_2$ elastoplasticity (matrix) and linear elasticity (fibers):

• J2Tensor_vect.py: a simple $J_2$ plasticity model in plane stress. This code comes directly from an FE package, demonstrating how PRNNs can embed existing material models with little to no changes in code;
• prnn.py: A PyTorch network class that implements the PRNN, with single-layer encoder and decoder;
• rnn.py: Implements GRU and LSTM networks with variational Gaussian dropout. For comparing predictions and learning performance with PRNNs;
• utils.py: Implements a custom dataset class for handling stress and strain paths, and a class for training and evaluating networks, with the ability of saving and loading checkpoints;