Few-shot Learning under Domain Shift: Attentional Contrastive Calibrated Transformer of Time Series for Fault Diagnosis under Sharp Speed Variation

Shen Liu,Jinglong Chen,Shuilong He,Zhen Shi,Zitong Zhou
DOI: https://doi.org/10.1016/j.ymssp.2022.110071
IF: 8.4
2023-01-01
Mechanical Systems and Signal Processing
Abstract:The domain shift of sample distribution caused by sharp speed variation dissatisfies the general assumption of stationary conditions, which renders a severe challenge for a majority of existing intelligent fault diagnosis methods. Moreover, data deficiencies in industrial applications further compromise diagnostic accuracy and reliability. To break the predicament of fault diagnosis under sharp speed variation with few samples, we developed an Attentional Contrastive Cali-brated Transformer (ACCT) of time series. First, a plurality of convolution layers is used to capture low-level local structure features. Then, the transformer is applied to the sequences of split patches for modeling global dependencies and extracting the domain invariant features. Meanwhile, a data augmentation strategy based on regional mixing is used to enhance the generalization. Furthermore, to obtain a more discriminative feature representation, we designed a regularization based on unsupervised contrastive learning for calibration of attention distri-bution. The results demonstrated that transformers have an aptitude for analyzing time series data even under sharp variation, which do not need to deliberately consider extra modules for cross-domain disentanglement. The proposed method is superior to several advanced trans-formers in three case studies under speed transient conditions with few samples.
What problem does this paper attempt to address?