【技术实现步骤摘要】
一种缺陷样本的生成方法、装置、电子设备及存储介质
[0001]本申请实施例涉及人工智能
,尤其涉及一种缺陷样本的生成方法、装置、电子设备及存储介质。
技术介绍
[0002]近年来,缺陷检测在许多领域都具有重要的意义。在制造业中,缺陷检测可以帮助企业及时发现生产线上的缺陷,防止缺陷产品流入市场。这不仅可以保证产品质量,提高用户满意度,同时也可以降低企业的成本和损失。或在医疗领域,缺陷检测可以帮助医生及时发现疾病的病灶和异常,从而提高诊断准确率和治疗效果。例如,在医学影像领域,缺陷检测可以帮助医生发现肿瘤、血管病变等病灶,从而辅助诊断和治疗。
[0003]目前,利用深度学习技术进行缺陷检测是主流的技术手段。然而,基于深度学习的监督方法需要大量的标记数据来训练模型,但实际上,标注数据往往是稀缺的,特别是对于缺陷数据的情况下。首先,由于真实的缺陷样本往往比正常样本难以获得,缺陷检测的样本数量往往有限,这会限制缺陷检测系统的性能。通过缺陷样本生成技术,可以合成大量的缺陷样本,从而使得缺陷检测系统可以更全面地学习缺陷的特征和模式,从而提高检测性能。
[0004]当前,已经有一些基于非监督学习的缺陷样本生成方法,例如,基于生成对抗网络(GAN)的方法,变分自编码器(VAE)等方法。这些方法可以通过随机生成数据来增加缺陷样本数量,但由于数据生成的不确定性,生成的缺陷样本可能不够准确和可靠,也存在着一定的缺陷偏差。为此,研究人员提出了扩散模型(Diffusion Model)缓解上述生成式模型的问题。扩散模型是一 ...
【技术保护点】
【技术特征摘要】
1.一种缺陷样本的生成方法,其特征在于,所述方法包括:对标准高斯噪声进行随机采样,将随机采样得到的噪声的分布作为当前时间步的噪声特征f
T
,并将所述当前时间步的噪声特征f
T
和预先获取的目标缺陷对应的语义标签y输入至预先训练好的缺陷生成网络CUNet中,通过所述预先训练好的缺陷生成网络CUNet预测出所述当前时间步的上一个时间步的扩散噪声的分布ε
θ
(f
T
,T,y);基于所述上一个时间步的扰动噪声的分布ε
θ
(f
T
,T,y)随机采样得到所述上一个时间步的扩散噪声;并根据所述上一个时间步的扩散噪声和所述当前时间步的噪声特征f
T
,计算所述上一个时间步的噪声特征f
T
‑1;将所述上一个时间步作为所述当前时间步,重复执行上述操作,直到计算出第0个时间步的噪声特征f0;基于第0个时间步的噪声特征f0生成所述目标缺陷对应的局部缺陷图像;将所述局部缺陷图像输入预先训练好的缺陷融合网络中,通过所述缺陷融合网络生成所述目标缺陷对应的缺陷样本。2.根据权利要求1所述的方法,其特征在于,在对标准高斯噪声进行随机采样之前,所述方法还包括:若所述缺陷生成网络CUNet不满足预先设置的第一收敛条件,则在预先构建的局部缺陷集I
defect
中提取出预定数量的缺陷样本作为当前批次的训练样本;针对所述当前批次的训练样本中的每一个训练样本,在预先构建的扩散步长集合中随机选择一个步长作为该训练样本对应的步长;使用所述当前批次的训练样本以及各个训练样本对应的步长对所述缺陷生成网络CUNet进行训练,重复执行上述操作,直到所述缺陷生成网络满足所述第一收敛条件。3.根据权利要求2所述的方法,其特征在于,在预先构建的局部缺陷集I
defect
中提取出预定数量的缺陷样本作为当前批次的训练样本之前,所述方法还包括:对预先构建的真实的缺陷样本集中的每一个缺陷样本进行标注,并将缺陷样本集中的真实的缺陷剪切下来形成原始的局部缺陷集将所述原始的局部缺陷集中的每一个缺陷样本分别作为第一缺陷样本,在所述原始的缺陷样本集中随机选择的一个缺陷样本作为各个第一缺陷样本对应的第二缺陷样本,并将各个第一缺陷样本与其对应的第二缺陷样本进行混合,得到各个新的虚拟样本和各个新的虚拟样本对应的标签;基于各个新的虚拟样本、各个新的虚拟样本对应的标签以及所述原始的局部缺陷集构建所述局部缺陷集I
defect
。4.根据权利要求1所述的方法,其特征在于,在将所述局部缺陷图像输入预先训练好的缺陷融合网络中之前,所述方法还包括:若所述缺陷融合网络不满足预先设置的第二收敛条件,则将所述局部缺陷图像添加至预先构建的正常样本中,得到不完美的缺陷样本集在所述不完美的缺陷样本集中随机选择一个不完美的缺陷样本作为当前训练样本;
使用所述当前训练样本对所述缺陷融合网络进行训练,重复执行上述操作,直到所述缺陷融合网络满足所述第二收敛条件。5.根据权利要4所述的方法,其特征在于,使用所述当前训练样本对所述缺陷融合网络进行训练,包括:若判别器不满足预先设置的第三收敛条件,则将所述当前训练样本输入至所述缺陷融合网络中,通过所述缺陷融合网络输出生成的缺陷样本x
fake
;使用所述生成的缺陷样本x
fake
对判别器进行训练,直到所述判别器满足所述第三收敛条件;基于所述生成的缺陷样本x
fake
和训练好的判别器对所述缺陷融合网络进行训练。6.根据权利要求5所述的方法,其特征在于,使用所述生成的缺陷样本x
fake
对判别器进行训练,包括:将所述生...
【专利技术属性】
技术研发人员:黄开竹,杨超龙,杨关禹,赵伟光,
申请(专利权)人:昆山杜克大学,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。