arXiv:2410.10125cs.SDeess.AS2024-10被引 6

用生成模型合成心脏听诊音,提升分类模型鲁棒性。

Generative Deep Learning and Signal Processing for Data Augmentation of Cardiac Auscultation Signals: Improving Model Robustness Using Synthetic Audio

  • 结合扩散模型生成合成音频,扩充稀缺标注数据
  • 在多个数据集上同时提升模型内分布与分布外表现
  • 适合需要高鲁棒性的医疗信号分类研究者

准确解读心脏听诊信号对心血管疾病诊断至关重要,但标注数据匮乏制约了分类模型的训练。以往研究多关注模型性能,而忽视了模型鲁棒性——即在分布内与分布外数据上的表现,以马修相关系数等指标衡量。本文通过传统音频处理与WaveGrad、DiffWave扩散模型生成条件合成音频,构建增强数据集,用于训练基于卷积神经网络的分类模型。实验表明,在多个数据集上,该方法显著提升了模型的分布内与分布外性能,不仅提高准确率,还改善了平衡准确率和马修相关系数,有效缓解数据不平衡问题,从而构建更具泛化能力的鲁棒分类器。

原文摘要 · Abstract (English)

Accurately interpreting cardiac auscultation signals plays a crucial role in diagnosing and managing cardiovascular diseases. However, the paucity of labelled data inhibits classification models' training. Researchers have turned to generative deep learning techniques combined with signal processing to augment the existing data and improve cardiac auscultation classification models to overcome this challenge. However, the primary focus of prior studies has been on model performance as opposed to model robustness. Robustness, in this case, is defined as both the in-distribution and out-of-distribution performance by measures such as Matthew's correlation coefficient. This work shows that more robust abnormal heart sound classifiers can be trained using an augmented dataset. The augmentations consist of traditional audio approaches and the creation of synthetic audio conditionally generated using the WaveGrad and DiffWave diffusion models. It is found that both the in-distribution and out-of-distribution performance can be improved over various datasets when training a convolutional neural network-based classification model with this augmented dataset. With the performance increase encompassing not only accuracy but also balanced accuracy and Matthew's correlation coefficient, an augmented dataset significantly contributes to resolving issues of imbalanced datasets. This, in turn, helps provide a more general and robust classifier.

生成模型医疗信号数据增强扩散模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。