A machine learning approach to predict tensile strength of TiO2/CeO2-reinforced AA7075 nanocomposites fabricated by ultrasonic-assisted stir casting
摘要
The application of machine learning in the manufacturing sector is gaining attention for improving both material properties and fabrication techniques. In the present study, a machine learning framework has been developed to predict the ultimate tensile strength of ceramic particles-reinforced aluminium metal matrix composites/nanocomposites fabricated via ultrasonic-assisted stir casting. Supervised machine learning models, including Elastic Net, Support Vector Regression, AdaBoost, Decision Tree, Bagging, Random Forest, Extra Trees, Gradient Boosting, and Extra Gradient Boosting, were tested and evaluated. Each ML model was trained using features such as alloy composition, reinforcement characteristics, such as particle size, density and amount, and casting parameters. The evaluation of ML models was conducted in two stages, based on R2–values, mean absolute error, root mean square error, and deviation between predicted UTS and actual UTS of the AA7075 and its nanocomposites fabricated in this work. Among these, the random forest model was selected as the final model, which exhibited the best predictive accuracy, achieving a training R2 value of 0.982, a test R2 of 0.945, and a mean absolute error of 17.601. Bayesian optimization was further employed for hyperparameter tuning to enhance the model's generalization ability. For the final validation of the best-performing ML model, AA7075 nanocomposites were fabricated via ultrasonic-assisted stir casting using TiO2 and CeO2 ceramic nanoparticles as reinforcements. A minimum deviation of 0.65% between the experimental and predicted ultimate tensile strength values indicated strong agreement and confirmed the model’s capability to predict the tensile strength of the AA7075 nanocomposites.
Graphical abstract