Development of machine learning regression models for the prediction of tensile strength of friction stir processed AA8090/SiC surface composites

Abstract

Friction Stir Processing is a state-of-the-art technology for microstructure refinement, material property enhancement, and fabrication of surface composites. Machine learning approaches have garnered significant interest as prospective models for modeling various production systems. The present work aims to develop four machine learning models, namely linear regression, support vector regression, artificial neural network and extreme gradient boosting to predict the influence of FSP parameters such as tool rotational speed, tool traverse speed and groove width on ultimate tensile strength of friction stir processed AA8090/SiC surface composites. These models were developed through Python programming and the original dataset was divided into 80% for the training phase and 20% for the testing phase. The performance of the models was evaluated by root mean squared error, mean absolute error and R2. Based on the results and graphical visualization, it was observed that the XGBoost model outperformed other models with high accuracy in predicting UTS of AA8090/SiC surface composites. © 2024 The Author(s). Published by IOP Publishing Ltd.