A General Framework Combining Generative Adversarial Networks and Mixture Density Networks for Inverse Modeling in Microstructural Materials Design
This software is an application of a general framework combining generative adversarial networks and mixture density networks for inverse modeling in microstructural materials design. The efficacy of the proposed approach is tested on a case study where material microstructure is design given a desired optical matieral property.
To use this software, what the algorithm requires as input is a float number from 0 to 1 representing the value for optical matrial property. Then the mixture density networks will sample various latent variable vectors, following by generative adversarial networks to produce designed microstructure images. The detailed drscription about architecture and training processing of the proposed framework can be found in the published paper given below.
- Python 3.6.3
- Numpy 1.18.1
- Sklearn 0.20.0
- Keras 2.3.1
- Pickle 4.0
- TensorFlow 2.1.0
- h5py 2.9.0
- Scipy 1.2.0
model
: This folder contains scripts and saved model for mixture density networks and generative adversarial networks.data
: This folder contains two example datasets.
- Run commend below, which uses trained mixture density networks to sample various latent variable vectors given a desired optical material property.
python MDN.py
- Run commend below, which uses sampled latent variable vectors to produce designed material microstructures.
python ScalableG.py
The Rigorous Couple Wave Analysis simulation is supported by Prof. Cheng Sun’s lab at Northwestern University. This work is supported in part by the following grants: NIST award 70NANB19H005; DOE awards DE-SC0014330, DE-SC0019358.
Zijiang Yang, Dipendra Jha, Arindam Paul, Wei-keng Liao, Alok Choudhary, Ankit Agrawal. "A General Framework Combining Generative Adversarial Networks and Mixture Density Networks for Inverse Modeling in Microstructural Materials Design." Accepted by Workshop on machine learning for engineering modeling, simulation and design @ NeurIPS 2020
Zijiang Yang [email protected]; Ankit Agrawal [email protected]