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 条
  • [11] Few-shot learning via graph embeddings with convolutional networks for low-data molecular property prediction
    Luis Torres
    Joel P. Arrais
    Bernardete Ribeiro
    Neural Computing and Applications, 2023, 35 : 13167 - 13185
  • [12] Few-shot learning via graph embeddings with convolutional networks for low-data molecular property prediction
    Torres, Luis
    Arrais, Joel P.
    Ribeiro, Bernardete
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (18) : 13167 - 13185
  • [13] Grouping of Maintenance Actions with Deep Reinforcement Learning and Graph Convolutional Networks
    Kerkkamp, David
    Bukhsh, Zaharah A.
    Zhang, Yingqian
    Jansen, Nils
    ICAART: PROCEEDINGS OF THE 14TH INTERNATIONAL CONFERENCE ON AGENTS AND ARTIFICIAL INTELLIGENCE - VOL 2, 2022, : 574 - 585
  • [14] Schema-Aware Deep Graph Convolutional Networks for Heterogeneous Graphs
    Manchanda, Saurav
    Zheng, Da
    Karypis, George
    2021 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2021, : 480 - 489
  • [15] Convolutional Graph Neural Networks
    Gama, Fernando
    Marques, Antonio G.
    Leus, Geert
    Ribeiro, Alejandro
    CONFERENCE RECORD OF THE 2019 FIFTY-THIRD ASILOMAR CONFERENCE ON SIGNALS, SYSTEMS & COMPUTERS, 2019, : 452 - 456
  • [16] Property Prediction of Functional Organic Molecular Crystals with Graph Neural Networks
    O'Connor, Dana
    Buitrago, Paola A.
    PRACTICE AND EXPERIENCE IN ADVANCED RESEARCH COMPUTING 2024, PEARC 2024, 2024,
  • [17] Comparison of Atom Representations in Graph Neural Networks for Molecular Property Prediction
    Pocha, Agnieszka
    Dana, Tomasz
    Podlewska, Sabina
    Tabor, Jacek
    Maziarka, Lukasz
    2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [18] Multi-view attention graph convolutional networks for the host prediction of phages
    Ma, Lijia
    Gao, Peng
    Zhou, Wenxiang
    Lin, Qiuzhen
    Bai, Yuan
    Fang, Min
    Du, Zhihua
    Li, Jianqiang
    KNOWLEDGE-BASED SYSTEMS, 2025, 308
  • [19] Multivariate sequence prediction for graph convolutional networks based on ESMD and transfer entropy
    Li, Xin
    Tang, Guoqiang
    MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (35) : 83493 - 83511
  • [20] Graph neural networks with molecular segmentation for property prediction and structure-property relationship discovery
    Chen, Zhudan
    Li, Dazi
    Liu, Minghui
    Liu, Jun
    COMPUTERS & CHEMICAL ENGINEERING, 2023, 179