A novel feature engineering approach for predicting melt pool depth during LPBF by machine learning models

Abstract

Melt pool geometry is a deterministic factor affecting the characteristics of metal Additive Manufacturing (AM) components. The wide array of physical and thermal phenomena involved during the formation of the AM melt pool, along with the great variety of alloy compositions and AM methods, coupled with the clear influence of multiple process parameters, make it difficult to predict the melt pool geometry under a given set of conditions. Therefore, using Artificial Intelligence (AI) approaches such as Machine Learning (ML) is necessary for accurate predictions. Using a physics-informed feature selection strategy along with the application of atomic features for the first time, this work aims to offer accurately trained models relying on existing high-fidelity data for most common alloys in AM academia and industry, i.e., 316 L stainless steel, Ti6Al4V, and AlSi10Mg. Multiple ML algorithms were trained, and the results revealed that the average R² and RMSE obtained by the K-fold cross-validation (K = 5) were significantly enhanced when laser and material properties, inspired by the analytical models for AM melt pool geometry, were used as the model features. Removing the excess features and applying atomic features further enhanced the accuracy of the models. As a result, R² for the XGBoost, CatBoost, and GPR models were 0.907, 0.889, and 0.882, respectively, while the hold-out cross-validation led to 0.978, 0.976, and 0.945, respectively. Furthermore, the results showed that the XGBoost model outperforms the Rosenthal equation. This approach provides a pathway to more accurately predict the properties of metal AM components.