1. 项目概述:心音信号分类实战
最近完成了一个心音信号(PCG)分类的小项目,过程中踩了不少坑也积累了一些可直接运行的代码。这个项目主要使用小波变换和深度学习技术对心音信号进行分类,核心是提取心音信号的特征表示并构建分类模型。
心音信号是心脏活动产生的声学信号,包含丰富的病理信息。传统心音分析依赖医生听诊,存在主观性强、效率低等问题。通过信号处理和深度学习技术实现自动分类,可以辅助医生诊断,提高工作效率。
2. 技术方案设计
2.1 整体技术路线
项目采用的技术路线如下:
- 数据预处理:对原始心音信号进行去噪和标准化
- 特征提取:使用小波变换获取时频特征
- 模型构建:基于预训练CNN模型进行迁移学习
- 模型评估:验证分类性能
2.2 关键技术选择
2.2.1 小波变换特征提取
小波变换相比傅里叶变换更适合分析非平稳信号。我们使用连续小波变换(CWT)获取心音信号的时频表示,生成尺度图(scalogram)作为CNN的输入。
关键参数设置:
- 小波基函数:Morlet小波
- 尺度数:64
- 采样频率:1000Hz
- 频率范围:20-400Hz
2.2.2 深度学习模型选择
考虑模型性能和计算资源限制,我们测试了两种预训练模型:
- GoogLeNet:22层深度,计算量适中
- 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})); end3.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 数据不平衡问题
问题现象:正常心音样本远多于异常样本,导致模型偏向多数类。
解决方案:
- 数据增强:对少数类样本进行时间拉伸、添加噪声等
- 类别权重:在损失函数中为不同类别分配不同权重
- 过采样:使用SMOTE等方法生成合成样本
5.2 小波参数选择
问题现象:不同小波基函数和参数对结果影响大。
解决方案:
- 通过网格搜索确定最优小波基
- 根据心音特性调整尺度范围
- 可视化不同参数下的尺度图,选择区分度好的
5.3 模型过拟合
问题现象:训练集准确率高但验证集性能差。
解决方案:
- 增加Dropout层和L2正则化
- 使用早停策略
- 采用数据增强扩充训练集
6. 性能对比与结果分析
我们在自有数据集上对比了不同方法的性能:
| 方法 | 准确率 | 灵敏度 | 特异性 | 参数量 |
|---|---|---|---|---|
| 传统特征+SVM | 82.3% | 78.5% | 85.1% | - |
| 原始信号+CNN | 87.6% | 83.2% | 90.1% | 2.3M |
| 小波+GoogLeNet | 93.8% | 91.5% | 95.2% | 7.0M |
| 小波+SqueezeNet | 92.1% | 89.7% | 93.8% | 1.2M |
从结果可以看出:
- 小波变换能有效提取心音特征
- 迁移学习显著提升小样本下的性能
- SqueezeNet在参数量大幅减少的情况下保持了不错的性能
7. 实际应用建议
- 数据采集:确保心音采集环境安静,使用专业心音传感器
- 预处理:必须进行带通滤波(20-400Hz)去除噪声
- 模型部署:考虑使用SqueezeNet等轻量模型以便在移动端部署
- 结果解释:模型输出应结合临床其他检查结果综合判断
这个项目完整代码已开源,包含数据预处理、特征提取、模型训练和评估的全流程。通过合理调整参数,可以适应不同的心音分类任务。在实际应用中还需要考虑实时性要求、计算资源限制等因素,选择最适合的方案。