Rotating machinery fault diagnosis for imbalanced data based on decision tree and fast clustering algorithm

Zhang, Xiaochen; Jiang, Dongxiang; Long, Quan; Han, Te

doi:10.21595/jve.2017.18373

Journal of Vibroengineering

Browse Journal

Submit article

Published: 30 September 2017

Check for updates

Rotating machinery fault diagnosis for imbalanced data based on decision tree and fast clustering algorithm

Xiaochen Zhang¹

Dongxiang Jiang²

Quan Long³

Te Han⁴

^{1, 2, 3, 4}State Key Lab of Power Systems, Department of Thermal Engineering, Tsinghua University, Beijing, 100084, China

Corresponding Author:

Xiaochen Zhang

Cite the article Download PDF

Downloads 1761

WoS Core Citations 12

CrossRef Citations 14

Abstract

To diagnose rotating machinery fault for imbalanced data, a kind of method based on fast clustering algorithm and decision tree is proposed. Combined with wavelet packet decomposition and isometric mapping (Isomap), sensitive features of different faults can be obtained so the imbalanced fault sample set is constituted. Then the fast clustering algorithm is applied to search core samples from the majority data of the imbalanced fault sample set. Consequently, the balanced fault sample set consisted of the clustered data and the minority data is built. After that, decision tree is trained with the balanced fault sample set to get the fault diagnosis model. Finally, gearbox fault data set and rolling bearing fault data set are used to test the fault diagnosis model. The experiment results show that proposed fault diagnosis model could accurately diagnose the rotating machinery fault for imbalanced data.

1. Introduction

With the rapid progress of modern science and technology, large rotating machinery equipment such as wind turbines, gas turbines and others become more and more complex. Considering the impact of unpredictable factors, failures of the rotating machinery equipment are different to avoid [1-3]. Since these failures would cause serious economic loss. To avoid huge economic loss, accurate and timely fault diagnosis is very significant for the rotating machinery. Recently, novel technologies and algorithms have been widely applied to rotating machinery fault diagnosis [4, 5]. For rotating machinery, types of faults are various while some failures don't happen very often. Herein the fault sample set would be imbalanced. Therefore, it is necessary to develop rotating machinery fault diagnosis technology for imbalanced fault sample set.

Decision tree is a kind of classification algorithm which has been widely applied in fault diagnosis. It adopts the top-down recursive regulation and attribute values are compared in the internal nodes of the decision tree. And conclusions can be got at the leaf nodes [6-8]. Compared with artificial neural network, the classification principle of decision tree is simple and easy to understand. At the same time, the classification model based on decision tree calculates fast. To increase the comprehensibility and usability of the decision tree in the process of establishing the decision tree, the pruning method can be used to prevent the decision tree from being too complicated. Even though pruning decision tree can avoid the over-fitting of the decision tree, the information gain of the decision tree will still be easily affected in the imbalanced training sample set [9-10]. The result of the information gain will still be biased towards those features with more values. Therefore, the pre-processing of the data set is necessary.

Since the cluster algorithm could classify the data according to their similarity, an approach based on fast clustering algorithm is adopted to search the core samples from the majority data of the imbalanced data set. This kind of fast clustering algorithm is proposed by Alex Rodriguez and Alessandro Laio in 2014. The main idea of this clustering algorithm [11-13] is automatically excluding the outliers from the original data set, so it is suitable for extracting the core samples and balance the imbalanced data set.

A kind of method based on fast clustering algorithm and decision tree is proposed to diagnose rotating machinery fault for imbalanced data. The vibration signal of the rotating machinery is decomposed into different frequency bands by wavelet packet decomposition, thus the original features can be got. After that, Isomap is applied to reduce the dimension of original features and obtain sensitive features. The fast clustering algorithm is used to construct the balanced data set. Finally, decision tree is trained to get the fault diagnosis model.

2. Decision tree and fast clustering algorithm

2.1. Fundamentals of decision tree

