No project description provided
Project description
NeuROM, a NN-PGD architecture based on the HiDeNN Framework
NeuROM provides a interpretable Deep Neural Network architecture that uses Tensor decomposition to provide a parametric reduced-order model. This interpretability of the network comes from the HiDeNN architecture that provides an interpolation framework through a deep neural network which weights and biaises are constrained so that the interpolation mnatches a finite element interpolation (P1 or P2). The first hidden layers plays the role of the shape functions while the last layer, called interpolation layer in the reminder of the document, utilises the output of the shape functions to interpolate the output. Training the weights of that last hidden layer is the same as solving a FEM problem on a fixed mesh. The weights of the interpolation layer directly correspond to the nodal values associated to each shape function. Therefore prescibing dirichlet boundary conditions is straight forward by freezing the weights associated to the prescribed values of fixed DoFs. Learning the parameters associated with the first layers however accounts to mesh adaptation.
This code provides an implementation of a HiDeNN. The input of the layer is the coordinate $\underline{x}$ where the output is evaluated and the value of the parameters $\underline{\mu}$ for which the output is computed. In this case the output of the network is the displacement $\underline{u}\left(\underline{x},\underline{\mu}\right)$
Folder's structure
.
├── neurom
│ ├── HiDeNN_PDE.py
| |
│ ├── src
│ | ├── PDE_Library.py
│ | ├── Pre_processing.py
│ | └── Training.py
| |
| ├── Post
| │ └── Plots.py
| |
│ └── main.py
│
├── Geometries
│ ├── Beam.geo
│ └── Hole.geo
│
│
├── Illustrations
│ └── NeuROM_1D.gif
|
│
├── README.md
│
└── TrainedModels
├── ROM_1Para_np_10_nmodes_1_npara_1
└── ROM_1Para_np_50_nmodes_1_npara_1
Architecture of the NN
There are two level of modelling involved. The space interpolation in itself can be trained to get a solution for a given problem. Such interpolation is achieved using the model MeshNN
. Building on that interpolation model and using a similar interpolation for the parametric space, a reduced-order model can e trained as well using the NeuROM
class.
Data entry
A Config
file needs to be filled with the job parameters (name of the geometry, solvers' parameters, post-precessing required, etc.) A default file is specified at the top the main.py
script but any other file can be passed as an argument when running the main script as python main.py -cf 'Configuration/CONFIG_FILE.toml'
.
Pre-processing
Mesh
class that builds the Geometry and the Mesh_object
from the
Inputs:
config["geometry"]["Name"]
:config["interpolation"]["order"]
,config["interpolation"]["dimension"]
The method .AddBCs
then allows to define Dirichlet boundary conditions and specify the volume element of the structure that should be considered.
Inputs:
- List of BCs (Physical surface, value, normal direction)
- volume index
Then the methods
.MeshGeo()
.ReadMesh()
.AssemblyMatrix()
(if 1D) respectively mesh the geometry (if the mesh does not already exist), parse the mesh file, assemble the matrix later used for the assembly of each shape functions
Space interpolation
Given a mesh object created using the Mesh
class, MeshNN
gives a space interpolation of the quantity of interest using first or second order FE shape functions.
- MeshNN class that, given a mesh, "assemble" the shape functions accordingly
model = MeshNN_2D(Mesh_object)
(ormodel = MeshNN_1D(Mesh_object)
in 1D) Creates an interpolation model- Mesh_object being the previously created mesh
Reduced-order modelling
Given a hypercube ParameterHypercube
defining the parametric space, the Space dirichlet boundary conditions a mesh and the maximum number of requested modes, a reduced-order model od the parametric field is built
ROM_model = NeuROM(Mesh_object,ParameterHypercube,ConfigFile)
Illustration of the ROM parametric solution
Training the NN
The Volumic forces are accounted for in the loss function through the right hand side (RHS) function and the loss function is the potential energy.
The trainable parameters can be changed on the fly.
-
model.Freeze_Mesh()
Freezes the mesh so that only the nodale values are trained -
model.UnFreeze_Mesh()
Unfreezes the mesh so that the coordinates values can be trained -
model.Freeze_FEM()
Freezes the nodale values so that only the coordinates are trained -
model.UnFreeze_FEM()
Unfreezes the nodale so that FEM problem can be solved
Reduced-order modelling
The Volumic forces are accounted for in the loss function through the right hand side (RHS) function and the loss function is the potential energy.
The trainable parameters can be changed on the fly.
-
ROM_model.Freeze_Mesh()
Freezes the space mesh so that only the nodale values are trained -
ROM_model.UnFreeze_Mesh()
Unfreezes the space mesh so that the coordinates values can be trained -
ROM_model.Freeze_Space()
Freezes the space nodale values so that only the coordinates are trained -
ROM_model.UnFreeze_Space()
Unfreezes the space nodale so that FEM problem can be solved -
ROM_model.Freeze_MeshPara()
Freezes the parametric mesh so that only the nodale values are trained -
ROM_model.UnFreeze_MeshPara()
Unfreezes the parametric mesh so that the coordinates values can be trained -
ROM_model.Freeze_Para()
Freezes the parametric nodale values so that only the coordinates are trained -
ROM_model.UnFreeze_Para()
Unfreezes the parametric nodale so that FEM problem can be solved
Licensing
- Copyright (C) 2024, UMR 7649, École Polytechnique, IPP, CNRS, MΞDISIM Team, INRIA
- This program is free software: you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation, either version 3 of the License, or
- (at your option) any later version.
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
- You should have received a copy of the GNU General Public License
- along with this program. If not, see https://www.gnu.org/licenses/. */
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for NeuROM_Py-3.1.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 441828c5096f47c983fb447ff4bc564afee025d4b6d6dbad9043abb8d0f42a30 |
|
MD5 | 3ba5b80b3b5d53c1784575df3a5aeec6 |
|
BLAKE2b-256 | d89d0b7039b781a2a0c16bb898c62c489a6865db594ec433485d88ef1f820f12 |