Continuous FMD-Net: A Physics-Driven Algorithm-Unrolling Network with Spatio-Temporal Dual Sparsity for Adaptive Fault Diagnosis

Author(s)

Abstract
Extracting and demodulating weak fault signatures from rotating machinery under heavy background noise remains a significant challenge. Conventional methods such as Feature Mode Decomposition (FMD) rely on manual parameter tuning and are highly susceptible to mode mixing. To address these limitations, this paper proposes Continuous FMD-Net, a fully differentiable framework for adaptive bearing fault diagnosis. The network unrolls the FMD algorithm into deep neural layers and incorporates three key designs: (i) a trainable time-domain spatial attention mask that adaptively determines the effective window length L of each channel’s filter via backpropagation; (ii) a channel gating mechanism that prunes redundant channels to automatically optimize the number of modes K; and (iii) a cross-correlation orthogonality penalty, embedded in a physics-informed joint loss function, that enforces mutual repulsion among parallel channels to effectively suppress mode mixing. Experiments on the Case Western Reserve University (CWRU) bearing dataset demonstrate that the proposed network achieves an accuracy approaching 100% under 0 dB noise, and reaches 90.13% on the −8 dB strong-noise blind test set, significantly outperforming all compared methods. Convergence curves and UMAP visualizations further confirm that the network learns highly decoupled and discriminative fault features. This work provides a feasible pathway for integrating physical demodulation priors with end-to-end deep learning.

Keywords
Rolling bearing fault diagnosis; Algorithm unrolling network; Feature mode decomposition; Mode mixing; Physics-informed machine learning; Adaptive filtering

Cite this paper
Meng li, Jiaqi Xie, Hangyuan Gao, Continuous FMD-Net: A Physics-Driven Algorithm-Unrolling Network with Spatio-Temporal Dual Sparsity for Adaptive Fault Diagnosis , SCIREA Journal of Mechanical Engineering. Volume 6, Issue 1, February 2026 | PP. 1-24. 10.54647/mechanical460143

References

