GrainGNN: A dynamic graph neural network for predicting 3D grain microstructure

Yigong Qin,Stephen DeWitt,Balasubramaniam Radhakrishnan,George Biros
2024-02-01
Abstract:We propose GrainGNN, a surrogate model for the evolution of polycrystalline grain structure under rapid solidification conditions in metal additive manufacturing. High fidelity simulations of solidification microstructures are typically performed using multicomponent partial differential equations (PDEs) with moving interfaces. The inherent randomness of the PDE initial conditions (grain seeds) necessitates ensemble simulations to predict microstructure statistics, e.g., grain size, aspect ratio, and crystallographic orientation. Currently such ensemble simulations are prohibitively expensive and surrogates are necessary.
Computational Engineering, Finance, and Science
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to efficiently predict the evolution of multi - grain structures under rapid solidification conditions during metal additive manufacturing. Specifically, although high - fidelity simulations (such as phase - field simulations) can accurately predict solidification microstructures, they are extremely computationally costly and require a large amount of time and computational resources. Therefore, the paper proposes a dynamic graph neural network model named GrainGNN as an alternative to high - fidelity simulations to significantly reduce computational costs while maintaining high prediction accuracy. ### Problem Background 1. **Limitations of High - Fidelity Simulations**: - High - fidelity simulations usually use multi - component partial differential equations (PDEs) to describe phase transitions and interface movements during solidification. - Due to the randomness of initial conditions (such as the position of grain seeds), multiple simulations are required to statistically obtain key parameters such as grain size, aspect ratio, and crystallographic orientation. - Such simulation calculations are very costly, especially in large - scale additive manufacturing processes where a large number of molten pool simulations need to be processed. 2. **Requirement for Alternative Models**: - In order to reduce computational costs, an efficient alternative model needs to be developed that can significantly increase the computational speed while ensuring prediction accuracy. - The alternative model needs to be able to handle complex grain growth and coarsening processes, including interface motion and topological changes. ### Main Contributions of the GrainGNN Model 1. **Graph Representation**: - Use a graph structure to represent the grain microstructure, where nodes in the graph represent grain and grain - boundary intersections, and edges represent the connection relationships between grains. - Compress the spatial representation through hand - designed features (such as grain size, boundary length, etc.) to reduce computational complexity. 2. **Graph Neural Network**: - Use two neural networks to predict the evolution of graph features and topological events of grains: - A regression network is used to predict interface motion. - A classification network is used to estimate the probability of topological change events. - The network adopts an encoder - decoder architecture, where the encoder uses a multi - layer Transformer LSTM and the decoder uses a single - layer perceptron. 3. **Graph Update Algorithm**: - Propose a graph update algorithm that combines network outputs to evolve the topological structure of the microstructure graph. - The time complexity of this algorithm is \( O(\text{#grains}) \) and it can handle unseen \( G \), \( R \), domain sizes, grain size distributions, and a larger number of grains. 4. **Performance Evaluation**: - Verify the prediction accuracy of GrainGNN by comparing it with high - fidelity phase - field simulations. - The results show that GrainGNN achieves 80% - 90% point - to - point accuracy and is almost consistent with phase - field simulations in the distribution of key parameters. - In terms of computational speed, the inference speed of GrainGNN is 150 to 2000 times faster than that of high - fidelity phase - field simulations. ### Summary GrainGNN has successfully solved the high - computational - cost problem of grain microstructure prediction in the metal additive manufacturing process by using graph neural networks and an efficient graph update algorithm, providing a feasible solution for practical applications.