Grain Knowledge Graph Representation Learning: A New Paradigm for Microstructure-Property Prediction

Abstract

The mesoscopic structure significantly affects the properties of polycrystalline materials. Current artificial-based microstructure-performance analyses are expensive and require rich expert knowledge. Recently, some machine learning models have been used to predict the properties of polycrystalline materials. However, they cannot capture the complex interactive relationship between the grains in the microstructure, which is a crucial factor affecting the material’s macroscopic properties. Here, we propose a grain knowledge graph representation learning method. First, based on the polycrystalline structure, an advanced digital representation of the knowledge graph is constructed, embedding ingenious knowledge while completely restoring the polycrystalline structure. Then, a heterogeneous grain graph attention model (HGGAT) is proposed to realize the effective high-order feature embedding of the microstructure and to mine the relationship between the structure and the material properties. Through benchmarking with other machine learning methods on magnesium alloy datasets, HGGAT consistently demonstrates superior accuracy on different performance labels. The experiment shows the rationality and validity of the grain knowledge graph representation and the feasibility of this work to predict the material’s structural characteristics.