Ensemble Learning for Comprehensive Hardness Prediction of High-Entropy Alloys
摘要
Due to the vast compositional space, conventional “trial-and-error” and computational methods struggle to effectively uncover the composition-property relationships in high-entropy alloys (HEAs). In this study, we proposed an ensemble averaging-based machine learning (ML) strategy to predict the hardness of a diverse range of HEAs. Initially, we constructed a composition-hardness database of HEAs comprising over nine hundred data points, covering nearly thirty elements. Six key features affecting the hardness of HEAs were identified through a three-step feature selection process. A highly reliable ensemble model was subsequently developed to predict the hardness of HEAs. Furthermore, SHapley Additive exPlanation (SHAP) analysis was conducted to interpret the contributions of the selected features to the hardness of HEAs. This work presents an efficient approach for establishing high-predictability and generalizability ML models using an ensemble learning strategy, demonstrating promising potential in developing advanced HEAs.
Graphical Abstract