Machine Learning Approaches for Classification of Composite Materials
Dmytro Tymoshchuk, Iryna Didych, Pavlo Maruschak, Oleh Yasniy, Andrii Mykytyshyn, Mykola Mytnyk
Abstract
The paper presents a comparative analysis of various machine learning algorithms for the classification of epoxy composites reinforced with basalt fiber and modified with inorganic fillers. The classification is based on key thermophysical characteristics, in particular, the mass fraction of the filler, temperature, and thermal conductivity coefficient. A dataset of 16,056 interpolated samples was used to train and evaluate more than a dozen models. Among the tested algorithms, the MLP neural network model showed the highest accuracy of 99.7% and balanced classification metrics F1-measure and G-Mean. Ensemble methods, including XGBoost, CatBoost, ExtraTrees, and HistGradientBoosting, also showed high classification accuracy. To interpret the results of the MLP model, SHAP analysis was applied, which confirmed the predominant influence of the mass fraction of the filler on decision-making for all classes. The results of the study confirm the high effectiveness of machine learning methods for recognizing filler type in composite materials, as well as the potential of interpretable AI in materials science tasks.