Prediction of electronic work function of the second phase in binary magnesium alloy based on machine learning method

In this paper, 150 pieces of the work function of Mg-based solid solutions and Mg-containing second phases calculated by our group based on density functional theory (DFT) were collected to build a data set. According to the analysis of Pearson and Spearman correlation, 14 features were used as inpu...

Full description

Saved in:
Bibliographic Details
Published inJournal of materials research Vol. 37; no. 21; pp. 3792 - 3802
Main Authors Wei, Xiaoxiu, Wang, Jianfeng, Wang, Chao, Zhu, Shijie, Wang, Liguo, Guan, Shaokang
Format Journal Article
LanguageEnglish
Published Cham Springer International Publishing 14.11.2022
Springer Nature B.V
Subjects
Online AccessGet full text

Cover

Loading…
More Information
Summary:In this paper, 150 pieces of the work function of Mg-based solid solutions and Mg-containing second phases calculated by our group based on density functional theory (DFT) were collected to build a data set. According to the analysis of Pearson and Spearman correlation, 14 features were used as input variables. Four machine learning (ML) models including multiple linear regression, support vector regression (SVR), gradient boosting regression tree, extreme gradient boosting tree were established to predict the work function of second phases. The result shows that the SVR model has the highest accuracy and best generalization ability to predict the work function. The work function of Mg 7 Zn 3 phase predicted from the SVR model is very close to that calculated from DFT, which suggests that ML is an effective method to predict the work function in Mg alloy. Graphical abstract
ISSN:0884-2914
2044-5326
DOI:10.1557/s43578-022-00752-6