For decision tree, the top-down recursive method is adopted. The attribute values are compared in the internal nodes of the decision tree, and then the branches are developed from the internal nodes according to the different attributes. Finally, the conclusion can be concluded at the leaf nodes. Therefore, how to construct a decision tree with high precision and small scale is the core of the decision tree algorithm. Fig. 1 shows the schematic diagram of the decision tree. In Fig. 1, each non-leaf node represents an attribute of the training samples. The attribute value indicates the value corresponding to the attribute. And leaf nodes represent the sample category attributes.

Fig. 1Schematic diagram of the decision tree

In the process of training the decision tree, it is significant to choose the basis for testing attributes. The information gain is usually used as the basis for the generation of nodes. By selecting the attribute with the highest information gain as the testing attribute of the current node, the mixing level of training samples at this node will be reduced to the lowest. In order to precisely define the information gain, we first define the entropy, a metric that is widely used in information theory.

If the sample set is $S$ , and the value of the attribute $A$ is $c$ , then the entropy of the sample set S is defined as [14, 15]:

1

E n t r o p y (S) = - \sum_{i = 1}^{c} p_{i} l o g_{2} p_{i},

where $p_{i}$ is the proportion of the $i$ th attribute value sample set.

Then the information gain can be formed as:

2

G a i n (S, A) = E n t r o p y (S) - \sum_{v \in V (A)} \frac{|S_{v}|}{|S|} E n t r o p y (S_{v}),

where $V (A)$ is the range of the attribute $A$ , $S_{v}$ represents the sample set that the value of the attribute $A$ is $v$ .

In different candidate attributes, the attribute with the largest information gain is selected as the classification basis of the current decision node. Then the new decision nodes are constantly created, and finally a decision tree to classify the training sample set can be established.

2.2. Pruning decision tree

To avoid the over-fitting and reduce the complexity of the decision tree, it is necessary to prune the decision tree. The pruning decision tree is to delete the most unreliable branches by statistical methods, so the probability of over-fitting can be weakened. Generally, the main pruning methods include pre-pruning and post-pruning [16, 17]. The pre-pruning is usually based on the statistical significance to determine whether the current node needs to divide continuously. It is difficulty to choose a suitable threshold which directly determine the classification degree of the decision tree. Compared with pre-pruning, post-pruning allows the decision tree to grow sufficiently. Then the extra branches would be pruned, so the post-pruning can obtain a more accurate decision tree. The pruning decision tree would consider the coding length of the decision tree. The minimum description length (MDL) principle is adopt to optimize the decision tree. The basic idea is to construct the decision tree with the shortest coding length.

Although pruning the decision tree can avoid the over-fitting of the decision tree, the information gain of the decision tree will still be easily affected in the imbalanced training sample set. The result of the information gain will still be biased towards those features with more values. Therefore, in order to improve the classification performance of decision tree for imbalanced sample set, a fast clustering algorithm is applied to balance the sample set.

2.3. Fast clustering algorithm

A kind of fast clustering algorithm is adopted to balance the original data set. The basic idea of this clustering algorithm is automatically excluding the outliers, so it is suitable for extracting the core samples from the imbalanced data set. The assumptions of the algorithm are that cluster centers are surrounded by neighbors with lower local density. In addition, these cluster centers are at a relatively large distance from any points with a higher local density. The local density can be calculated by Cut-off kernel. With Cut-off kernel, the local density $ρ_{i}$ of data point $i$ can be formulated as [11, 12]:

3

ρ_{i} = \sum_{j} χ (d_{i j} - d_{c}),

4

