[发明专利]故障诊断模型的训练方法、装置、电子设备及存储介质

权利要求书

1.一种故障诊断模型的训练方法，其特征在于，所述故障诊断模型包括卷积层和域适配层，所述训练方法包括：

将第一设备对应的样本目标域信号和第二设备对应的样本源域信号分别输入到各自对应的卷积层中，得到第一目标域特征和第一源域特征；其中，所述第一设备与所述第二设备相关联；

将所述第一目标域特征和所述第一源域特征分别输入到各自对应的域适配层中，得到第二目标域特征和第二源域特征，基于所述第二目标域特征和所述第二源域特征，确定所述故障诊断模型的域适配损失；

基于所述第二目标域特征和所述第二源域特征，确定所述第一设备分别归属于预设故障类型的第一概率，基于所述第一概率以及所述第一设备对应的目标故障类型，确定所述故障诊断模型的第一分类损失；

若所述故障诊断模型的域适配损失和第一分类损失满足第一训练条件，则将当前的故障诊断模型确定为目标故障诊断模型。

2.根据权利要求1所述的故障诊断模型的训练方法，其特征在于，所述故障诊断模型还包括域混淆层，所述训练方法还包括：

将所述第二目标域特征和所述第二源域特征输入到各自对应的域混淆层中，得到第三目标域特征和第三源域特征，基于所述第三目标域特征和所述第三源域特征，确定所述故障诊断模型的域混淆损失；

基于所述第三目标域特征和所述第三源域特征，确定所述第一设备分别归属于预设故障类型的第二概率，基于所述第二概率以及所述第一设备对应的目标故障类型，确定所述故障诊断模型的第二分类损失；

若所述故障诊断模型的域适配损失、域混淆损失以及第二分类损失满足第二训练条件，则将当前的故障诊断模型确定为目标故障诊断模型。

3.根据权利要求1所述的故障诊断模型的训练方法，其特征在于，所述故障诊断模型对应有多级域适配层，所述基于所述第二目标域特征和所述第二源域特征，确定所述故障诊断模型的域适配损失，包括：

针对每级域适配层，基于该级域适配层对应的第二目标域特征和第二源域特征，确定该级域适配层对应的域适配损失；

基于每级域适配层对应的域适配损失，确定所述故障诊断模型的域适配损失。

4.根据权利要求2所述的故障诊断模型的训练方法，其特征在于，所述域混淆层对应有二元领域分类器，所述基于所述第三目标域特征和所述第三源域特征，确定所述故障诊断模型的域混淆损失，包括：

将所述第三目标域特征和所述第三源域特征分别输入到所述二元领域分类器中，得到所述第三目标域特征所属的第一领域类别，以及所述第三源域特征所属的第二领域类别；

基于所述第一领域类别和所述第二领域类别，确定所述故障诊断模型的域混淆损失。

5.根据权利要求1所述的故障诊断模型的训练方法，其特征在于，通过如下方式获取所述样本目标域信号：

基于所述第一设备对应的多个第一信号中的上极值点和下极值点，确定所述第一设备对应的上包络线和下包络线；

基于所述上包络线和所述下包络线，确定所述第一设备对应的均值包络线，并基于所述均值包络线和所述多个第一信号，确定所述第一设备的中间信号；

根据所述第一设备对应的多个第一信号和中间信号，确定所述第一设备对应的样本信号。

6.根据权利要求5所述的故障诊断模型的训练方法，其特征在于，通过如下方式获取所述第一信号：

获取所述第一设备对应的多个第二信号；

针对每类第二信号，计算该类第二信号的平均值和标准差，基于所述平均值和标准差，对每个该类的第二信号进行标准化处理，得到第一信号。

7.根据权利要求1所述的故障诊断模型的训练方法，其特征在于，在训练得到目标故障诊断模型之后，所述训练方法还包括：

将第一设备对应的目标域信号和第二设备对应的源域信号输入到所述目标故障诊断模型中，获取所述目标故障诊断模型输出的所述第一设备所属的当前故障类型；

基于所述第一设备所属的当前故障类型，生成并显示针对所述第一设备的故障处理提示信息。