Harnessing machine learning for predicting mechanical properties of lightweight Mg alloys

Sandeep Jain,Reliance Jain,Mahesh Patel,Baidehish Sahoo,Ayan Bhowmik
DOI: https://doi.org/10.1016/j.matlet.2024.137597
IF: 3
2024-10-25
Materials Letters
Abstract:In this study, we introduce a machine learning methodology for predicting mechanical properties in Mg-based multicomponent alloys, incorporating alloying elements and processing methods as input variables. We employed eight ML models and assessed their efficacy using metrics such as R 2 , RMSE, and MAE after using five-fold cross validation grid search hyperparameters tuning process. Following this, we implemented the top-performing Extra Tree (ET), Random Forest (RF) and XGB models to predict mechanical properties for Mg alloys. The best performance was observed with an R 2 of 95.1 % and 97.2 %, RMSE of 7 % and 8 %, and MAE of 4.7 % and 5 % for UTS and YS respectively using the Extra Tree model. This research not only showcases the efficiency of ML techniques with minimal intervention but also offers valuable insights paving the way for accelerated design of Mg-alloys for tailored application in the future.
materials science, multidisciplinary,physics, applied
What problem does this paper attempt to address?