Skip to main content

ORIGINAL RESEARCH article

Front. Mater.
Sec. Computational Materials Science
Volume 11 - 2024 | doi: 10.3389/fmats.2024.1474609

Enhancing material property prediction with ensemble deep graph convolutional networks

Provisionally accepted
  • West Virginia University, Morgantown, United States

The final, formatted version of the article will be published soon.

    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 (∆E f ), band gap (E g ), density (ρ), equivalent reaction energy per atom (E rxn,atom ), energy per atom (E atom ) and atomic density (ρ atom ) in 33,990 stable inorganic materials. These findings support the broader application of ensemble methods to enhance predictive accuracy in the field.

    Keywords: Material property prediction, Graph neural networks, Ensemble model, prediction ensemble, Model ensemble

    Received: 01 Aug 2024; Accepted: 23 Sep 2024.

    Copyright: © 2024 Rahman, Bhandari, Nasrabadi, Romero and Gyawali. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence: Chowdhury Mohammad Abid Rahman, West Virginia University, Morgantown, United States

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.