Decision Support Tool in the Selection of Powder for 3D Printing

Materials (Basel). 2024 Apr 18;17(8):1873. doi: 10.3390/ma17081873.

Abstract

The work presents a tool enabling the selection of powder for 3D printing. The project focused on three types of powders, such as steel, nickel- and cobalt-based and aluminum-based. An important aspect during the research was the possibility of obtaining the mechanical parameters. During the work, the possibility of using the selected algorithm based on artificial intelligence like Random Forest, Decision Tree, K-Nearest Neighbors, Fuzzy K-Nearest Neighbors, Gradient Boosting, XGBoost, AdaBoost was also checked. During the work, tests were carried out to check which algorithm would be best for use in the decision support system being developed. Cross-validation was used, as well as hyperparameter tuning using different evaluation sets. In both cases, the best model turned out to be Random Forest, whose F1 metric score is 98.66% for cross-validation and 99.10% after tuning on the test set. This model can be considered the most promising in solving this problem. The first result is a more accurate estimate of how the model will behave for new data, while the second model talks about possible improvement after optimization or possible overtraining to the parameters.

Keywords: 3D printing; K-nearest neighbors; XGBoost; decision tree; fuzzy K-nearest neighbors; gradient boosting; machine learning algorithms; random forest.

Grants and funding

This research was funded by DWD/3/34/2019 concluded between the State Treasury—Minister of Science and Higher Education and the AGH University of Science and Technology. Stanisław Staszic in Krakow.