Abstract
A multidomain variance-learnable prototypical network (MVPN) is proposed to learn transferable knowledge from a large-scale dataset containing sufficient samples of multiple faults for few-shot diagnosis of novel faults (i.e., disjoint with fault types in the large-scale dataset). Signal characterizations in time, frequency, and time–frequency domains are first constructed to make full use of information contained in severely limited labeled data. Mahalanobis distance is proposed as a criterion for improving classification performance by considering different spreads between classes in the embedding space. The spread variance of each class is learned by constructing an additional deep learning network in the original prototypical network. Multidomain signals are used to learn the prototype representations and spread variances separately, and are finally fused for classification. With the proposed MVPN, deeper variance-learnable embedding learning from wider domain characterizations improves the ability of few-shot fault diagnosis. Experiments are conducted to evaluate the performance of MVPN using datasets collected from a benchmark bearing and a Delta 3-D printer. Results indicate that the proposed MVPN performs competitively compared to state-of-the-art few-shot learning algorithms.
Similar content being viewed by others
References
Case Western Reserve University Bearing Data Center. https://engineering.case.edu/bearingdatacenter
Chen, X. H., Zhang, B. K., & Gao, D. (2021). Bearing fault diagnosis base on multi-scale CNN and LSTM model. Journal of Intelligent Manufacturing, 32, 971–987.
Das, D., & Lee, C. S. G. (2020). A two-stage approach to few-shot learning for image recognition. IEEE Transactions on Image Processing, 29, 3336–3350.
Das, D., Moon, J. H., & George Lee, C. S. (2020). Few-shot image recognition with manifolds. In International symposium on visual computing (ISVC) (pp. 3–14).
Garcia, V., & Bruna, J. (2018). Few-shot learning with graph neural networks. https://doi.org/10.48550/arXiv.1711.04043
Huang, S., Zeng, X., Wu, S., Yu, Z., Azzam, M., & Wong, H.-S. (2021). Behavior regularized prototypical networks for semi-supervised few-shot image classification. Pattern Recognition, 112, 107765.
Hui, B., Liu, L., Chen, J., Zhou, X., & Nian, Y. (2020). Few-shot relation classification by context attention-based prototypical networks with BERT. EURASIP Journal on Wireless Communications and Networking, 2020(1), 118.
Jalayer, M., Orsenigo, C., & Vercellis, C. (2021). Fault detection and diagnosis for rotating machinery: A model based on convolutional LSTM, fast Fourier and continuous wavelet transforms. Computers in Industry, 125, 103378.
Ji, Z., Chai, X., Yu, Y., Pang, Y., & Zhang, Z. (2020). Improved prototypical networks for few-shot learning. Pattern Recognition Letters, 140, 81–87.
Ji, Z., Chai, X., Yu, Y., & Zhang, Z. (2021). Reweighting and information-guidance networks for few-shot learning. Neurocomputing, 423, 13–23.
Lee, H., Jeong, H., Koo, G., Ban, J., & Kim, S. W. (2021). Attention recurrent neural network-based severity estimation method for interturn short-circuit fault in permanent magnet synchronous machines. IEEE Transactions on Industrial Electronics, 68(4), 3445–3453.
Li, C., Cabrera, D., Sancho, F., Sanchez, R. V., Cerrada, M., & de Oliveira, J. V. (2021). One-shot fault diagnosis of three-dimensional printers through improved feature space learning. IEEE Transactions on Industrial Electronics, 68(9), 8768–8776.
Li, C., Sanchez, R.-V., Zurita, G., Cerrada, M., Cabrera, D., & Vásquez, R. E. (2015). Multimodal deep support vector classification with homologous features and its application to gearbox fault diagnosis. Neurocomputing, 168, 119–127.
Li, C., Tao, Y., Ao, W., Yang, S., & Bai, Y. (2018). Improving forecasting accuracy of daily enterprise electricity consumption using a random forest based on ensemble empirical mode decomposition. Energy, 165, 1220–1227.
Liu, C., Fu, Y., Xu, C., Yang, S., Li, J., Wang, C., & Zhang, L. (2021). Learning a few-shot embedding model with contrastive learning. In Proceedings of the AAAI conference on artificial intelligence (Vol. 35, No. 10).
Long, J., Chen, Y. B., Yang, Z., Huang, Y. W., & Li, C. (2022). A novel self-training semi-supervised deep learning approach for machinery fault diagnosis. International Journal of Production Research. https://doi.org/10.1080/00207543.2022.2032860
Long, J., Mou, J. D., Zhang, L. W., Zhang, S. H., & Li, C. (2021). Attitude data-based deep hybrid learning architecture for intelligent fault diagnosis of multi-joint industrial robots. Journal of Manufacturing Systems, 61, 736–745.
Medina, R., Macancela, J. C., Lucero, P., Cabrera, D., Sanchez, R. V., & Cerrada, M. (2022). Gear and bearing fault classification under different load and speed by using Poincare plot features and SVM. Journal of Intelligent Manufacturing, 33, 1031–1055.
Peng, J. Y., Kimmig, A., Wang, D. K., Niu, Z. B., Zhi, F., Wang, J. H., Liu, X. F., & Ovtcharova, J. (2022). A systematic review of data-driven approaches to fault diagnosis and early warning. Journal of Intelligent Manufacturing. https://doi.org/10.1007/s10845-022-02020-0
Rai, A., & Upadhyay, S. H. (2016). A review on signal processing techniques utilized in the fault diagnosis of rolling element bearings. Tribology International, 96, 289–306.
Saravanan, N., & Ramachandran, K. I. (2010). Incipient gear box fault diagnosis using discrete wavelet transform (DWT) for feature extraction and classification using artificial neural network (ANN). Expert Systems with Applications, 37(6), 4168–4181.
Snell, J., Swersky, K., & Zemel, R. S. (2017). Prototypical networks for few-shot learning. ArXiv170305175 Cs Stat. Retrieved December 15, 2020, from http://arxiv.org/abs/1703.05175
Sun, W. X., Chen, J., & Li, J. Q. (2007). Decision tree and PCA-based fault diagnosis of rotating machinery. Mechanical Systems and Signal Processing, 21, 1300–1317.
Sun, X., Xv, H., Dong, J., Zhou, H., Chen, C., & Li, Q. (2021). Few-shot learning for domain-specific fine-grained image classification. IEEE Transactions on Industrial Electronics, 68(4), 3588–3598.
Sung, F., Yang, Y., Zhang, L., Xiang, T., Torr, P. H. S., & Hospedales, T. M. (2018). Learning to compare: Relation network for few-shot learning. ArXiv171106025. Retrieved March 7, 2021, from http://arxiv.org/abs/1711.06025
Vinyals, O., Blundell, C., Lillicrap, T., Kavukcuoglu, K., & Wierstra, D. (2017). Matching networks for one shot learning. ArXiv160604080 Cs Stat. Retrieved April 6, 2021, from http://arxiv.org/abs/1606.04080
Wang, Y., Yao, Q., Kwok, J. T., & Ni, L. M. (2020). Generalizing from a few examples: A survey on few-shot learning. ACM Computing Surveys, 53(3), 1–34.
Zhang, P., Bai, Y., Wang, D., Bai, B., & Li, Y. (2020). Few-shot classification of aerial scene images via meta-learning. Remote Sensing, 13(1), 108.
Zhou, X. K., Liang, W., Shimizu, S. H., Ma, J. H., & Jin, Q. (2021). Siamese neural network based few-shot learning for anomaly detection in industrial cyber-physical systems. IEEE Transactions on Industrial Informatics, 17, 5790–5798.
Acknowledgements
This research is partially supported by the National Natural Science Foundation of China (72171049, 52175080, 52005103 and 52205556), Young Elite Scientists Sponsorship Program by CAST (2022QNRC001), the Guangdong Basic and Applied Basic Research Foundation (2022A1515012004, 2021A1515110521 and 2020A1515110139), Department of Education of Guangdong in China (2021ZDJS083). The valuable comments and suggestions from the editors and the three anonymous reviewers are very much appreciated.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Competing interests
The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Appendix
Appendix
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Long, J., Chen, Y., Huang, H. et al. Multidomain variance-learnable prototypical network for few-shot diagnosis of novel faults. J Intell Manuf 35, 1455–1467 (2024). https://doi.org/10.1007/s10845-023-02123-2
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10845-023-02123-2