Skip to main content
Log in

Multi-residual unit fusion and Wasserstein distance-based deep transfer learning for mill load recognition

  • Original Paper
  • Published:
Signal, Image and Video Processing Aims and scope Submit manuscript

Abstract

This paper proposes the ball mill load recognition algorithm (MRUF-WD) based on multi-residual unit fusion (MRUF) and Wasserstein distance transfer learning to address the problem of accurately judging the working state parameters during ball mill grinding operations. The proposed algorithm can significantly enhance the processing capability of under-labeled or unlabeled data, while ensuring excellent transfer capability even when there is a large data gap between the target and source domains. The MRUF-WD transfer algorithm first inputs collected ball mill vibration data under different working conditions into the MRUF feature extraction network based on ResNet-18, where MRUF can extract feature information from the upper, middle, and lower layers of the network for learning. This approach equips the algorithm with enhanced feature extraction capability, leading to a significant improvement in the processing capability for under-labeled or unlabeled data. This algorithm has a better ability to deal with under-labeled or unlabeled data through its improved feature extraction capability. It then inputs a classifier that uses Wasserstein distance as a loss function to measure distance between source and target domain data, as well as a domain alignment network. The Wasserstein distance loss function, also known as EDM (Earth Mover’s Distance), measures the distance between two probability distributions and exhibits broader applicability, better robustness, and improved continuity when the probability distribution of the data changes, compared to other common distance metrics. After the parameters are updated by back propagation, the algorithm uses the invariant features learned in the source domain to identify the load in the target domain.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

Data availability

The experimental data in Sect. 4.1 were obtained in the laboratory of Jiangxi University of Science and Technology.

References

  1. Luo, X., Huang, Y., Li, B., Liu, J.: Development of PSO-LSSVM-based ball mill load parameter prediction and monitoring system. Noise Vib. Control 42(4), 144–151 (2022)

    Google Scholar 

  2. Liu, Z., Tang, J., Chai, T., Yu, W.: A mill load parameter prediction method based on selective integrated modeling of multimodal feature subsets. J. Autom. 47(8), 1921–1931 (2021)

    Google Scholar 

  3. Tang, J., Qiao, J., Liu, Z., Zhou, X., Yu, G., Zhao, J.: Multi-channel mechanical signal analysis evaluation and optimal combination for mill load parameter prediction. J. Beijing Inst. Technol. 46(9), 997–1007 (2020)

    Google Scholar 

  4. Sui, L.: Research on soft measurement modeling of key parameters of wet ball mill based on multi-task and transfer learning. Taiyuan University of Technology (2018)

  5. Tang, J., Qiao, J.F., Liu, Z., Zhou, X.J., Yu, G., Zhao, J.J.: Mechanism characteristic analysis and soft measuring method review for ball mill load based on mechanical vibration and acoustic signals in the grinding process (Review). Miner. Eng. 28(1), 294–311 (2018)

    Article  Google Scholar 

  6. Wang, X., Sun, K., Zhang, H., Xiong, W., Yang, C.: Mill load identification method for ball milling process based on grinding signal. IFAC-PapersOnLine 54(21), 7–12 (2021)

    Article  Google Scholar 

  7. Yang, D., Karimi, H.R., Pawelczyk, M.: A new intelligent fault diagnosis framework for rotating machinery based on deep transfer reinforcement learning. Control. Eng. Pract. 134, 105475 (2023)

    Article  Google Scholar 

  8. Huang, M., Yin, J., Yan, S., et al.: A fault diagnosis method of bearings based on deep transfer learning. Simul. Model. Pract. Theory 122, 102659 (2023)

    Article  Google Scholar 

  9. Liu, G., Shen, W., Gao, L., et al.: Automated broad transfer learning for cross-domain fault diagnosis. J. Manuf. Syst. 66, 27–41 (2023)

    Article  Google Scholar 

  10. He, K.M., Zhang, X.Y., Ren, S.Q., Sun, J.: Deep residual learning for image recognition. In: 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 770–778 (2016)

  11. Wang, R., An, S., Liu, W., Li, L.: Invertible residual blocks in deep learning networks. IEEE Trans. Neural Netw. Learn. Syst. 1–7 (2023)

  12. Arjovsky, M., Chintala, S., Bottou, L.: Wasserstein gan. arXiv:1701.07875 (2017)

  13. Long, M., Cao, Y., Wang, J., Jordan, M.I.: Learning transferable features with deep adaptation networks. In: Computer Science (2015)

  14. Gretton, A., Borgwardt, K.M., Rasch, M.J., Scholkopf, B., Smola, A.: A kernel two-sample test. J. Mach. Learn. Res. 13, 723–773 (2012)

    MathSciNet  Google Scholar 

  15. Xu, K., Li, S., Jiang, X., An, Z., Wang, J., Yu, T.: A renewable fusion fault diagnosis network for the variable speed conditions under unbalanced samples. Neurocomputing 379, 12–29 (2020)

    Article  Google Scholar 

  16. Goodfellow, I., Pouget-Abadie, J., Mirza, M., Bing, X., Warde-Farley, D., Ozair, S., Courville, A., Bengio, Y.: Generative adversarial networks. Commun. ACM 63(11), 139–144 (2020)

    Article  MathSciNet  Google Scholar 

  17. Ganin, Y., Ustinova, E., Ajakan, H.: Domain-adversarial training of neural networks. J. Mach. Learn. Res. 17(1), 2096 (2016)

    MathSciNet  Google Scholar 

Download references

Funding

The National Natural Science Foundation of China (NSFC) project “Research on energy consumption of vibratory crushing based on multi-scale cohesive particle model” [Grant No. 51464017] and the Science and Technology Project of Jiangxi Provincial Department of Education, “State monitoring and load intelligent identification of wolframite milling process” [Grant No. 200827] provided partial support for this study.

Author information

Authors and Affiliations

Authors

Contributions

Xu wrote the main manuscript text and Xiao prepared figures 2, 3, 6. Luo provided deep learning model guidance and experimental equipment. All authors reviewed the manuscript.

Corresponding author

Correspondence to Xiaoyan Luo.

Ethics declarations

Competing interests

The authors declare no competing interests.

Ethical approval

This study did not involve human or animal subjects, and thus, no ethical approval was required. The study protocol adhered to the guidelines established by the journal.

Additional information

Publisher’s Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Xu, H., Luo, X. & Xiao, W. Multi-residual unit fusion and Wasserstein distance-based deep transfer learning for mill load recognition. SIViP 18, 3187–3196 (2024). https://doi.org/10.1007/s11760-023-02981-6

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11760-023-02981-6

Keywords

Navigation