arrow
Return

Structure-aware graph neural network based deep transfer learning framework for enhanced predictive analytics on diverse materials datasets

delete2024-01-02
delete11
delete
OA
AI
V
Vishu Gupta
K
Kamal Choudhary
B
Brian DeCost
F
Francesca Tavazza
C
Carelyn E. Campbell
廖维 cover
廖维 (Wei‐keng Liao)
A
Alok Choudhary
A
Ankit Agrawal *
DOI:10.1038/s41524-023-01185-3delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Modern data mining methods have demonstrated effectiveness in comprehending and predicting materials properties. An essential component in the process of materials discovery is to know which material(s) will possess desirable properties. For many materials properties, performing experiments and density functional theory computations are costly and time-consuming. Hence, it is challenging to build accurate predictive models for such properties using conventional data mining methods due to the small amount of available data. Here we present a framework for materials property prediction tasks using structure information that leverages graph neural network-based architecture along with deep-transfer-learning techniques to drastically improve the model's predictive ability on diverse materials (3D/2D, inorganic/organic, computational/experimental) data. We evaluated the proposed framework in cross-property and cross-materials class scenarios using 115 datasets to find that transfer learning models outperform the models trained from scratch in 104 cases, i.e., approximate to 90%, with additional benefits in performance for extrapolation problems. We believe the proposed framework can be widely useful in accelerating materials discovery in materials science.

Journal

npj Computational Materials cover
npj Computational Materials
IF:
11.9
Papers:
2.4K
Citations:
1.7W

Organization

N
national institute of standards & technology (nist) - usa
Scholars:
9.7K
Papers: 9.0K
Citations: 4
N
Northwestern University
Scholars:
6.1W
Papers: 5.3W
Citations: 3.9K