[发明专利]故障检测方法、装置及存储介质

说明书

本公开是关于一种故障检测方法、装置及存储介质。该方法包括：获取待检测样本；从服务器处获取全局模型参数，并基于所述全局模型参数更新所述终端中的局部模型；基于所述更新的局部模型，对所述待检测样本进行故障检测；其中，所述全局模型参数由全局模型确定，所述全局模型基于终端发送的局部近似样本分布和样本数训练得到。采用本公开的方法，可以在解决各终端的样本非独立同分布的技术问题和局部中断数据异常的技术问题的同时，不暴露各终端的样本分布信息，得到符合全局数据特征的预测模型，从而在确保预测精度的基础上保护了用户隐私。

技术领域

本公开涉及计算机技术领域，尤其涉及故障检测方法、装置及存储介质。

背景技术

相关技术中，可以采用预先训练好的检测模型来对产品的故障进行检测分类，检测模型可以根据产品的历史故障数据训练得到。工业场景下，不同企业/工厂之间出于数据隐私及安全性的考虑，导致各自终端中的数据不能自由流通，形成数据孤岛。同时，各企业/工厂终端中的数据又存在互相补充的特质，可以为故障检测分类等应用提供更多的样本数据，提高检测模型的检测精度。因此，可以采用横向联邦学习模型，通过“数据不动模型动”的方式来解决这个问题。

横向联邦学习普通直接均值聚合的方法，需要各终端中的数据为服从独立同分布的。但是不同企业/工厂设备的终端中的数据受到工况、生产条件等影响，往往是非独立同分布的，常规方法没有很好解决这个问题。

发明内容

为克服相关技术中存在的问题，本公开提供一种故障检测方法、装置及存储介质。

根据本公开实施例的第一方面，提供一种故障检测方法，包括：

获取待检测样本；

从服务器处获取全局模型参数，并基于所述全局模型参数更新所述终端中的局部模型；

基于所述更新的局部模型，对所述待检测样本进行故障检测；

其中，所述全局模型参数由所述全局模型确定，所述全局模型基于终端发送的局部近似样本分布和样本数训练得到。

在一种实施方式中，所述终端发送的局部近似样本分布采用以下方式获得：

从服务器处获取初始化全局模型参数，根据所述初始化全局模型参数确定终端的待检测样本中各样本的标签，所述标签包括一个或多个标签类型；

基于各样本的标签确定所述终端的局部样本分布，所述局部样本分布表征与每一类型标签对应的样本数占总样本数的比例；

对每一类型的标签进行近似处理，将各标签对应的局部样本分布近似处理为局部近似样本分布。

在一种实施方式中，所述全局模型采用如下方式训练得到：

响应于被服务器选中，对终端中的局部模型进行训练，得到局部更新参数；

发送所述局部更新参数至服务器，以使服务器对接收到的局部更新参数进行聚合，得到全局模型；

其中，所述终端被服务器选中的概率由所述局部近似样本分布和样本数确定。

根据本公开实施例的第二方面，提供一种故障检测方法，包括：

接收终端发送的局部近似样本分布和样本数，基于所述局部近似样本分布和样本数训练得到全局模型；

发送全局模型参数至终端，以使终端根据接收到的全局模型参数更新终端中的局部模型，并根据更新后的局部模型对待检测样本进行故障检测。

在一种实施方式中，所述终端发送的局部近似样本分布由终端对终端中的局部样本分布近似处理获得。

在一种实施方式中，所述基于所述局部近似样本分布和样本数训练得到全局模型，包括：