Classification of Cast Iron Alloys through Convolutional Neural Networks Applied on Optical Microscopy Images

Marta Bárcena,Lara Lloret Iglesias,Diego Ferreño,Isidro Carrascal
DOI: https://doi.org/10.1002/srin.202400120
IF: 2.1258
2024-08-31
steel research international
Abstract:This study introduces an innovative method for classifying cast iron alloys. Leveraging convolutional neural networks, including MobileNet and U‐Net, it automates the traditional, subjective process of graphite morphology classification, achieving remarkable accuracies. This advancement promises a revolution in material selection, propelled by a synthetic dataset tailored to the ISO 945‐1:2019 standard, paving the way for more efficient industrial applications. Classification of cast iron alloys based on graphite morphology plays a crucial role in materials science and engineering. Traditionally, this classification has relied on visual analysis, a method that is not only time‐consuming but also suffers from subjectivity, leading to inconsistencies. This study introduces a novel approach utilizing convolutional neural networks—MobileNet for image classification and U‐Net for semantic segmentation—to automate the classification process of cast iron alloys. A significant challenge in this domain is the limited availability of diverse and comprehensive datasets necessary for training effective machine learning models. This is addressed by generating a synthetic dataset, creating a rich collection of 2400 pure and 1500 mixed images based on the ISO 945‐1:2019 standard. This ensures a robust training process, enhancing the model's ability to generalize across various morphologies of graphite particles. The findings showcase a remarkable accuracy in classifying cast iron alloys (achieving an overall accuracy of 98.9 ± 0.4%—and exceeding 97% for all six classes—for classification of pure images and ranging between 84% and 93% for semantic segmentation of mixed images) and also demonstrate the model's ability to consistently identify and graphite morphology with a level of precision and speed unattainable through manual methods.
metallurgy & metallurgical engineering
What problem does this paper attempt to address?