Prediction of friction stir welding performances of dissimilar AA3003-H12 and C12200-H01 using machine learning algorithms

Abstract

Tests specimens were prepared by friction stir welding of two dissimilar metals aluminum and copper. The specimens were subjected to mechanical tests to calculate the ultimate tensile strength, yield strength, percentage elongation, and impact energy. Four different machine learning algorithms (AdaBoost, CatBoost, Gradient Boosting, and XGBoost) were applied for developing the ML models in predicting the performance parameters such as ultimate strength, yield strength, percentage elongation, and impact energy. Pin type, weld speed, rotational speed, and shoulder diameter were considered as the input parameters for the model. Training, testing, and validation were carried out by considering 60%, 20%, and 20% of the available data respectively. In terms of accuracy (lower MAE, lower RMSE, greater R² value, and lower AAD%), CatBoost model, Gradient Boosting model, and XGBoost model performed better than the AdaBoost model in predicting the ultimate tensile strength, yield strength, percentage elongation, and impact energy. Compared to other models, AdaBoost model has only few hyperparameters for fine-tuning. During hyperparameters tuning, AdaBoost model showed accuracy only within a narrow range of values of features.