An Improved Random Forest Magnesium Alloy Prediction Method Based on Particle Swarm Optimization

Yu Zhang,Shi Cheng Wei,Bo Wang,Lin Wei Li
DOI: https://doi.org/10.4028/p-l251t7
2023-05-19
Materials Science Forum
Abstract:Publication date: 18 May 2023 Source: Materials Science Forum Vol. 1088 Author(s): Yu Zhang, Shi Cheng Wei, Bo Wang, Lin Wei Li With the development of the material databases' construction, the use of machine learning methods to process data mining to discover new materials has gradually become a hot topic. The mechanical properties of Mg alloys are related to their components and processing technologies, therefore, it is possible to build prediction model between components, processing technologies and mechanical properties. In order to improve the design efficiency of Mg alloys, using machine learning methods to build a prediction model for the mechanical properties of Mg alloys is of vital importance. To achieve efficient material design, this paper proposed an improved random forest (RF) method based on the Particle Swarm Optimization (PSO) algorithm, and built a Mg alloy performance prediction model. Experiments showed that the accuracy was greatly improved compared with the original RF model, and the prediction accuracy of mechanical properties can reach more than 90%.
What problem does this paper attempt to address?