Multi-task graph neural networks for simultaneous prediction of global and atomic properties in ferromagnetic systems

02/04/2022
by   Massimiliano Lupo Pasini, et al.
0

We introduce a multi-tasking graph convolutional neural network, HydraGNN, to simultaneously predict both global and atomic physical properties and demonstrate with ferromagnetic materials. We train HydraGNN on an open-source ab initio density functional theory (DFT) dataset for iron-platinum (FePt) with a fixed body centered tetragonal (BCT) lattice structure and fixed volume to simultaneously predict the mixing enthalpy (a global feature of the system), the atomic charge transfer, and the atomic magnetic moment across configurations that span the entire compositional range. By taking advantage of underlying physical correlations between material properties, multi-task learning (MTL) with HydraGNN provides effective training even with modest amounts of data. Moreover, this is achieved with just one architecture instead of three, as required by single-task learning (STL). The first convolutional layers of the HydraGNN architecture are shared by all learning tasks and extract features common to all material properties. The following layers discriminate the features of the different properties, the results of which are fed to the separate heads of the final layer to produce predictions. Numerical results show that HydraGNN effectively captures the relation between the configurational entropy and the material properties over the entire compositional range. Overall, the accuracy of simultaneous MTL predictions is comparable to the accuracy of the STL predictions. In addition, the computational cost of training HydraGNN for MTL is much lower than the original DFT calculations and also lower than training separate STL models for each property.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
09/20/2021

Prediction of properties of metal alloy materials based on machine learning

Density functional theory and its optimization algorithm are the main me...
research
03/11/2020

Global Attention based Graph Convolutional Neural Networks for Improved Materials Property Prediction

Machine learning (ML) methods have gained increasing popularity in explo...
research
11/15/2018

Predicting thermoelectric properties from crystal graphs and material descriptors - first application for functional materials

We introduce the use of Crystal Graph Convolutional Neural Networks (CGC...
research
06/02/2020

Careful analysis of XRD patterns with Attention

The important peaks related to the physical properties of a lithium ion ...
research
08/02/2022

Curvature-informed multi-task learning for graph networks

Properties of interest for crystals and molecules, such as band gap, ela...
research
11/14/2018

MT-CGCNN: Integrating Crystal Graph Convolutional Neural Network with Multitask Learning for Material Property Prediction

Developing accurate, transferable and computationally inexpensive machin...
research
06/04/2021

Detect the Interactions that Matter in Matter: Geometric Attention for Many-Body Systems

Attention mechanisms are developing into a viable alternative to convolu...

Please sign up or login with your details

Forgot password? Click here to reset