Evaluating Generalized Feature Importance via Performance Assessment of Machine Learning Models for Predicting Elastic Properties of Materials

26 June 2023, Version 2
This content is a preprint and has not undergone peer review at the time of posting.

Abstract

Identifying key descriptors and understanding important features across different classes of materials are crucial for machine learning (ML) tools to both predict material properties and reveal the physics underlying any process of interest. Traditionally, the predictive modeling of elastic properties of materials is limited to only a few classes of materials and a small set of ML tools despite the broad applications of these materials. Users now have a broad choice of ML models ranging from simple regression models to graph neural networks (GNN) for predicting structure-property relationships. While in recent years, GNNs have outshined traditional ML models in terms of predictability, their intensive data requirement and lack of interpretability may limit practical applicability. Here, we develop a domain-segmented feature space using a diverse set of material attributes and perform a predictive analysis using state-of-the-art ML tools using elastic modulus prediction as a representative example. By deducing a model-independent overall ranking based on feature importance learned by each model, the knowledge is then transferred to GNNs. Our findings indicate a threshold limit on the predictability of traditional ML models, but our approach of transferring task-specific feature importance knowledge to the GNNs can enhance their performance by reducing their data requirement while retaining considerable accuracy.

Keywords

Machine Learning
Elastic Properties
Feature Importance
SHAP
mRMR
Graph Neural Network

Supplementary materials

Title
Description
Actions
Title
Supplementary Information for Evaluating Generalized Feature Importance via Performance Assessment of Machine Learning Models for Predicting Elastic Properties of Materials
Description
Supplementary Information
Actions

Supplementary weblinks

Comments

Comments are not moderated before they are posted, but they can be removed by the site moderators if they are found to be in contravention of our Commenting Policy [opens in a new tab] - please read this policy before you post. Comments should be used for scholarly discussion of the content in question. You can find more information about how to use the commenting feature here [opens in a new tab] .
This site is protected by reCAPTCHA and the Google Privacy Policy [opens in a new tab] and Terms of Service [opens in a new tab] apply.