[ 1 ]	Lei, Y., Lin, J., He, Z., & Zuo, M. J. (2013). A review on empirical mode decomposition in fault diagnosis of rotating machinery. Mechanical Systems and Signal Processing, 35(1–2), 108–126. https://doi.org/10.1016/j.ymssp.2012.09.015
[ 2 ]	Randall, R. B., & Antoni, J. (2011). Rolling element bearing diagnostics — A tutorial. Mechanical Systems and Signal Processing, 25(2), 485–520. https://doi.org/10.1016/j.ymssp.2010.07.017
[ 3 ]	Nectoux, P., Gouriveau, R., Medjaher, K., Ramasso, E., Chebel-Morello, B., Zerhouni, N., & Varnier, C. (2012). PRONOSTIA: An experimental platform for bearings accelerated degradation tests. In IEEE International Conference on Prognostics and Health Management (pp. 1–8). Denver, CO, USA.
[ 4 ]	Jardine, A. K. S., Lin, D., & Banjevic, D. (2006). A review on machinery diagnostics and prognostics implementing condition-based maintenance. Mechanical Systems and Signal Processing, 20(7), 1483–1510. https://doi.org/10.1016/j.ymssp.2005.09.012
[ 5 ]	Peng, Z. K., & Chu, F. L. (2004). Application of the wavelet transform in machine condition monitoring and fault diagnostics: A review with bibliography. Mechanical Systems and Signal Processing, 18(2), 199–221. https://doi.org/10.1016/S0888-3270(03)00075-X
[ 6 ]	Zhao, R., Yan, R., Chen, Z., Mao, K., Wang, P., & Gao, R. X. (2019). Deep learning and its applications to machine health monitoring. Mechanical Systems and Signal Processing, 115, 213–237. https://doi.org/10.1016/j.ymssp.2018.05.050
[ 7 ]	Antoni, J. (2006). The spectral kurtosis: A useful tool for characterising non-stationary signals. Mechanical Systems and Signal Processing, 20(2), 282–307. https://doi.org/10.1016/j.ymssp.2004.09.001
[ 8 ]	Wang, D., Tsui, K. L., & Miao, Q. (2018). Prognostics and health management: A review of vibration based bearing and gear health indicators. IEEE Access, 6, 665–676. https://doi.org/10.1109/ACCESS.2017.2774261.
[ 9 ]	He, M., & He, D. (2017). Deep learning based approach for bearing fault diagnosis. IEEE Transactions on Industry Applications, 53(3), 3057–3065. https://doi.org/10.1109/TIA.2017.2661250
[ 10 ]	Yan, R., Gao, R. X., & Chen, X. (2014). Wavelets for fault diagnosis of rotary machines: A review with applications. Signal Processing, 96, 1–15. https://doi.org/10.1016/j.sigpro.2013.04.015
[ 11 ]	Huang, N. E., Shen, Z., Long, S. R., Wu, M. C., Shih, H. H., Zheng, Q., & Liu, H. H. (1998). The empirical mode decomposition and the Hilbert spectrum for nonlinear and non-stationary time series analysis. Proceedings of the Royal Society of London Series A, 454(1971), 903–995. https://doi.org/10.1098/rspa.1998.0193
[ 12 ]	Rilling, G., & Flandrin, P. (2008). One or two frequencies? The empirical mode decomposition answers. IEEE Transactions on Signal Processing, 56(1), 85–95. https://doi.org/10.1109/TSP.2007.906771
[ 13 ]	Wu, Z., & Huang, N. E. (2009). Ensemble empirical mode decomposition: A noise-assisted data analysis method. Advances in Adaptive Data Analysis, 1(1), 1–41. https://doi.org/10.1142/S1793536909000047
[ 14 ]	Torres, M. E., Colominas, M. A., Schlotthauer, G., & Flandrin, P. (2011). A complete ensemble empirical mode decomposition with adaptive noise. In 2011 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) (pp. 4144–4147). Prague, Czech Republic. https://doi.org/10.1109/ICASSP.2011.5947265
[ 15 ]	Dragomiretskiy, K., & Zosso, D. (2014). Variational mode decomposition. IEEE Transactions on Signal Processing, 62(3), 531–544. https://doi.org/10.1109/TSP.2013.2288675
[ 16 ]	Li, Z., Chen, J., Zi, Y., & Pan, J. (2017). Independence-oriented VMD to identify fault feature for wheel set bearing fault diagnosis of high speed locomotive. Mechanical Systems and Signal Processing, 85, 512–529. https://doi.org/10.1016/j.ymssp.2016.08.042
[ 17 ]	McDonald, G. L., & Zhao, Q. (2017). Multipoint optimal minimum entropy deconvolution and convolution fix: Application to vibration fault detection. Mechanical Systems and Signal Processing, 82, 461–477. https://doi.org/10.1016/j.ymssp.2016.05.036
[ 18 ]	Miao, Y., Zhao, M., & Lin, J. (2020). Identification of mechanical compound-fault based on the improved parameter-adaptive variational mode decomposition. ISA Transactions, 84, 82–95. https://doi.org/10.1016/j.isatra.2018.10.008
[ 19 ]	LeCun, Y., Bengio, Y., & Hinton, G. (2015). Deep learning. Nature, 521(7553), 436–444. https://doi.org/10.1038/nature14539
[ 20 ]	Zhang, W., Peng, G., Li, C., Chen, Y., & Zhang, Z. (2017). A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals. Sensors, 17(2), 425. https://doi.org/10.3390/s17020425
[ 21 ]	Zhao, R., Yan, R., Wang, J., & Mao, K. (2017). Learning to monitor machine health with convolutional bi-directional LSTM networks. Sensors, 17(2), 273. https://doi.org/10.3390/s17020273
[ 22 ]	Hou, L., & Bergmann, N. W. (2013). Novel industrial wireless sensor networks for machine condition monitoring and fault diagnosis. IEEE Transactions on Instrumentation and Measurement, 61(10), 2787–2798. https://doi.org/10.1109/TIM.2012.2200817
[ 23 ]	Chen, Z., Gryllias, K., & Li, W. (2020). Mechanical fault diagnosis using convolutional neural networks and extreme learning machine. Mechanical Systems and Signal Processing, 133, 106272. https://doi.org/10.1016/j.ymssp.2019.106272
[ 24 ]	Antoni, J., & Randall, R. B. (2006). The spectral kurtosis: Application to the vibratory surveillance and diagnostics of rotating machines. Mechanical Systems and Signal Processing, 20(2), 308–331. https://doi.org/10.1016/j.ymssp.2004.09.002
[ 25 ]	Shao, H., Jiang, H., Zhao, H., & Wang, F. (2017). A novel deep autoencoder feature learning method for rotating machinery fault diagnosis. Mechanical Systems and Signal Processing, 95, 187–204. https://doi.org/10.1016/j.ymssp.2017.03.034
[ 26 ]	Karniadakis, G. E., Kevrekidis, I. G., Lu, L., Perdikaris, P., Wang, S., & Yang, L. (2021). Physics-informed machine learning. Nature Reviews Physics, 3(6), 422–440. https://doi.org/10.1038/s42254-021-00314-5
[ 27 ]	Raissi, M., Perdikaris, P., & Karniadakis, G. E. (2019). Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations. Journal of Computational Physics, 378, 686–707. https://doi.org/10.1016/j.jcp.2018.10.045
[ 28 ]	Zhang, S., Zhang, S., Wang, B., & Habetler, T. G. (2020). Deep learning algorithms for bearing fault diagnostics — A comprehensive study. IEEE Access, 8, 29857–29881. https://doi.org/10.1109/ACCESS.2020.2972859
[ 29 ]	Wang, H., Liu, Z., Peng, D., & Qin, Y. (2022). Understanding and learning discriminant features based on multiattention 1DCNN for wheelset bearing fault diagnosis. IEEE Transactions on Industrial Informatics, 18(9), 5735–5745. https://doi.org/10.1109/TII.2021.3136069
[ 30 ]	Smith, W. A., & Randall, R. B. (2015). Rolling element bearing diagnostics using the Case Western Reserve University data: A benchmark study. Mechanical Systems and Signal Processing, 64–65, 100–131. https://doi.org/10.1016/j.ymssp.2015.04.021