Assessment of the Quality and Mechanical Parameters of Castings Using Machine Learning Methods

Abstract

The aim of the work is to investigate the effectiveness of selected classification algorithms and their extensions in assessing microstructure of castings. Experiments were carried out in which the prepared algorithms and machine learning methods were tested in various conditions and configurations, as well as for various input data, which are photos of castings (photos of the microstructure) or information about the material (e.g., type, composition). As shown by the literature review, there are few scientific papers on this subject (i.e., in the use of machine learning to assess the quality of the microstructure and the obtained strength properties of cast iron). The effectiveness of machine learning algorithms in assessing the quality of castings will be tested using the most universal methods. Results obtained by classic machine learning methods and by neural networks will be compared with each other, taking into account aspects such as interpretability of results, ease of model implementation, algorithm simplicity, and learning time.