一种基于扩散模型的孕产妇心电图数据增强方法

本发明提供了一种基于扩散模型的孕产妇心电图数据增强方法，涉及心电图信号处理领域，该方法通过获取心电信号，记录心电信号的元数据，并统一信号采样频率；对统一信号采样频率的心电信号进行滑动切片，并初始化标签向量；设置输入噪声模块，生成长度为、采样频率为fs的噪声序列，设置LN个不同噪声等级；设置LOSS函数，训练模型，计算KL散度损失，反向传播更新参数，迭代训练得到心电信号扩散模型；应用训练后的心电信号扩散模型，生成满足多标签诊断需求的心电图数据。本发明通过扩散模型学习已有数据库的ECG信号，从而可以根据需求生成所需要的单发或并发异位心搏的心电数据。