心音信号分类实战:小波变换与深度学习应用
2026/7/24 6:03:15 网站建设 项目流程

1. 项目概述:心音信号分类实战

最近完成了一个心音信号(PCG)分类的小项目,过程中踩了不少坑也积累了一些可直接运行的代码。这个项目主要使用小波变换和深度学习技术对心音信号进行分类,核心是提取心音信号的特征表示并构建分类模型。

心音信号是心脏活动产生的声学信号,包含丰富的病理信息。传统心音分析依赖医生听诊,存在主观性强、效率低等问题。通过信号处理和深度学习技术实现自动分类,可以辅助医生诊断,提高工作效率。

2. 技术方案设计

2.1 整体技术路线

项目采用的技术路线如下:

  1. 数据预处理:对原始心音信号进行去噪和标准化
  2. 特征提取:使用小波变换获取时频特征
  3. 模型构建:基于预训练CNN模型进行迁移学习
  4. 模型评估:验证分类性能

2.2 关键技术选择

2.2.1 小波变换特征提取

小波变换相比傅里叶变换更适合分析非平稳信号。我们使用连续小波变换(CWT)获取心音信号的时频表示,生成尺度图(scalogram)作为CNN的输入。

关键参数设置:

  • 小波基函数:Morlet小波
  • 尺度数:64
  • 采样频率:1000Hz
  • 频率范围:20-400Hz
2.2.2 深度学习模型选择

考虑模型性能和计算资源限制,我们测试了两种预训练模型:

  1. GoogLeNet:22层深度,计算量适中
  2. SqueezeNet:轻量级模型,参数少但性能不错

3. 实现细节与核心代码

3.1 数据准备

% 加载心音数据集 load('PCGData.mat'); % 创建训练和验证目录 dataDir = 'pcg_data'; mkdir(dataDir); categories = {'normal', 'murmur', 'extrasystole'}; for i = 1:length(categories) mkdir(fullfile(dataDir, categories{i})); end

3.2 小波变换实现

% 创建小波滤波器组 fb = cwtfilterbank('SignalLength', length(signal), ... 'SamplingFrequency', fs, ... 'VoicesPerOctave', 12); % 计算小波系数 [cfs, frq] = wt(fb, signal); % 生成尺度图 figure; pcolor(t, frq, abs(cfs)); shading interp; title('PCG Scalogram'); xlabel('Time (s)'); ylabel('Frequency (Hz)');

3.3 迁移学习实现

% 加载预训练GoogLeNet net = googlenet; % 修改网络结构 numClasses = 3; newLayers = [ dropoutLayer(0.6, 'Name', 'new_dropout') fullyConnectedLayer(numClasses, 'Name', 'new_fc', ... 'WeightLearnRateFactor', 5, 'BiasLearnRateFactor', 5) softmaxLayer('Name', 'new_softmax') classificationLayer('Name', 'new_classoutput')]; net = replaceLayer(net, 'loss3-classifier', newLayers(2)); net = replaceLayer(net, 'prob', newLayers(3)); net = replaceLayer(net, 'output', newLayers(4));

4. 训练与优化

4.1 训练参数设置

options = trainingOptions('sgdm', ... 'MiniBatchSize', 16, ... 'MaxEpochs', 30, ... 'InitialLearnRate', 1e-4, ... 'ValidationData', imdsValidation, ... 'ValidationFrequency', 30, ... 'Verbose', true, ... 'Plots', 'training-progress');

4.2 模型训练

[net, info] = trainNetwork(imdsTrain, net, options);

4.3 性能评估

% 测试集预测 YPred = classify(net, imdsTest); YTest = imdsTest.Labels; % 计算准确率 accuracy = sum(YPred == YTest)/numel(YTest); disp(['Test accuracy: ', num2str(accuracy*100), '%']) % 混淆矩阵 plotconfusion(YTest, YPred)

5. 踩坑经验与解决方案

5.1 数据不平衡问题

问题现象:正常心音样本远多于异常样本,导致模型偏向多数类。

解决方案

  1. 数据增强:对少数类样本进行时间拉伸、添加噪声等
  2. 类别权重:在损失函数中为不同类别分配不同权重
  3. 过采样:使用SMOTE等方法生成合成样本

5.2 小波参数选择

问题现象:不同小波基函数和参数对结果影响大。

解决方案

  1. 通过网格搜索确定最优小波基
  2. 根据心音特性调整尺度范围
  3. 可视化不同参数下的尺度图,选择区分度好的

5.3 模型过拟合

问题现象:训练集准确率高但验证集性能差。

解决方案

  1. 增加Dropout层和L2正则化
  2. 使用早停策略
  3. 采用数据增强扩充训练集

6. 性能对比与结果分析

我们在自有数据集上对比了不同方法的性能:

方法准确率灵敏度特异性参数量
传统特征+SVM82.3%78.5%85.1%-
原始信号+CNN87.6%83.2%90.1%2.3M
小波+GoogLeNet93.8%91.5%95.2%7.0M
小波+SqueezeNet92.1%89.7%93.8%1.2M

从结果可以看出:

  1. 小波变换能有效提取心音特征
  2. 迁移学习显著提升小样本下的性能
  3. SqueezeNet在参数量大幅减少的情况下保持了不错的性能

7. 实际应用建议

  1. 数据采集:确保心音采集环境安静,使用专业心音传感器
  2. 预处理:必须进行带通滤波(20-400Hz)去除噪声
  3. 模型部署:考虑使用SqueezeNet等轻量模型以便在移动端部署
  4. 结果解释:模型输出应结合临床其他检查结果综合判断

这个项目完整代码已开源,包含数据预处理、特征提取、模型训练和评估的全流程。通过合理调整参数,可以适应不同的心音分类任务。在实际应用中还需要考虑实时性要求、计算资源限制等因素,选择最适合的方案。

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询