Abstract
Defect detection plays a vital role in ensuring product quality and safety within industrial casting processes. In these dynamic environments, the occasional emergence of new defects in the production line poses a significant challenge for supervised methods. We present a defect detection framework to effectively detect novel defect patterns without prior exposure during training. Our method is based on contrastive learning applied to the Faster R-CNN model, enhanced with a contrastive head to obtain discriminative representations of different defects. By training on an diverse and comprehensive labeled dataset, our method achieves comparable performance to the supervised baseline model, showcasing commendable defect detection capabilities. To evaluate the robustness of our approach, we authentically replicate a real-world use case by deliberately excluding several defect types from the training data. Remarkably, in this new context, our proposed method significantly improves detection performance of the baseline model, particularly in situations with very limited training data, showcasing a remarkable 34.7% enhancement. Our research highlights the potential of the proposed method in real-world environments where the number of available images may be limited or inexistent. By providing valuable insights into defect detection in challenging scenarios, our framework could contribute to ensuring efficient and reliable product quality and safety in industrial manufacturing processes.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Rafiei, M., Raitoharju, J., Iosifidis, A.: Computer vision on X-ray data in industrial production and security applications: a comprehensive survey. IEEE Access 11, 2445–2477 (2023)
Chandola, V., Banerjee, A., Kumar, V.: Anomaly detection: a survey. ACM Comput. Surv. (CSUR) 41(3), 1–58 (2009)
Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. Commun. ACM 60(6), 84–90 (2017)
Simonyan, K., Zisserman, A.: Very Deep Convolutional Networks for Large-Scale Image Recognition (2015)
Köhler, M., Eisenbach, M., Gross, H.-M.: Few-Shot Object Detection: A Comprehensive Survey (2022)
Da Silva, R.R., Mery, D.: Accuracy estimation of detection of casting defects in X-ray images using some statistical techniques. Insight - Non-Destructive Test. Condition Monit. 49(10), 603–609 (2007)
Mery, D., Arteta, C.: Automatic defect recognition in X-ray testing using computer vision. In: 2017 IEEE Winter Conference on Applications of Computer Vision (WACV), Santa Rosa, CA, USA, pp. 1026–1035. IEEE (2017)
Nguyen, T.P., Choi, S., Park, S.-J., Park, S.H., Yoon, J.: Inspecting method for defective casting products with convolutional neural network (CNN). Int. J. Precision Eng. Manuf.-Green Technol. 8(2), 583–594 (2021)
Kuo, J.-K., Wu, J.-J., Huang, P.-H., Cheng, C.-Y.: Inspection of sandblasting defect in investment castings by deep convolutional neural network. Int. J. Adv. Manuf. Technol. 120(3–4), 2457–2468 (2022)
Jiang, X., Wang, X., Chen, D.: Research on defect detection of castings based on deep residual network. In: 2018 11th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics (CISP-BMEI), pp. 1–6. IEEE (2018)
Wang, Y., Hu, C., Chen, K., Yin, Z.: Self-attention guided model for defect detection of aluminium alloy casting on X-ray image. Comput. Electr. Eng. 88, 106821 (2020)
Ren, S., He, K., Girshick, R., Sun, J.: Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2017)
Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: unified, real-time object detection. In: 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, pp. 779–788. IEEE (2016)
Jiao, L., et al.: A survey of deep learning-based object detection. IEEE Access 7, 128837–128868 (2019)
Ferguson, M., Ak, R., Lee, Y.-T.T., Law, K.H.: Automatic localization of casting defects with convolutional neural networks. In: 2017 IEEE International Conference on Big Data (Big Data), Boston, MA, pp. 1726–1735. IEEE (2017)
Du, W., Shen, H., Fu, J., Zhang, G., He, Q.: Approaches for improvement of the X-ray image defect detection of automobile casting aluminum parts based on deep learning. NDT E Int. 107, 102144 (2019)
Lin, T.-Y., Dollar, P., Girshick, R., He, K., Hariharan, B., Belongie, S.: Feature pyramid networks for object detection. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, pp. 936–944. IEEE (2017)
He, K., Gkioxari, G., Dollár, P., Girshick, R.: Mask R-CNN. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 2961–2969 (2017)
Xing, J., Jia, M.: A convolutional neural network-based method for workpiece surface defect detection. Measurement 176, 109185 (2021)
Du, W., Shen, H., Fu, J., Zhang, G., Shi, X., He, Q.: Automated detection of defects with low semantic information in X-ray images based on deep learning. J. Intell. Manuf. 32(1), 141–156 (2021)
Alzubaidi, L., et al.: A survey on deep learning tools dealing with data scarcity: definitions, challenges, solutions, tips, and applications. J. Big Data 10(1), 46 (2023)
Gong, Y., Luo, J., Shao, H., Li, Z.: A transfer learning object detection model for defects detection in X-ray images of spacecraft composite structures. Compos. Struct. 284, 115136 (2022)
Jaiswal, A., Babu, A.R., Zadeh, M.Z., Banerjee, D., Makedon, F.: A survey on contrastive self-supervised learning. Technologies 9(1), 2 (2020)
Chen, T., Kornblith, S., Norouzi, M., Hinton, G.: A Simple Framework for Contrastive Learning of Visual Representations (2020)
Khosla, P., et al.: Supervised contrastive learning. In: Advances in Neural Information Processing Systems, vol. 33, pp. 18661–18673 (2020)
Sun, B., Li, B., Cai, S., Yuan, Y., Zhang, C.: FSCE: few-shot object detection via contrastive proposal encoding. In: 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Nashville, TN, USA, pp. 7348–7358. IEEE (2021)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, pp. 770–778. IEEE (2016)
Acknowledgment
Eneko intxausti, Carlos Cernuda and Ekhi Zugasti are part of the Intelligent Systems for Industrial Systems research group of Mondragon Unibertsitatea (IT1676-22), supported by the Department of Education, Universities and Research of the Basque Country. They are also supported by the DREMIND project of the Basque Government under Grant KK-2022/00049 from the ELKARTEK program.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 Springer Nature Switzerland AG
About this paper
Cite this paper
Intxausti, E., Zugasti, E., Cernuda, C., Leibar, A.M., Elizondo, E. (2024). Towards Robust Defect Detection in Casting Using Contrastive Learning. In: Vasconcelos, V., Domingues, I., Paredes, S. (eds) Progress in Pattern Recognition, Image Analysis, Computer Vision, and Applications. CIARP 2023. Lecture Notes in Computer Science, vol 14469. Springer, Cham. https://doi.org/10.1007/978-3-031-49018-7_43
Download citation
DOI: https://doi.org/10.1007/978-3-031-49018-7_43
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-49017-0
Online ISBN: 978-3-031-49018-7
eBook Packages: Computer ScienceComputer Science (R0)