Predicting the tensile properties of heat treated and non-heat treated LPBFed AlSi10Mg alloy using machine learning regression algorithms

Abstract

In this study, the ability of machine learning algorithms to predict tensile properties of both heat-treated and non-heat treated LPBFed AlSi10Mg alloy is investigated. The data was analyzed using various Machine Learning Regression (MLR) models such as Linear Regression (LR), Gaussian Process Regression (GPR), Random Forest Regression (RFR), and Decision Tree (DT). The AlSi10Mg alloys, heat-treated and non heat-treated, had different tensile characteristics. The tensile characteristics were forecasted using trained and evaluated MLR models. Because the performance of various MLR models has been verified by several performance indicators, such as Root Mean Square Error (RMSE), R² (coefficient of determination), Mean Square Error (MSE), and Mean Absolute Error (MAE). Moreover, scatter plots were made for checking the accuracy of the forecast. The GPR model demonstrated better prediction performance than the other three models, i.e., higher R² values and lower error values for the heat-treated samples. For predicting the UTS value of non-heat treated samples, the LR model performs very well with R² of 1.000. In that case, GPR has the better predictive performance for the other tensile features in non-heat treated samples. Summing up, it is obvious that GPR is well capable of predicting tensile properties of AlSi10Mg alloy with high precision. This indicates how important GPR is to additive manufacturing to achieve great quality.