A multi-strategy hybrid machine learning model for predicting glass-formation ability of metallic glasses based on imbalanced datasets

Xiaowei Liu,Zhilin Long,Wei Zhang,Lingming Yang,Zhuang Li
DOI: https://doi.org/10.1016/j.jnoncrysol.2023.122645
IF: 4.458
2023-09-24
Journal of Non-Crystalline Solids
Abstract:The glass-forming ability (GFA) is essential to developing and broadly applying metallic glasses. However, existing research rarely considers the imbalance of GFA data, which leads to low prediction accuracy and generalization ability of machine learning (ML) models. In this study, three strategies, a genetic algorithm to select features, a SMOGN technique to enhance data, and a stacking model to train data, are used to solve the problem of low prediction accuracy of unbalanced data. The cross-validated R 2 =0.85 and testing R 2 =0.87 for the stacking model incorporating the three strategies are much higher than the currently reported models. In addition, the ML model reasonably explains how different features affect GFA through the shapely additive explanation (SHAP) method, which deepens the understanding of GFA. This study demonstrates the effectiveness of the proposed strategy for GFA prediction in an imbalanced data environment.
materials science, multidisciplinary, ceramics
What problem does this paper attempt to address?