χ (x) = \{\begin{array}{l} 0, x \geq 0, \\ 1, x < 0, \end{array}

where $d_{i j}$ is the distance between data point $i$ and data point $j$ and $d_{c}$ is the cut-off distance.

In Eq. (3), the local density $ρ_{i}$ represents the number of the data points which are closer to data point $i$ compared with $d_{c}$ . Then the distance $δ_{i}$ can be expressed as:

5

δ_{i} = \underset{j : ρ_{j} > ρ_{i}}{m i n} (d_{i j}) .

It is clear that the distance $δ_{i}$ means the minimum distance between the point $i$ and the point with higher density, except that the point $i$ possesses the highest density. For the point with highest density, we conventionally take:

6

δ_{i} = m a x_{j} (d_{i j}) .

The local density $ρ_{i}$ and distance $δ_{i}$ for each data point can be calculated. Therefore, the weight of clustering center $γ_{i}$ is constructed as:

7

γ_{i} = ρ_{i} δ_{i} .

Obviously, the clustering centers are the points with larger weights. So, the sequence $n_{i}$ can be constructed as:

8

n_{q_{i}} = a r g m i n d_{q_{i} q_{j}} (q_{j}), i \geq 2, i > j,

where sequence $q_{i}$ is the index number of local density $ρ_{i}$ sorted in descending order. The sequence $n_{i}$ represents the index number of the point closest to point $i$ , while the local density of this point is larger than point $i$ .

After that, the non-clustering center points can be categorized as:

9

c_{q_{i}} = c_{n_{q_{i}}},

where $c$ are the labels of the clustering centers.

In the imbalanced data set, the mean local density of each cluster can be calculated. Then the cluster can be divided into core points and halo points by comparing with the mean local density.

Fig. 2“Synthetic point distributions” data set

a) Raw data

b) Clustered by fast clustering algorithm

c) Clustered by K-means clustering algorithm

To test the effectiveness of the fast clustering algorithm, the “Synthetic point distributions” data set is applied [11]. The distribution of the “Synthetic point distributions” data set are shown in Fig. 2. From Fig. 2(b), we can see that the core points of five class data are all correctly selected from the raw “Synthetic point distributions” data set. Fig. 2(c) shows the clustering result of the K-means clustering algorithm, it is clear that the raw data are approximately divided into five categories while the noise points are not eliminated. This illustrates that the fast clustering algorithm can be well applied to extract the core samples from the imbalanced data set.

3. Rotating machinery fault diagnosis for imbalanced data

3.1. Feature extraction and dimension reduction

The vibration signal of the rotating machinery usually consists of a series of complex components. To decompose the vibration signal into different frequency bands. The wavelet packet decomposition is applied. The output of the $j$ th layer can be defined as $d_{l}^{j, n}$ during the process of decomposing the vibration signal, where $l$ represents the time serial number. The decomposition formulas are as follows [18-20]:

10

d_{l}^{j + 1,2 n} = \sum_{k = - \infty}^{\infty} h_{0} (2 l - k) d_{k}^{j, n},

11

d_{l}^{j + 1,2 n + 1} = \sum_{k = - \infty}^{\infty} h_{1} (2 l - k) d_{k}^{j, n},

where $d_{l}^{j + 1,2 n}$ and $d_{l}^{j + 1,2 n + 1}$ are the outputs of the ( $j + 1$ )th layer.

The wavelet “coiflet” is adopted. The condition corresponds to a “coiflet” of order $L$ is defined as:

12

