小样本下SE-ResNet与元迁移学习的变工况轴承故障诊断

Bearing fault diagnosis under few-shot and variable working conditions using SE-ResNet and Meta-Transfer learning

  • 摘要: 针对轴承在变工况下样本分布不同、故障样本少和一些小样本算法特征提取有限,导致轴承故障诊断精度低及模型泛化能力弱的问题,提出了小样本下嵌入压缩、激励的残差网络(SE-ResNet)与元迁移学习(MTL)的变工况轴承故障诊断方法。将采集的不同工况下轴承一维振动信号通过连续小波变换(CWT)转换成对应工况下的时频图像,从而将轴承故障诊断问题转换为图像识别问题;引入压缩-激励注意力机制,构建了一种SE-ResNet的骨干网络模型,以聚焦于更有效的特征通道,增强特征提取表征能力;借助迁移学习能提供良好的深层网络初始参数和元学习能快速学习的优势,依次进行预训练与元迁移训练,得到利用少量样本微调便能达到高精度的元迁移网络,进而实现变工况下轴承的故障诊断;通过两个基准数据集和实验室搭建的轴承故障模拟试验台进行验证,并与其他方法进行对比分析,结果表明,所提方法在小样本、变工况下对轴承故障诊断具有更高的识别精度和泛化性能。

     

    Abstract: Traditional bearing fault diagnosis methods often suffer from low accuracy and weak model generalization under varying working conditions due to diverse sample distributions, scarcity of fault samples, and limited feature extraction capabilities of some few-shot learning algorithms. To address these challenges, this paper proposes a novel method for variable condition bearing fault diagnosis that combines a squeeze-and-excitation residual network (SE-ResNet) with meta-transfer learning (MTL). One-dimensional bearing vibration signals collected under different working conditions are converted into time-frequency images using continuous wavelet transform (CWT), thereby transforming the bearing fault diagnosis task into an image recognition problem. A squeeze-and-excitation (SE) attention mechanism is introduced to construct an SE-ResNet backbone network model. This focuses on more effective feature channels, thereby enhancing feature extraction and representation capabilities. Leveraging the advantages of transfer learning (which provides robust initial deep network parameters) and meta-learning (which enables rapid adaptation), the model undergoes sequential pre-training and meta-transfer training. This process yields a high-precision meta-transfer network that can be fine-tuned with only a small number of samples, ultimately achieving accurate bearing fault diagnosis under variable working conditions. The proposed method is validated using two benchmark datasets and a bearing fault simulation test bench developed in the laboratory. Comparative analysis with other methods demonstrates that the proposed method exhibits higher recognition accuracy and superior generalization performance for bearing fault diagnosis under both few-shot and variable working conditions.

     

/

返回文章
返回