计算机科学
卷积神经网络
人工智能
降噪
噪音(视频)
推论
模式识别(心理学)
小波
方位(导航)
机器学习
特征提取
先验概率
断层(地质)
数据挖掘
深度学习
变压器
频道(广播)
足迹
噪声测量
图像复原
信号(编程语言)
故障检测与隔离
稳健性(进化)
高斯噪声
人工神经网络
计算机视觉
利用
概率逻辑
棱锥(几何)
作者
Zhichao Gong,Jiarula Yasenjiang,Wenhao Wang,Yingjun Zhao,Hebo Hao
摘要
ABSTRACT Rolling bearings play a vital role in rotating machinery, yet their fault diagnosis under real industrial conditions is challenged by high noise levels, limited labeled data, and class imbalance. To address these issues, this paper proposes a cross‐modal fusion network (CMFN) for few‐shot bearing diagnosis, designed to exploit the complementarity of one‐dimensional vibration sequences and two‐dimensional time–frequency spectrograms. The signal branch employs a noise‐aware pyramid convolutional extractor to capture multi‐scale temporal patterns and incorporates sample‐wise reweighting for enhanced robustness. Meanwhile, the image branch integrates a noise‐to‐vector enhanced efficient channel attention (N2V–ECA) module, leveraging noise priors derived from the signal branch to reduce channel bias and suppress noise‐dominated responses. A feature‐wise linear modulation (FiLM) fusion module further conditions image features on signal semantics, enabling fine‐grained cross‐modal alignment with lightweight computation. Extensive experiments with rigorous statistical validation (including fivefold cross‐validation) were conducted on the Jiangnan University and Harbin Institute of Technology bearing datasets under strict few‐shot settings (30 training samples per class). CMFN consistently outperformed state‐of‐the‐art single‐ and dual‐modal baselines, including recent heavy‐duty Transformer architectures. It achieved an outstanding average test accuracy of 99.0% with an extremely lightweight footprint of only 0.86M parameters and a 2.92 ms inference latency. Furthermore, it demonstrated exceptional noise resilience (maintaining 79.0%–97.0% robust accuracy across −5 to 3 dB SNR) and obtained 97.2% average accuracy on a heterogeneous unseen aero‐engine dataset without architectural adaptation. These results thoroughly demonstrate its superiority as an integrated, highly generalizable framework for industrial edge deployment.
科研通智能强力驱动
Strongly Powered by AbleSci AI