arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.14398cs.LGeess.SP

利用自监督学习与联邦学习最大化机器状态监测的模型泛化能力

Maximizing Model Generalization for Machine Condition Monitoring with Self-Supervised Learning and Federated Learning

University of Kentucky(肯塔基大学)

Matthew Russell, Peng Wang

AI总结:

本文研究机器状态监测中数据无标签、类别少且存在领域偏移的问题,提出以 Barlow Twins 自监督学习最大化源域特征通用性,并结合联邦学习跨机器共享知识,实验表明该方法在无标签目标域和新故障条件下优于监督学习。

摘要:

深度学习(DL)能够从原始状态监测数据中诊断故障并评估机器健康状况,而无需人工设计的统计特征。然而,对于现有的深度学习方法而言,实际制造应用仍然极其困难。机器数据通常没有标签,并且来自非常少的健康状态(例如,仅有正常运行数据)。此外,随着过程参数变化以及新的故障类别出现,模型常常会遇到领域偏移。传统的监督学习可能难以学习到紧凑且具有判别性的表示,以泛化到这些未见的目标领域,因为它依赖于拥有足够多的类别来用决策边界划分特征空间。采用领域自适应的迁移学习(TL)试图将这些模型适配到无标签的目标领域,但它假设存在相似的底层结构,而当新故障出现时这种结构可能并不存在。本研究提出将重点放在最大化源领域上的特征通用性,并通过权重迁移将模型复制到目标领域来应用迁移学习。具体而言,采用 Barlow Twins 的自监督学习(SSL)通过关注数据的语义属性,可能比监督学习产生更具判别性的健康状态监测特征。此外,用于分布式训练的联邦学习(FL)通过在多台客户端机器之间共享信息,有效扩大训练数据的有效规模和多样性,也可能改善泛化能力。结果表明,当源训练数据包含非常少的明显类别时,Barlow Twins 在出现新电机故障的无标签目标领域中优于监督学习。引入联邦学习还可以通过在机器之间扩散健康状态知识而提供轻微优势。

英文摘要:

Deep Learning (DL) can diagnose faults and assess machine health from raw condition monitoring data without manually designed statistical features. However, practical manufacturing applications remain extremely difficult for existing DL methods. Machine data is often unlabeled and from very few health conditions (e.g., only normal operating data). Furthermore, models often encounter shifts in domain as process parameters change and new categories of faults emerge. Traditional supervised learning may struggle to learn compact, discriminative representations that generalize to these unseen target domains since it depends on having plentiful classes to partition the feature space with decision boundaries. Transfer Learning (TL) with domain adaptation attempts to adapt these models to unlabeled target domains but assumes similar underlying structure that may not be present if new faults emerge. This study proposes focusing on maximizing the feature generality on the source domain and applying TL via weight transfer to copy the model to the target domain. Specifically, Self-Supervised Learning (SSL) with Barlow Twins may produce more discriminative features for monitoring health condition than supervised learning by focusing on semantic properties of the data. Furthermore, Federated Learning (FL) for distributed training may also improve generalization by efficiently expanding the effective size and diversity of training data by sharing information across multiple client machines. Results show that Barlow Twins outperforms supervised learning in an unlabeled target domain with emerging motor faults when the source training data contains very few distinct categories. Incorporating FL may also provide a slight advantage by diffusing knowledge of health conditions between machines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