Classifiers ensemble of transfer learning for improved drill wear classification using convolutional neural network

Main Article Content

Jarosław Kurek
Joanna Aleksiejuk-Gawron
Izabella Antoniuk
Jarosław Górski
Albina Jegorowa
Michał Kruk
Arkadiusz Orłowski
Jakub Pach
Bartosz Świderski
Grzegorz Wieczorek


Keywords : classifiers ensemble, convolutional neural networks, data augmentation, deep learning, tool condition monitoring
Abstract

In this paper we introduce the enhanced drill wear recognition method, based on classifiers ensemble, obtained using transfer learning and data augmentation methods. Red, green and yellow classes are used to describe the current drill state. The first one corresponds to the case when drill should be immediately replaced. The second one denotes a tool that is still in a good condition. The final class refers to the case when a drill is suspected of being worn out, and a human expert evaluation would be required. The proposed algorithm uses three different, pretrained network models and adjusts them to the drill wear classification problem. To ensure satisfactory results, each of the methods used was required to achieve accuracy above 90% for the given classification task. Final evaluation is achieved by voting of all three classifiers. Since the initial data set was small (242 instances), the data augmentation method was used to artificially increase the total number of drill hole images. The experiments performed confirmed that the presented approach can achieve high accuracy, even with such a limited set of training data.

Article Details

How to Cite
Kurek, J., Aleksiejuk-Gawron, J., Antoniuk, I., Górski, J., Jegorowa, A., Kruk, M., Orłowski, A., Pach, J., Świderski, B., & Wieczorek, G. (2019). Classifiers ensemble of transfer learning for improved drill wear classification using convolutional neural network. Machine Graphics and Vision, 28(1/4), 13–23. https://doi.org/10.22630/MGV.2019.28.1.2
References

K. Jemielniak, T. Urbański, J. Kossakowska, S. Bombiński. Tool condition monitoring based on numerous signal features. Int. J. Adv. Manuf. Technol., vol. 59, pp. 73-81, 2012. (Crossref)

S. S. Panda, A. K. Singh, D. Chakraborty, S. K. Pal. Drill wear monitoring using back propagation neural network. Journal of Materials Processing Technology, vol. 172, pp. 283-290, 2006. (Crossref)

R. J. Kuo, Multi-sensor integration for on-line tool wear estimation through artificial neural networks and fuzzy neural network. Engineering Applications of Artificial Intelligence, vol. 13, pp. 249-261, 2000. (Crossref)

J. Kurek, M. Kruk, S. Osowski, P. Hoser, G. Wieczorek, A. Jegorowa, J. Górski, J. Wilkowski, K. Śmietańska, J. Kossakowska. Developing automatic recognition system of drill wear in standard laminated chipboard drilling process. Bulletin of the Polish Academy of Sciences. Technical Sciences, vol. 64, pp. 633-640, 2016. (Crossref)

J. Kurek, G. Wieczorek, M. Kruk, A. Jegorowa, S. Osowski. Transfer learning in recognition of drill wear using convolutional neural network. 18th International Conference on Computational Problems of Electrical Engineering (CPEE), pp. 1-4. IEEE. September 2017. (Crossref)

J. Kurek, I. Antoniuk, J. Górski, A. Jegorowa, B. Świderski, M. Kruk, G. Wieczorek, J. Pach, A. Orłowski, and J. Aleksiejuk-Gawron. Data augmentation techniques for transfer learning improvement in drill wear classification using convolutional neural network. Machine Graphics & Vision, 28(1/4):3–12, 2019. doi:10.22630/MGV.2019.28.1.1. (Crossref)

J. Kurek, B. Świderski, A. Jegorowa, M. Kruk, S. Osowski. Deep learning in assessment of drill condition on the basis of images of drilled holes. Proc. SPIE 10225 Eighth International Conference on Graphic and Image Processing (ICGIP 2016), pp. 102251V, February 8, 2017. (Crossref)

L. Deng, D. Yu. Deep Learning: Methods and Applications. Foundations and Trends in Signal Processing, vol. 7, pp. 3-4, 2014. (Crossref)

Y. Bengio. Learning Deep Architectures for AI. Foundations and Trends in Machine Learning, vol. 2, no. 1, pp. 1-127, 2009. (Crossref)

I. Goodfellow, Y. Bengio, A. Courville. Deep learning. MIT Press, 2016.

J. Schmidhuber. Deep Learning in Neural Networks: An Overview. Neural Networks, vol. 61, pp. 85-117, 2015. (Crossref)

A. Krizhevsky, I. Sutskever, G. Hinton. Image net classification with deep convolutional neural networks. Advances in Neural Information Processing Systems, vol. 25, pp. 1-9, 2012.

O. Russakovsky, J. Deng, H. Su et al. ImageNet Large Scale Visual Recognition Challenge. International Journal of Computer Vision (IJCV), vol. 115, no. 3, pp. 211-252, 2015. (Crossref)

BVLC AlexNet Model. Online: https://github.com/BVLC/caffe/tree/master/models/bvlc_alexnet.

Matlab 2017a – User manual, Natick, MA, USA: The MathWorks, Inc, 2017.

ImageNet. Online: http://www.image-net.org.

B. Scholkopf, A. Smola. Learning with Kernels, Cambridge: MIT Press, 2002.

M. Kruk, B. Świderski, S. Osowski, J. Kurek, M. Słowińska, I. Walecka. Melanoma recognition using extended set of descriptors and classifiers. Eurasip Journal on Image and Video Processing, vol. 43, pp. 1-10, 2015. (Crossref)

V. N. Vapnik. Statistical Learning Theory, New York: Wiley, 1998.

Description of Matlab image transformations. Online: https://www.mathworks.com/help/deeplearning/examples/image-augmentation-using-image-processing-toolbox.html.

Statistics
Recommend Articles
Similar Articles

You may also start an advanced similarity search for this article.

Most read articles by the same author(s)