Enhancing material property prediction with ensemble deep graph convolutional networks

被引:1
作者
Rahman, Chowdhury Mohammad Abid [1 ]
Bhandari, Ghadendra [2 ]
Nasrabadi, Nasser M. [1 ]
Romero, Aldo H. [2 ]
Gyawali, Prashnna K. [1 ]
机构
[1] West Virginia Univ, Lane Dept Comp Sci & Elect Engn, Morgantown, WV 26506 USA
[2] West Virginia Univ, Dept Phys & Astron, Morgantown, WV USA
来源
FRONTIERS IN MATERIALS | 2024年 / 11卷
基金
美国国家科学基金会;
关键词
material property prediction; graph neural networks; ensemble model; prediction ensemble; model ensemble;
D O I
10.3389/fmats.2024.1474609
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Machine learning (ML) models have emerged as powerful tools for accelerating materials discovery and design by enabling accurate predictions of properties from compositional and structural data. These capabilities are vital for developing advanced technologies across fields such as energy, electronics, and biomedicine, potentially reducing the time and resources needed for new material exploration and promoting rapid innovation cycles. Recent efforts have focused on employing advanced ML algorithms, including deep learning-based graph neural networks, for property prediction. Additionally, ensemble models have proven to enhance the generalizability and robustness of ML and Deep Learning (DL). However, the use of such ensemble strategies in deep graph networks for material property prediction remains underexplored. Our research provides an in-depth evaluation of ensemble strategies in deep learning-based graph neural network, specifically targeting material property prediction tasks. By testing the Crystal Graph Convolutional Neural Network (CGCNN) and its multitask version, MT-CGCNN, we demonstrated that ensemble techniques, especially prediction averaging, substantially improve precision beyond traditional metrics for key properties like formation energy per atom ( Delta E f ) , band gap ( E g ) , density ( rho ) , equivalent reaction energy per atom ( E rxn,atom ) , energy per atom ( E atom ) and atomic density ( rho atom ) in 33,990 stable inorganic materials. These findings support the broader application of ensemble methods to enhance predictive accuracy in the field.
引用
收藏
页数:14
相关论文
共 50 条
[41]   Graph Neural Networks With Convolutional ARMA Filters [J].
Bianchi, Filippo Maria ;
Grattarola, Daniele ;
Livi, Lorenzo ;
Alippi, Cesare .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (07) :3496-3507
[42]   GRAPH-TIME CONVOLUTIONAL NEURAL NETWORKS [J].
Isufi, Elvin ;
Mazzola, Gabriele .
2021 IEEE DATA SCIENCE AND LEARNING WORKSHOP (DSLW), 2021,
[43]   FAST GRAPH CONVOLUTIONAL RECURRENT NEURAL NETWORKS [J].
Kadambari, Sai Kiran ;
Chepuri, Sundeep Prabhakar .
CONFERENCE RECORD OF THE 2019 FIFTY-THIRD ASILOMAR CONFERENCE ON SIGNALS, SYSTEMS & COMPUTERS, 2019, :467-471
[44]   Graph convolutional neural networks via scattering [J].
Zou, Dongmian ;
Lerman, Gilad .
APPLIED AND COMPUTATIONAL HARMONIC ANALYSIS, 2020, 49 (03) :1046-1074
[45]   GRAPH CONVOLUTIONAL NEURAL NETWORKS IN THE COMPANION MODEL [J].
Shi, John ;
Chaudhari, Shreyas ;
Moura, Jose M. E. .
2024 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2024), 2024, :7045-7049
[46]   EEvoU-Net: An ensemble of evolutionary deep fully convolutional neural networks for medical image segmentation [J].
Hassanzadeh, Tahereh ;
Essam, Daryl ;
Sarker, Ruhul .
APPLIED SOFT COMPUTING, 2023, 143
[47]   Prediction of pitch type and location in baseball using ensemble model of deep neural networks [J].
Lee, Jae Sik .
JOURNAL OF SPORTS ANALYTICS, 2022, 8 (02) :115-126
[48]   Enhancing material property prediction via a dual stream model fusing spatial and topological information [J].
Xie, Jiangnan ;
Zheng, Xiaolong ;
Ying, Zhihua ;
Zheng, Hui ;
Zheng, Liang .
COMPUTATIONAL MATERIALS SCIENCE, 2025, 250
[49]   Scalable Graph Neural Networks with Deep Graph Library [J].
Zheng, Da ;
Wang, Minjie ;
Gan, Quan ;
Song, Xiang ;
Zhang, Zheng ;
Karypis, Geroge .
WSDM '21: PROCEEDINGS OF THE 14TH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2021, :1141-1142
[50]   Learning Graph Neural Networks with Deep Graph Library [J].
Zheng, Da ;
Wang, Minjie ;
Gan, Quan ;
Zhang, Zheng ;
Karypis, George .
WWW'20: COMPANION PROCEEDINGS OF THE WEB CONFERENCE 2020, 2020, :305-306