WMSST-CNN融合模型在轴承故障诊断中的应用

1. 项目背景与核心价值

轴承故障诊断一直是工业设备健康监测领域的重点难题。传统方法在面对非平稳振动信号时，往往难以准确捕捉故障特征。我在实际项目中发现，当轴承出现早期微弱故障时，振动信号中的冲击成分往往被噪声淹没，采用常规的FFT分析几乎无法识别。

小波多尺度同步压缩变换(WMSST)的出现为这个问题提供了突破性的解决方案。与普通小波变换相比，WMSST通过多尺度同步压缩操作，能将分散的时频能量重新聚集，形成清晰的时频脊线。去年在处理某风电场的齿轮箱故障案例时，我们对比了STFT、CWT和WMSST三种方法，只有WMSST在转速波动情况下仍能稳定提取出故障特征频率。

2. 技术方案设计思路

2.1 WMSST-CNN融合架构

WTCNN模型的核心创新在于将WMSST的时频分析优势与CNN的特征学习能力有机结合。具体实现路径如下：

1. 信号预处理层：对原始振动信号进行去趋势和归一化处理
2. WMSST变换层：生成高分辨率的时频图像
3. CNN特征提取层：采用多尺度卷积核捕捉局部特征
4. 分类决策层：通过softmax输出故障类型概率

关键提示：WMSST的尺度参数选择直接影响时频图像质量。经过多次实验，我们发现对于轴承故障信号，采用Morlet小波基，尺度范围设为1-128，能较好平衡计算效率和分辨率。

2.2 时频图像生成细节

WMSST的实现包含三个关键步骤：

1. 连续小波变换(CWT)：

   [cfs,frq] = cwt(signal, 'amor', Fs);

2. 瞬时频率估计：

   omega = abs(diff(unwrap(angle(cfs))));

3. 同步压缩操作：

   Ts = (0:length(signal)-1)/Fs; [~,idx] = min(abs(frq - omega),[],1); sst = accumarray([idx(:),round(Ts*length(frq))'],abs(cfs(:)));

在实际应用中，我们发现对WMSST结果进行对数压缩能增强可视化效果：

imagesc(20*log10(abs(sst)+eps))

3. MATLAB实现关键代码解析

3.1 数据准备模块

CWRU轴承数据需要特别注意采样对齐问题。我们采用以下预处理流程：

function [trainData, testData] = prepareData(dataPath, splitRatio) % 加载.mat文件中的振动数据 load(fullfile(dataPath, 'bearing_data.mat')); % 标签one-hot编码 labels = categorical(labels); labels = onehotencode(labels,2); % 时频变换批处理 parfor i = 1:length(signals) tfImages(:,:,i) = wmsst_transform(signals(i,:), Fs); end % 数据集划分 [trainInd,~,testInd] = dividerand(size(tfImages,3),... splitRatio,0,1-splitRatio); trainData = {tfImages(:,:,trainInd), labels(trainInd,:)}; testData = {tfImages(:,:,testInd), labels(testInd,:)}; end

3.2 CNN网络构建

采用动态调整感受野的设计策略：

function net = createWTCNN(inputSize, numClasses) layers = [ imageInputLayer(inputSize) % 多尺度卷积分支 convolution2dLayer(3,16,'Padding','same') batchNormalizationLayer reluLayer convolution2dLayer(5,16,'Padding','same') batchNormalizationLayer reluLayer % 特征融合 depthConcatenationLayer(2,'Name','concat') maxPooling2dLayer(2,'Stride',2) convolution2dLayer(3,32,'Padding','same') batchNormalizationLayer reluLayer convolution2dLayer(3,32,'Padding','same') batchNormalizationLayer reluLayer fullyConnectedLayer(numClasses) softmaxLayer classificationLayer ]; options = trainingOptions('adam', ... 'InitialLearnRate',0.001, ... 'MaxEpochs',30, ... 'Plots','training-progress'); net = trainNetwork(trainData, layers, options); end

4. 实战问题与解决方案

4.1 时频图像质量优化

常见问题：转速波动导致时频脊线断裂 解决方法：采用转速同步平均技术

function avgImage = rpm_sync_average(signal, rpm, Fs) period = 60/rpm * Fs; segments = buffer(signal, round(period), 0); avgSignal = mean(segments, 2); avgImage = wmsst_transform(avgSignal, Fs); end

4.2 样本不均衡处理

对于CWRU数据中不同故障类型样本量差异大的问题，我们采用：

1. 时域波形平移增强
2. 添加高斯噪声增强
3. SMOTE过采样技术

实现代码：

function [augImages, augLabels] = augmentData(images, labels) augImages = []; augLabels = []; for i = 1:size(images,3) % 原始样本 augImages = cat(3, augImages, images(:,:,i)); augLabels = [augLabels; labels(i,:)]; % 时移增强 shifted = circshift(images(:,:,i), randi(10)); augImages = cat(3, augImages, shifted); augLabels = [augLabels; labels(i,:)]; % 噪声增强 noisy = imnoise(images(:,:,i), 'gaussian'); augImages = cat(3, augImages, noisy); augLabels = [augLabels; labels(i,:)]; end end

5. 性能评估与对比实验

我们在CWRU数据集上进行了三组对比实验：

迁移学习实现关键代码：

function net = transferLearning(pretrainedNet, newData, numClasses) layers = pretrainedNet.Layers; % 替换最后三层 layers(end-2:end) = [ fullyConnectedLayer(numClasses) softmaxLayer classificationLayer ]; options = trainingOptions('adam', ... 'InitialLearnRate',0.0001, ... 'MaxEpochs',10); net = trainNetwork(newData, layers, options); end

6. 工程应用建议

1. 实时性优化：将WMSST变换转换为C++ MEX函数可提升5-8倍速度

   mex wmsst_mex.cpp -I/usr/local/include -L/usr/local/lib -lfftw3

2. 模型轻量化：采用深度可分离卷积减少参数量

   convolution2dLayer(3,1,'Padding','same') convolution2dLayer(1,16,'Padding','same')

3. 跨设备适配：开发统一的时频图像标准化方法

   function normImg = normalize_tfimage(img) img = img - min(img(:)); normImg = img / max(img(:)); normImg = histeq(normImg); % 直方图均衡化 end

在实际部署中，我们发现将采样率统一为12.8kHz，时频图像尺寸规范化为128×128像素，能获得最佳的跨设备兼容性。对于不同型号的轴承，建议先进行少量样本的迁移学习微调。