BearingFM: Towards a foundation model for bearing fault diagnosis by domain knowledge and contrastive learning

Zou Lai, Chen Yang, Shulin Lan, Lihui Wang, Weiming Shen and Liehuang Zhu

International Journal of Production Economics, 2024, vol. 275, issue C

Abstract: Monitoring bearing failures in production equipment can effectively prevent finished product quality issues and unplanned factory downtime, thereby reducing supply chain uncertainties and risk. Therefore, monitoring bearing failures in production equipment is important for improving supply chain sustainability. Due to the generalization limitations of neural network models, specific models must be trained for specific tasks. However, in real industrial scenarios, there is a severe lack of labeled samples, making it difficult to deploy fault diagnosis models across massive amounts of equipment in workshops. In order to solve the above issue, this paper proposes a cloud-edge-end collaborative semi-supervised learning framework, which provides multi-level computing power and data support for building a foundation model. A data augmentation method based on the bearing fault mechanism is proposed, which effectively preserves the inherent essential characteristics in vibration signals by normalizing frequency and adding noise in specific frequency bands. A novel contrastive learning model is designed, which narrows the distances between positive samples and widens the distances between negative samples in the high-dimensional space through cross comparisons in the time dimension and knowledge dimension, thereby extracting the most essential characteristics from the unlabeled signals. Multiple sets of experiments conducted on four datasets demonstrate that the proposed approach achieves an approximately 98% fault classification accuracy with only 1.2% labeled samples.

Keywords: Fault diagnosis; Foundation model; Semi-supervised learning; Cloud-edge-end collaboration; Contrastive learning (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0925527324001762
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:proeco:v:275:y:2024:i:c:s0925527324001762

DOI: 10.1016/j.ijpe.2024.109319

Access Statistics for this article

International Journal of Production Economics is currently edited by Stefan Minner

More articles in International Journal of Production Economics from Elsevier
Bibliographic data for series maintained by Catherine Liu ().