The Influence of Heatmap Correlation-based Feature Selection on Predictive Modeling of Low Alloy Steel Mechanical Properties Using Artificial Neural Network (ANN) Algorithm

Abstract

This study aims to evaluate the influence of heatmap correlation-based feature selection on predictive modeling of low alloy steel mechanical properties using an artificial neural network (ANN) algorithm. Heatmap correlation was used to determine the chemical elements most correlated to the low alloy steel mechanical properties, such as Yield strength (YS) and Tensile strength (TS). There were 15 input variables of chemical elements in this study, and after feature selection, 11 input variables were obtained for YS, and 13 input variables were obtained for TS. The ANN model was validated using K-fold 10 cross-validation and evaluated using loss metric, Mean Absolute Error (MAE), and Root Mean Square Error (RMSE). The results showed that modeling with feature selection was able to improve the YS prediction, with a decrease in value of 6.83% in MAE and 4.97% in RMSE, while the TS prediction decreased by 16.46% in MAE and 18.34% in RMSE after feature selection. These results indicate that the use of feature selection provides better performance compared to the model without feature selection, and heatmap correlation can be used as an alternative to improve model performance in predictive modeling of low alloy steel mechanical properties using the ANN algorithm.