A Multiattribute Learning Model for Zero-Sample Mechanical Fault Diagnosis
Li Cai, Hongpeng Yin, Jingdong Lin, Dandan Zhao, Yan Qin
Abstract
The scarcity of fault samples is a common scenario in the field of fault diagnosis. In the context of mechanical fault diagnosis, the emergence of new working conditions and fault modes renders the availability of samples of target (unseen) faults for model training unfeasible, thus limiting the performance of data-driven methods. Consequently, zero-sample learning and diagnosis of mechanical faults is a challenging task. In this regard, this article proposes a multiattribute learning model, inspired by the zero-shot learning paradigm, for zero-sample mechanical fault diagnosis. The key lies in the shared multiclass attribute classifiers. During the attribute learning process, a convolutional neural network is developed to construct multiclass attribute classifiers, which serve as a mapping between visual features and semantic features. These classifiers are transferred from readily available faults to enhance the capability of diagnosing unseen faults. By minimizing the difference among the fault attributes, the diagnosis of unseen faults is achieved, which includes fault location, size, working load, etc. Experiments on two real datasets verify the efficacy and the superiority of the proposed method.