公开(公告)号：CN118296302A

公开(公告)日：2024-07-05

申请号：CN202410540600.0

申请日：2024-04-30

Applicant: 浙江工业大学

Inventor： 黄亮 ,  张凌鸿 ,  徐翊宸 ,  池凯凯 ,  张书彬 ,  魏欣晨

IPC: G06F18/10 ,  G06F18/24 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/048 ,  G06F18/213 ,  G06N3/08

Abstract: 本发明公开了一种基于扩散模型的电磁信号调制分类方法和装置，首先获取电磁信号数据集，进行旋转与翻转操作，得到第一数据集；然后将第一数据集中每个样本复制为至少两份，得到第二数据集，然后将第二数据集中每份样本随机选择一段连续的序列作为掩码段，其余部分作为非掩码段，利用标准高斯分布将掩码段转化为随机纯噪声，得到掩码数据集；构建去噪扩散模型，采用第二数据集训练，利用训练好的去噪扩散模型，对掩码数据集中样本进行去噪，得到增强数据集。最后采用增强数据集训练自动调制识别模型，对待检测电磁信号进行检测。本发明利用去噪扩散模型以实现数据的重构来增强数据集，更好地解决了数据集不充分，模型泛化能力弱的问题。