条件生成扩散模型

即征即用,扩充故障样本

针对轴承故障诊断中的类别不平衡问题,通过嵌入残差结构和信号标签条件,生成高质量且类别可控的样本,从而提升不平衡诊断精度。

Yu, Tian · Li C. · 黄洁 · Xiangqu, Xiao · 张孝远 · 黎育红 · Bitao, Fu

Reliability Engineering and System Safety 2024

技术优势

质量多样性兼得

扩散模型通过马尔可夫链逐步加噪去噪,比GAN训练更稳定,缓解模式坍塌,生成样本既真实又多样。

类别可控

加入信号标签作为条件信息,采样时能指定故障类型生成样本。

特征表达更强

改进噪声预测网络,增强层内和层间特征交互,提升生成质量。

应用场景