Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/lamm-mit/fieldcompleter

GAN/convolutional and Transformer models to predict missing mechanical information given limited known data in part of the domain, and further characterize the composite geometries from the recovered mechanical fields for 2D and 3D complex microstructures
https://github.com/lamm-mit/fieldcompleter

analysis computer-vision convolutional-neural-networks deep-learning deep-neural-networks design engineering fracture materials materials-informatics non-destructive-testing science strain stress testing transformers

Last synced: 15 days ago
JSON representation

Host: GitHub
URL: https://github.com/lamm-mit/fieldcompleter
Owner: lamm-mit
Created: 2022-10-03T14:27:33.000Z (about 2 years ago)
Default Branch: main
Last Pushed: 2023-04-23T08:13:01.000Z (over 1 year ago)
Last Synced: 2024-05-20T20:08:07.126Z (6 months ago)
Topics: analysis, computer-vision, convolutional-neural-networks, deep-learning, deep-neural-networks, design, engineering, fracture, materials, materials-informatics, non-destructive-testing, science, strain, stress, testing, transformers
Language: Python
Homepage: https://onlinelibrary.wiley.com/doi/10.1002/adma.202301449
Size: 244 KB
Stars: 13
Watchers: 2
Forks: 3
Open Issues: 0
Metadata Files:
- Readme: README.md

Awesome Lists containing this project

README

# FieldCompleter

Z. Yang, M.J. Buehler, “Fill in the Blank: Transferrable Deep Learning Approaches to Recover Missing Physical Field Information,” Adv. Materials, https://doi.org/10.1002/adma.202301449, 2023

Solving materials engineering tasks is often hindered by limited information, such as in inverse problems with only boundary data information or design tasks with a simple objective but a vast search space. To address these challenges, we leverage multiple deep learning (DL) architectures to predict missing mechanical information given limited known data in part of the domain, and further characterize the composite geometries from the recovered mechanical fields for 2D and 3D complex microstructures. In 2D, we utilize a conditional generative adversarial network (GAN) to complete partially masked field maps and predict the composite geometry with convolutional models with great accuracy and generality by making precise predictions on field data with mixed stress/strain components, hierarchical geometries, distinct materials properties and various types of microstructures including ill-posed inverse problems. In 3D, a Transformer-based architecture is implemented to predict complete 3D mechanical fields from input field snapshots. The model manifests excellent performance regardless of microstructural complexity and recovers the entire bulk field even from a single surface field image, allowing internal structural characterization from only boundary measurements. The frameworks provide efficient ways for analysis and design with incomplete information and allow the direct inverse translation from properties back to materials structures.

![Overall workflow](https://github.com/lamm-mit/FieldCompleter/blob/main/Figure.png)

## 2D Mechanical Field Completion using a DeepFill Model
**Working directory**
```
2D_field_completer
```
**Requirements**
```
conda env create -f environment.yml
```
**Dataset**
- Example dataset: Stress field (σ₁₁) in the 2D digital composites with linear elasticity under uniaxial tension.
- The dataset can be found in the following link: https://www.dropbox.com/sh/6zkcrw2xzjtjugc/AACWo-znV2ntQC-zcvPc3KDea?dl=0.

**Training**
- We use transfer learning starting from a pretrained model trained on Places2 dataset (http://places2.csail.mit.edu/download.html).
- The pretrained checkpoint can be found here: https://www.dropbox.com/sh/eiy0n6xjc0e2a05/AADvGvn75n0WObEqBFwEletQa?dl=0.
- The hyperparameters and training details can be modified via the configuration file **train-S11-pretrained.yaml**.
```
python3 train.py --config configs/train-S11-pretrained.yaml
```

**Testing**
- The testing part is stored in **test.ipynb** including 2D field completion and inverse translation from field to geometry.
- The pretrained checkpoints for DeepFill model and CNN model can be found here: https://www.dropbox.com/sh/1d37uqr0nj73ky9/AADMBbRw8iZgLKy2o4fJlfW3a?dl=0. The paths to checkpoints need to be specified in **test.ipynb**.

## 3D Mechanical Field Completion using a Transformer model
**Working directory**
```
3D_field_completer
```
**Requirements**
```
conda env create -f environment.yml
```
**Dataset**
- Example dataset: Stress field (σ₁₁) in the 3D digital composites with linear elasticity under uniaxial compression.
- The dataset can be found in the following link: https://www.dropbox.com/sh/5gntfr7ittue5fh/AACE2D-GOeTHhR2zCMcUCXila?dl=0. **S11.npy** store matrix represent all 3D stress fields. **labels_train.npy** and **labels_test.npy** are train/test sequences representing geometries of 3D composites.

**Training**
- The training starts from scratch.
- The hyperparameters and training details can be modified directly in **vivit.py**.
```
python3 train.py
```

**Testing**
- The testing part is stored in **test.ipynb** including 3D field completion and inverse translation from field to geometry.
- The pretrained checkpoints for ViViT model and CNN model can be found here: https://www.dropbox.com/sh/ulz37l3ang5hfjf/AAB1dr2yX2AJw26bGSE582S4a?dl=0. The paths to checkpoints need to be specified in **test.ipynb**.

```
@article{YangBuehlerAdvMat_2023,
title = {Fill in the Blank: Transferrable Deep Learning Approaches to Recover Missing Physical Field Information},
author = {Z. Yang and M.J. Buehler},
journal = {Advanced Materials},
year = {2023},
volume = {},
pages = {},
url = {https://doi.org/10.1002/adma.202301449}
}
```