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 条
  • [31] Recurrent Graph Convolutional Networks for Spatiotemporal Prediction of Snow Accumulation Using Airborne Radar
    Zalatan, Benjamin
    Rahnemoonfar, Maryam
    2023 IEEE RADAR CONFERENCE, RADARCONF23, 2023,
  • [32] Variational Graph Convolutional Networks for Dynamic Graph Representation Learning
    Mir, Aabid A.
    Zuhairi, Megat F.
    Musa, Shahrulniza
    Alanazi, Meshari H.
    Namoun, Abdallah
    IEEE ACCESS, 2024, 12 : 161697 - 161717
  • [33] Deep Graph-Convolutional Image Denoising
    Valsesia, Diego
    Fracastoro, Giulia
    Magli, Enrico
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 8226 - 8237
  • [34] A Systematic Comparison Study on Hyperparameter Optimisation of Graph Neural Networks for Molecular Property Prediction
    Yuan, Yingfang
    Wang, Wenjun
    Pang, Wei
    PROCEEDINGS OF THE 2021 GENETIC AND EVOLUTIONARY COMPUTATION CONFERENCE (GECCO'21), 2021, : 386 - 394
  • [35] Enhancing Road Safety Through Accurate Detection of Hazardous Driving Behaviors With Graph Convolutional Recurrent Networks
    Khosravinia, Pooyan
    Perumal, Thinagaran
    Zarrin, Javad
    IEEE ACCESS, 2023, 11 : 52983 - 52995
  • [36] Adaptative Scaler-Moment Crystal Graph Attention Neural Network for Material Property Prediction
    Zhang, Weiwei
    Tang, Lixin
    Xu, Meiling
    IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTATIONAL INTELLIGENCE, 2024,
  • [37] GCN-WP - Semi-Supervised Graph Convolutional Networks for Win Prediction in Esports
    Bisberg, Alexander J.
    Ferrara, Emilio
    2022 IEEE CONFERENCE ON GAMES, COG, 2022, : 449 - 456
  • [38] Spam Review Detection with Graph Convolutional Networks
    Li, Ao
    Qin, Zhou
    Liu, Runshi
    Yang, Yiqun
    Li, Dong
    PROCEEDINGS OF THE 28TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT (CIKM '19), 2019, : 2703 - 2711
  • [39] Transfer Entropy in Graph Convolutional Neural Networks
    Moldovan, Adrian
    Cataron, Angel
    Andonie, Azvan
    2024 28TH INTERNATIONAL CONFERENCE INFORMATION VISUALISATION, IV 2024, 2024, : 207 - 213
  • [40] Gated Graph Convolutional Recurrent Neural Networks
    Ruiz, Luana
    Gama, Fernando
    Ribeiro, Alejandro
    2019 27TH EUROPEAN SIGNAL PROCESSING CONFERENCE (EUSIPCO), 2019,