\int t^{l} ϕ (t) d t = \{\begin{array}{l} 1, (l = 0), \\ 0, (l = 1,2, \dots, L - 1), \end{array}

13

\int t^{l} ψ (t) d t = 0, (l = 0,1, \dots, L - 1),

where Eq. (12) is the condition that the vanishing moment of scaling function equals to zero. And Eq. (13) is the condition that the vanishing moment of wavelet equals to zero.

Meanwhile:

14

ϕ (t) = \sum_{n} c_{n} ϕ (2 t - n),

where $c_{n}$ are the coefficients.

After decomposing the vibration signal of the rotating machine into different frequency bands, the energy of each sub-frequency band is calculated. Then the original features can be constructed. To reduce the dimension of the original features, Isomap is applied to obtain the sensitive features. The main calculating stages of the Isomap are as follows [21-22]:

(1) Neighborhood graph $G$ is constructed. For example, the original data set are $V (x_{i} \in V)$ . If $x_{j}$ is one of the nearest neighbors of $x_{i}$ , $G$ might contain the edge $x_{i} x_{j}$ when $|x_{i} - x_{j}| < ε$ ;

(2) Euclidean length is calculated. The shortest paths should be calculated for all pairs of data points;

(3) Embedded Data. With the method of multidimensional scaling (MDS), the new embedment of the data in Euclidean space can be searched.

3.2. Balanced data set and fault diagnosis model

The rotating machinery fault sample set usually contains various fault types. Since some fault types are frequent faults while others are accidental, ordinarily the rotating machinery fault sample set is an imbalanced data set. For each fault sample, a number of sensitive features can be obtained by Isomap. Then the distance between two fault samples can be formulated as:

15

d_{i j} = \sqrt{\sum_{k = 1}^{K} w_{k} (t_{i k} - t_{j k})^{2}},

where $t_{i k}$ and $t_{j k}$ are the sensitive features of the $i$ th fault sample and the $j$ th fault sample. $w_{k}$ is the weight of the $k$ th sensitive feature. $K$ represents the number of sensitive features for each fault sample.

Fig. 3Flowchart of building fault diagnosis model

After calculating the distance $d_{i j}$ , the local density $ρ_{i}$ and the distance $δ_{i}$ can be got based on Eqs. (3-6). Then referencing the Eq. (7), the weight $γ_{i}$ of each fault sample can be obtained. Considering the number of samples of the minority fault type, the same number of samples with higher weight $γ_{i}$ are selected from the majority type. Thus, the balanced fault sample set is constructed by whole samples from minority fault type and selected samples from the majority type. Eventually, the decision tree classification algorithm is applied to study the balanced fault sample set, so the fault diagnosis model can be obtained. Fig. 3 shows the flowchart of building fault diagnosis model.

It is clear that the balanced sample set are used to train the decision tree, while the testing samples are responsible for testing the classification accuracy of the trained decision tree. The trained decision tree would not be adopted until the classification accuracy is acceptable. Thus, the trained decision tree can be recognized as the fault diagnosis model.

4. The experiment results

Gearbox and rolling bearing are two kinds of common rotating machinery components. Therefore, fault data sets of gearbox and rolling bearing are both adopted to test the validity of the proposed fault diagnosis model.

4.1. Gearbox fault diagnosis

The gearbox failure simulation test bed is shown in Fig. 4. The main components include gearbox, motor, motor driver, wind wheel and accelerometer. The motor is responsible for driving the gearbox at different rotating speeds. The function of the wind wheel is as a load. An accelerometer is installed on the top of the gearbox and used for acquiring the vibration signal. The type of the tested gearbox is single-stage planetary transmission. Meanwhile the number of the planetary gear teeth is 20. In the testing, two kinds of planetary gear failures are adopted: half fracture and full fracture. To simulate the real working condition, the rotating speed is also considered in the testing. As is shown in Table 1, the rotating speed includes 157 r/min, 237 r/min and 317 r/min. The sampling frequency of the data acquisition is 25 kHz while the sampling time for each sample is 0.4 seconds. Thus, there are 10000 data points in each sample.

Table 1Testing for gearbox

	Planetary gear	Rotating speed (r/min)
Testing 1	Normal	157/237/317
Testing 2	Half fracture	157/237/317
Testing 3	Full fracture	157/237/317

Fig. 4Gearbox failure simulation test bed

With the method of wavelet packet decomposition, the gearbox vibration signal can be decomposed into different frequency bands. Then the energy of each frequency band is calculated, so the original features can be obtained. To get the sensitive features, Isomap dimension reduction algorithm is adopted. Thus, sensitive feature 1, sensitive feature 2 are selected. Fig. 5 show the distributions of sensitive features. In order to test the effectiveness of Isomap, calculating result by principal component analysis (PCA) is also shown in Fig. 5(a). From Fig. 5(a), it can be seen that the distribution areas of normal gear, half fracture and full fracture are mixed. In Fig. 5(b), the aliasing region only exists between normal gear and full fracture. Since the aliasing region would lead to the misjudgment between different failures. Therefore, testing data from the normal gear and full fracture are used to construct an imbalanced data set to test the classification effect of the proposed fault diagnosis model.

Fig. 6 show the distributions of the imbalanced data set under different proportions (normal: full fracture). Since the normal gear data is much easier to obtain than the full fracture data, the normal gear is defined as the majority class while the full fracture is the minority class. The number of the normal gear is 150 and the number of the full fracture data varies from 30 to 80.

Fig. 5Distributions of sensitive features

a) PCA

