Formation energy prediction of neutral single-atom impurities in 2D materials using tree-based machine learning
Issued Date
2024-09-01
Resource Type
eISSN
26322153
Scopus ID
2-s2.0-85200839296
Journal Title
Machine Learning: Science and Technology
Volume
5
Issue
3
Rights Holder(s)
SCOPUS
Bibliographic Citation
Machine Learning: Science and Technology Vol.5 No.3 (2024)
Suggested Citation
Kesorn A., Hunkao R., Na Talang C., Cholsuk C., Sinsarp A., Vogl T., Suwanna S., Yuma S. Formation energy prediction of neutral single-atom impurities in 2D materials using tree-based machine learning. Machine Learning: Science and Technology Vol.5 No.3 (2024). doi:10.1088/2632-2153/ad66ae Retrieved from: https://repository.li.mahidol.ac.th/handle/20.500.14594/100502
Title
Formation energy prediction of neutral single-atom impurities in 2D materials using tree-based machine learning
Corresponding Author(s)
Other Contributor(s)
Abstract
We applied tree-based machine learning algorithms to predict the formation energy of impurities in 2D materials, where adsorbates and interstitial defects are investigated. Regression models based on random forest, gradient boosting regression, histogram-based gradient-boosting regression, and light gradient-boosting machine algorithms are employed for training, testing, cross validation, and blind testing. We utilized chemical features from fundamental properties of atoms and supplemented them with structural features from the interaction of the added chemical element with its neighboring host atoms via the Jacobi-Legendre (JL) polynomials. Overall, the prediction accuracy yields optimal MAE ≈ 0.518 , RMSE ≈ 1.14 , and R 2 ≈ 0.855 . When trained separately, we obtained lower residual errors RMSE and MAE, and higher R2 value for predicting the formation energy in the adsorbates than in the interstitial defects. In both cases, the inclusion of the structural features via the JL polynomials improves the prediction accuracy of the formation energy in terms of decreasing RMSE and MAE, and increasing R2. This work demonstrates the potential and application of physically meaningful features to obtain physical properties of impurities in 2D materials that otherwise would require higher computational cost.