A chemically inspired convolutional neural network using electronic structure representation

Dong Hyeon Mok, Daeun Shin, Jonggeol Na, Seoin Back

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

In recent years, the development of appropriate crystal representations for accurate prediction of inorganic crystal properties has been considered as one of the essential tasks to accelerate materials discovery through high-throughput virtual screening (HTVS). However, many of them were developed aiming to predict the properties of the given structures, although property predictions of ground state structures using unrelaxed structures as inputs are much more important in practical HTVS. To tackle this challenge, we develop a chemically inspired convolutional neural network based on convolution block attention modules using the density of states of unrelaxed initial structures (IS-DOS) as inputs. Our model, Electronic Structure Network (ESNet), achieved the highest accuracy for predicting formation energy, proving that IS-DOS is an appropriate input for property prediction and the attention module is capable of properly featurizing DOS signals by capturing the contributions of each spin and orbital state. In addition, we statistically evaluated the stability screening performance of ESNet, measuring the computational cost and capability of materials discovery simultaneously. We found that ESNet outperformed previously reported models and various models with different types of input features and architectures. Indeed, ESNet successfully discovered 926 stable materials from 15 318 unrelaxed structures with 82% reduced computational cost compared to the complete DFT validation.

Original languageEnglish
JournalJournal of Materials Chemistry A
DOIs
StateAccepted/In press - 2023

Bibliographical note

Publisher Copyright:
© 2023 The Royal Society of Chemistry.

Fingerprint

Dive into the research topics of 'A chemically inspired convolutional neural network using electronic structure representation'. Together they form a unique fingerprint.

Cite this