b) Isomap

The imbalanced data sets shown in Fig. 6 are adopted as the training samples. The testing samples are composed of 300 samples. 150 samples are from normal gear data while another 150 samples are from full failure data. Meanwhile, decision tree models are also trained for comparing with the proposed fault diagnosis model. Table 2 and Fig. 7 show the classification accuracy comparisons between proposed fault diagnosis model and other methods. We can see that pruning has no effect on the improvement of decision tree classification accuracy for the imbalanced data set. To show the advantage of the fast clustering algorithm for extracting the core samples, K-means clustering algorithm is also adopted for comparing. Since K-means clustering algorithm is a similarity measure method based on Euclidean distance, so the algorithm is suitable for clusters of spherical shapes. Thus, the core samples extracted by K-means clustering algorithm are mostly gathered together. Therefore, the decision tree trained by these gathered core samples is not suitable to the entire imbalance data set. It is clear that the classification accuracy of the K-means clustering algorithm is less than other methods in Table 2.

Table 2Classification accuracy comparisons between fault diagnosis model and other methods

Proportion of the data set	150:30	150:40	150:50	150:60	150:70	150:80
Fault diagnosis model	92.33 %	92.00 %	92.00 %	93.00 %	92.67 %	92.33 %
Decision tree before pruning	87.67 %	88.33 %	90.00 %	90.00 %	90.00 %	90.67 %
Decision tree after pruning	86.00 %	88.33 %	89.33 %	89.00 %	89.67 %	89.67 %
K-means and decision tree	81.33 %	81.33 %	84.00 %	84.33 %	79.00 %	77.67 %

Table 3Training time comparisons between fault diagnosis model and other methods

Proportion of the data set	150:30	150:40	150:50	150:60	150:70	150:80
Fault diagnosis model	0.9048 s	0.9375 s	0.9450 s	0.9559 s	0.9588 s	0.9806 s
Decision tree before pruning	0.1886 s	0.1904 s	0.1914 s	0.1924 s	0.1927 s	0.1935 s
Decision tree after pruning	0.5883 s	0.6088 s	0.6227 s	0.6282 s	0.6364 s	0.6451 s
K-means and decision tree	0.6326 s	0.6655 s	0.6856 s	0.6910 s	0.6949 s	0.7149 s

The classification accuracies of the fault diagnosis model in different data sets are all more than 92 %. It is obvious that the fault diagnosis model has achieved better classification results than other methods.

Table 3 show the training time comparisons between fault diagnosis model and other methods. In this paper, the hardware environment is as follows: the processor is Intel Core i7-4600U. The memory is 8 GB and the operating system is Windows8. Meanwhile, the software environment is MATLAB R2013b. As can be seen from Table 3, the training time of four methods is less than 1 second. It would take time for clustering the data, so the training time of “fault diagnosis model” and “K-means and decision tree” is longer than other two methods. To sum up, the training time of the fault diagnosis model is acceptable since the fault diagnosis model has achieved better classification results than other methods.

Fig. 6Distributions of the imbalanced data set under different proportions (normal: full fracture)