A Multi-Task Multimodal Attention Graph Convolutional Network for Acoustic–Vibration Fusion-Based Rolling Bearing Fault Diagnosis
Source abstract
Abstract
Single-sensor-based fault diagnosis of rolling bearings often suffers from noise sensitivity, installation-dependent performance, and incomplete fault characterization. To address these limitations, this paper proposes a multi-task multimodal attention graph convolutional network (MTMAGNet) that integrates acoustic and vibration signals for bearing fault diagnosis. First, one-dimensional convolutional neural networks are used to extract modality-specific features. These features are then fused through a multi-modal attention mechanism to exploit the complementary information contained in the two signal sources. Based on the fused representations, a dynamic k-nearest neighbor graph is constructed to model relationships among samples, and a graph convolutional network is employed to learn discriminative structural features. Moreover, a multi-task learning scheme is introduced, in which fault classification serves as the primary task and modal classification is used as an auxiliary task to enhance feature learning and improve model generalization. Experimental results on a self-built acoustic–vibration test bench collected under three rotational speeds (1800 rpm, 2400 rpm, and 3000 rpm) demonstrate that the proposed method achieves high diagnostic accuracy and strong generalization performance under different fault conditions.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Machine Fault Diagnosis Techniques · Gear and Bearing Dynamics Analysis · Machine Learning and ELM
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Engineering Training Center, Shenyang University of Technology, Shenyang 110870, China
School of Mechanical Engineering, Shenyang University of Technology, Shenyang 110870, China
School of Information Science and Engineering, Shenyang University of Technology, Shenyang 110870, China
Funding
Liaoning Province Doctoral Research Startup Fund Project
References (41)
Show source-provided references
1. Yang, Y., Han, C., Ran, G., Ma, T., and Pan, J. (2025). Fault Diagnosis of Rolling Element Bearing Based on BiTCN-Attention and OCSSA Mechanism. Actuators, 14.10.3390/act14050218 ↗
2. An enhanced selective ensemble deep learning method for rolling bearing fault diagnosis with beetle antennae search algorithm10.1016/j.ymssp.2020.106752 ↗
3. Early fault diagnosis of bearings using an improved spectral kurtosis by maximum correlated kurtosis deconvolution10.3390/s151129363 ↗
4. Fault diagnosis of rolling bearing based on wavelet package transform and ensemble empirical mode decomposition10.1155/2013/792584 ↗
5. An intelligent fault diagnosis method of rolling bearings based on short-time Fourier transform and convolutional neural network10.1007/s11668-023-01616-9 ↗
6. Guo, Y., Yang, Y., Jiang, S., Jin, X., and Wei, Y. (2022). Rolling bearing fault diagnosis based on successive variational mode decomposition and the EP index. Sensors, 22.10.3390/s22103889 ↗
7. Modified multiscale weighted permutation entropy and optimized support vector machine method for rolling bearing fault diagnosis with complex signals10.1016/j.isatra.2020.12.054 ↗
8. A comprehensive evaluation of the effect of defect size in rolling element bearings on the statistical features of the vibration signal10.1016/j.ymssp.2020.107334 ↗
9. A fault information-guided variational mode decomposition (FIVMD) method for rolling element bearings diagnosis10.1016/j.ymssp.2021.108216 ↗
10. Selection of efficient degradation features for rolling element bearing prognosis using Gaussian process regression method10.1016/j.isatra.2020.12.020 ↗
11. Deep branch attention network and extreme multi-scale entropy based single vibration signal-driven variable speed fault diagnosis scheme for rolling bearing10.1016/j.aei.2022.101844 ↗
12. A fault diagnosis method for rotating machinery with semi-supervised graph convolutional network and images converted from vibration signals10.1109/JSEN.2023.3267427 ↗
13. Fault diagnosis method of low-speed rolling bearing based on acoustic emission signal and subspace embedded feature distribution alignment10.1109/TII.2020.3028103 ↗
14. Li, C., Chen, C., and Gu, X. (2023). Acoustic-based rolling bearing fault diagnosis using a co-prime circular microphone array. Sensors, 23.10.2139/ssrn.4334582 ↗
15. Shiri, H., Wodecki, J., Ziętek, B., and Zimroz, R. (2021). Inspection robotic UGV platform and the procedure for an acoustic signal-based fault detection in belt conveyor idler. Energies, 14.10.3390/en14227646 ↗
16. High-speed train wheel set bearing fault diagnosis and prognostics: Research on acoustic emission detection mechanism10.1016/j.ymssp.2022.109325 ↗
17. Sound based fault diagnosis for RPMs based on multi-scale fractional permutation entropy and two-scale algorithm10.1109/TVT.2021.3090419 ↗
18. Intelligent fault diagnosis of helical gearboxes with compressive sensing based non-contact measurements10.1016/j.isatra.2022.07.020 ↗
19. A fusion CWSMM-based framework for rotating machinery fault diagnosis under strong interference and imbalanced case10.1109/TII.2021.3125385 ↗
20. A novel multi-model cascade framework for pipeline defects detection based on machine vision10.1016/j.measurement.2023.113374 ↗
21. Yan, J., Liao, J., Gao, J., Zhang, W., Huang, C., and Yu, H. (2023). Fusion of audio and vibration signals for bearing fault diagnosis based on a quadratic convolution neural network. Sensors, 23.10.3390/s23229155 ↗
22. Tang, L., Wu, X., Wang, D., and Liu, X. (2023). A comparative experimental study of vibration and acoustic emission on fault diagnosis of low-speed bearing. IEEE Trans. Instrum. Meas., 72.10.1109/TIM.2023.3312761 ↗
23. Bearing fault diagnosis and prognosis using data fusion based feature extraction and feature selection10.1016/j.measurement.2021.110506 ↗
24. Extremely low-speed bearing fault diagnosis based on raw signal fusion and DE-1D-CNN network10.1007/s42417-023-01228-5 ↗
25. Fault diagnosis of rolling bearings based on acoustics and vibration engineering10.1109/ACCESS.2024.3466154 ↗
26. Bearing fault detection with vibration and acoustic signals: Comparison among different machine learning classification methods10.1016/j.engfailanal.2022.106515 ↗
27. Chu, Z., Xing, S., Han, B., and Wang, J. (2024). A novel cross-domain mechanical fault diagnosis method fusing acoustic and vibration signals by vision transformer. Sensors, 24.10.3390/s24165120 ↗
28. Liu, Y., Xiang, H., Jiang, Z., and Xiang, J. (2023). A domain adaptation ResNet model to detect faults in roller bearings using vibro-acoustic data. Sensors, 23.10.3390/s23063068 ↗
29. Train bogie bearings fault diagnosis model based on multimodal signal features and physics knowledge learning10.1109/TR.2024.3462713 ↗
30. Multi-sensor fusion fault diagnosis method of wind turbine bearing based on viewable neural networks10.1016/j.ress.2024.109980 ↗
31. Meng, Z., Zhu, J., Cao, S., Li, P., and Xu, C. (2023). Bearing fault diagnosis under multisensor fusion based on modal analysis and graph attention network. IEEE Trans. Instrum. Meas., 72.10.1109/TIM.2023.3301895 ↗
32. End-to-end heterogeneous graph neural networks for traffic assignment10.1016/j.trc.2024.104695 ↗
33. Cross-domain bearing fault diagnosis using dual-path convolutional neural networks and multi-parallel graph convolutional networks10.1016/j.isatra.2024.06.009 ↗
34. TodyNet: Temporal dynamic graph neural network for multivariate time series classification10.1016/j.ins.2024.120914 ↗
35. A class alignment method based on graph convolution neural network for bearing fault diagnosis in presence of missing data and changing working conditions10.1016/j.measurement.2022.111536 ↗
36. Spatial graph convolutional neural network via structured subdomain adaptation and domain adversarial learning for bearing fault diagnosis10.1016/j.neucom.2022.10.057 ↗
37. Ait Ichou, I., Elouaham, S., Nassiri, B., and Isknan, J. (2026). Deep Multimodal Learning for Heart Sound Classification Using CNN, Transformer, and BiLSTM with Attention. Symmetry, 18.10.3390/sym18040556 ↗
38. Kipf, T.N., and Welling, M. (2017, January 24–26). Semi-Supervised Classification with Graph Convolutional Networks. Proceedings of the 5th International Conference on Learning Representations (ICLR 2017), Toulon, France.
39. Li, R., Wang, S., Zhu, F., and Huang, J. (2018). Adaptive Graph Convolutional Neural Networks. Proceedings of the AAAI Conference on Artificial Intelligence, AAAI Press.10.1609/aaai.v32i1.11691 ↗
40. Few-shot learning for fault diagnosis with a dual graph neural network10.1109/TII.2022.3205373 ↗
41. Intelligent fault diagnosis of rolling bearings based on the visibility algorithm and graph neural networks10.1007/s40430-022-03913-0 ↗