Abstract
Tool breakage in manufacturing procedures can damage machined surfaces and machine tools. It is crucial to detect tool breakage in time and promptly respond to it. Due to the safety restrictions imposed in production, failure samples are significantly scarcer than normal samples, and this disequilibrium results in difficulty of failure detection. Therefore, we propose a new imbalanced data learning method for tool breakage detection. The key strategy is to balance the data distribution by producing valuable artificial samples for the minority class using an adversarial generative oversampling model based on a generative adversarial network (GAN). Unlike previous studies using GAN, we use the discriminator to screen samples generated by the generator and achieve effective oversampling. Multiple classifiers are adopted as the decision-making models to perform tool breakage detection. The proposed method is applied to a set of imbalanced experimental tool breakage data collected in a workshop. Compared with the best results of other oversampling solutions, the proposed method improves the breakage detection rate from 93.6% to 100%, which shows its practicability and validity. Additionally, evaluations are performed based on 12 imbalanced benchmark datasets. The results further substantiate the superiority of the proposed method to existing sampling methods.
Similar content being viewed by others
References
Abadi, M., Barham, P., Chen, J., Chen, Z., Davis, A., Dean, J., et al. (2016). Tensorflow: A system for large-scale machine learning. In 12th {USENIX} symposium on operating systems design and implementation ({OSDI} 16) (pp. 265–283)
Bassiuny, A. M., & Li, X. (2007). Flute breakage detection during end milling using Hilbert-Huang transform and smoothed nonlinear energy operator. International Journal of Machine Tools and Manufacture, 47(6), 1011–1020. https://doi.org/10.1016/j.ijmachtools.2006.06.016
Benkedjouh, T., Medjaher, K., Zerhouni, N., & Rechak, S. (2015). Health assessment and life prediction of cutting tools based on support vector regression. Journal of Intelligent Manufacturing, 26(2), 213–223. https://doi.org/10.1007/s10845-013-0774-6
Breiman, L. (2001). Random forests. Machine Learning, 45(1), 5–32. https://doi.org/10.1023/A:1010933404324
Bustillo, A., & Rodríguez, J. J. (2013). Online breakage detection of multitooth tools using classifier ensembles for imbalanced data. International Journal of Systems Science, 45(12), 2590–2602. https://doi.org/10.1080/00207721.2013.775378
Cao, H., Chen, X., Zi, Y., Ding, F., Chen, H., Tan, J., et al. (2008). End milling tool breakage detection using lifting scheme and Mahalanobis distance. International Journal of Machine Tools and Manufacture, 48(2), 141–151. https://doi.org/10.1016/j.ijmachtools.2007.09.001
Chang, C.-C., & Lin, C.-J. (2011). LIBSVM: A library for support vector machines. ACM Transactions on Intelligent Systems and Technology, 2(3), 1–27. https://doi.org/10.1145/1961189.1961199
Chawla, N. V., Bowyer, K. W., Hall, L. O., & Kegelmeyer, W. P. (2002). SMOTE: Synthetic minority over-sampling technique. Journal of Artificial Intelligence Research, 16, 321–357. https://doi.org/10.1613/jair.953
Chen, J. C., & Chen, W.-L. (1999). A tool breakage detection system using an accelerometer sensor. Journal of Intelligent Manufacturing, 10(2), 187–197. https://doi.org/10.1023/A:1008980821787
Cho, S., Asfour, S., Onar, A., & Kaundinya, N. (2005). Tool breakage detection using support vector machine learning in a milling process. International Journal of Machine Tools and Manufacture, 45(3), 241–249. https://doi.org/10.1016/j.ijmachtools.2004.08.016
Dietterich, T. G. (2000). Ensemble methods in machine learning. In International workshop on multiple classifier systems, Berlin, Heidelberg (pp. 1–15). Springer. https://doi.org/10.1007/3-540-45014-9_1.
Dong, Y., & Wang, X. (2011). A new over-sampling approach: Random-SMOTE for learning from imbalanced data sets. In 5th international conference on knowledge science, engineering and management, KSEM 2011, December 12, 2011 - December 14, 2011, Irvine, CA, United states, 2011 (Vol. 7091 LNAI, pp. 343–352). Lecture Notes in Computer Science): Springer Verlag. https://doi.org/10.1007/978-3-642-25975-3_30.
Douzas, G., & Bacao, F. (2018). Effective data generation for imbalanced learning using conditional generative adversarial networks. Expert Systems with Applications, 91, 464–471. https://doi.org/10.1016/j.eswa.2017.09.030
Dua, D., & Graff, C. (2019). UCI machine learning repository. http://archive.ics.uci.edu/ml.
Elsken, T., Metzen, J. H., & Hutter, F. (2018). Neural architecture search: A survey. arXiv preprint arXiv:1808.05377
Freyer, B. H., Heyns, P. S., & Theron, N. J. (2014). Comparing orthogonal force and unidirectional strain component processing for tool condition monitoring. Journal of Intelligent Manufacturing, 25(3), 473–487. https://doi.org/10.1007/s10845-012-0698-6
Goodfellow, I. J., Pouget-Abadie, J., Mirza, M., Xu, B., Warde-Farley, D., Ozair, S., et al. (2014). Generative adversarial nets. In Advances in neural information processing systems, 2014 (January ed., Vol. 3, pp. 2672–2680). https://doi.org/10.5555/2969033.2969125.
Grzymala-Busse, J. W., Stefanowski, J., & Wilk, S. (2005). A comparison of two approaches to data mining from imbalanced data. Journal of Intelligent Manufacturing, 16(6), 565–573. https://doi.org/10.1007/s10845-005-4362-2
Haixiang, G., Yijing, L., Shang, J., Mingyun, G., Yuanyue, H., & Bing, G. (2017). Learning from class-imbalanced data: Review of methods and applications. Expert Systems with Applications, 73, 220–239. https://doi.org/10.1016/j.eswa.2016.12.035
Han, H., Wang, W.-Y., & Mao, B.-H. (2005). Borderline-SMOTE: A new over-sampling method in imbalanced data sets learning. In International Conference on Intelligent Computing, ICIC 2005, August 23, 2005–August 26, 2005, Hefei, China, 2005 (PART I ed., Vol. 3644, pp. 878–887). Lecture Notes in Computer Science). Springer. https://doi.org/10.1007/11538059_91.
He, H., Bai, Y., Garcia, E. A., & Li, S. (2008). ADASYN: Adaptive synthetic sampling approach for imbalanced learning. In 2008 International joint conference on neural networks, IJCNN 2008, June 1, 2008–June 8, 2008, Hong Kong, China, 2008 (pp. 1322–1328, Proceedings of the International Joint Conference on Neural Networks): Institute of Electrical and Electronics Engineers Inc. https://doi.org/10.1109/IJCNN.2008.4633969.
Huang, P. B., Ma, C.-C., & Kuo, C.-H. (2015). A PNN self-learning tool breakage detection system in end milling operations. Applied Soft Computing, 37, 114–124. https://doi.org/10.1016/j.asoc.2015.08.019
Kingma, D. P., & Ba, J. (2014). Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980.
Kotsiantis, S. B. (2013). Decision trees: A recent overview. Artificial Intelligence Review, 39(4), 261–283. https://doi.org/10.1007/s10462-011-9272-4
Lee, J., & Park, K. (2019). GAN-based imbalanced data intrusion detection system. Personal and Ubiquitous Computing,. https://doi.org/10.1007/s00779-019-01332-y
Li, H., Wang, Y., Zhao, P., Zhang, X., & Zhou, P. (2015). Cutting tool operational reliability prediction based on acoustic emission and logistic regression model. Journal of Intelligent Manufacturing, 26(5), 923–931. https://doi.org/10.1007/s10845-014-0941-4
Li, X. (2001). Detection of tool flute breakage in end milling using feed-motor current signatures. IEEE/ASME Transactions on Mechatronics, 6(4), 491–498. https://doi.org/10.1109/3516.974863
Li, X., Ouyang, G., & Liang, Z. (2008). Complexity measure of motor current signals for tool flute breakage detection in end milling. International Journal of Machine Tools and Manufacture, 48(3–4), 371–379. https://doi.org/10.1016/j.ijmachtools.2007.09.008
Luo, J., Huang, J., & Li, H. (2021). A case study of conditional deep convolutional generative adversarial networks in machine fault diagnosis. Journal of Intelligent Manufacturing, 32(2), 407–425. https://doi.org/10.1007/s10845-020-01579-w
Mountrakis, G., Im, J., & Ogole, C. (2011). Support vector machines in remote sensing: A review. ISPRS Journal of Photogrammetry and Remote Sensing, 66(3), 247–259. https://doi.org/10.1117/12.903949
Pedregosa, F., Varoquaux, G., Gramfort, A., Michel, V., Thirion, B., Grisel, O., et al. (2011). Scikit-learn: Machine learning in Python. Journal of Machine Learning Research, 12, 2825–2830.
Ritou, M., Garnier, S., Furet, B., & Hascoet, J. Y. (2014). Angular approach combined to mechanical model for tool breakage detection by eddy current sensors. Mechanical Systems and Signal Processing, 44(1–2), 211–220. https://doi.org/10.1016/j.ymssp.2013.02.004
Salzberg, S. L. (1994). C45: Programs for Machine Learning. In J. Ross Quinlan (Ed.), Machine Learning. Burligton: Morgan Kaufmann Publisher, Inc.
Sun, S., Hu, X., & Zhang, W. (2020). Detection of tool breakage during milling process through acoustic emission. International Journal of Advanced Manufacturing Technology. https://doi.org/10.1007/s00170-020-05751-7
Wang, Y.-R., Sun, G.-D., & Jin, Q. (2020). Imbalanced sample fault diagnosis of rotating machinery using conditional variational auto-encoder generative adversarial network. Applied Soft Computing, 92, 106333. https://doi.org/10.1016/j.asoc.2020.106333
Wong, S. Y., Chuah, J. H., & Yap, H. J. (2020). Technical data-driven tool condition monitoring challenges for CNC milling: A review. International Journal of Advanced Manufacturing Technology. https://doi.org/10.1007/s00170-020-05303-z
Xu, G. D., Zhou, H. C., & Chen, J. H. (2018). CNC internal data based incremental cost-sensitive support vector machine method for tool breakage monitoring in end milling. Engineering Applications of Artificial Intelligence, 74, 90–103. https://doi.org/10.1016/j.engappai.2018.05.007
Zhang, K.-F., Yuan, H.-Q., & Nie, P. (2015). A method for tool condition monitoring based on sensor fusion. Journal of Intelligent Manufacturing, 26(5), 1011–1026. https://doi.org/10.1007/s10845-015-1112-y
Zhang, Y. Y., Li, X. Y., Gao, L., Wang, L. H., & Wen, L. (2018). Imbalanced data fault diagnosis of rotating machinery using synthetic oversampling and feature learning. Journal of Manufacturing Systems, 48, 34–50. https://doi.org/10.1016/j.jmsy.2018.04.005
Zhu, T. F., Lin, Y. P., & Liu, Y. H. (2017). Synthetic minority oversampling technique for multiclass imbalance problems. Pattern Recognition, 72, 327–340. https://doi.org/10.1016/j.patcog.2017.07.024
Acknowledgements
This work was supported by the National Key R&D Program of China [grant number 2018YFB1700500] and Science and Technology Commission of Shanghai Municipality [grant number 19511105302].
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Sun, S., Hu, X. & Liu, Y. An imbalanced data learning method for tool breakage detection based on generative adversarial networks. J Intell Manuf 33, 2441–2455 (2022). https://doi.org/10.1007/s10845-021-01806-y
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10845-021-01806-y