Local coordinate weight reconstruction for rolling bearing fault diagnosis

Chenxi Wu1 , Rong Jiang2 , Zhonghua Huang3 , Xin Wu4 , Zhe Liu5

1, 2, 3, 4, 5Hunan Provincial Key Laboratory of Vehicle Power and Transmission System, Hunan Institute of Engineering, Xiangtan, China

1, 2, 3, 4, 5Hunan Provincial Engineering Laboratory of Wind Power Operation, Maintenance and Testing, Hunan Institute of Engineering, Xiangtan, China

2, 3Corresponding authors

Journal of Vibroengineering, Vol. 22, Issue 7, 2020, p. 1583-1592. https://doi.org/10.21595/jve.2020.21460
Received 7 May 2020; received in revised form 9 July 2020; accepted 22 July 2020; published 5 November 2020

Copyright © 2020 Chenxi Wu, et al. This is an open access article distributed under the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
Creative Commons License
Table of Contents Download PDF Acknowledgements References
Cite this article
Views 48
Reads 18
Downloads 680
WoS Core Citations 0
CrossRef Citations 0

The high dimensionality data originating from rolling bearing measuring signals with non-linearity and low signal to noise ratio often contains too much disturbance like interference and redundancy for accurate condition identification. A novel manifold learning named Local coordinate weight reconstruction (LCWR) is proposed to remove such disturbance. Due to the different contribution of samples to their manifold structure, weight value is used for the contribution difference. By reconstructing local low-dimensional coordinates according to its weight function about geodesic distance in neighborhood, LCWR targets to reduce reconstruction error, preserve intrinsic structure of the high dimensionality data, eradicate disturbance and extract sensitive features as global low-dimensional coordinates. The experimental results show that the intraclass aggregation and interclass differences of global low-dimensional coordinates extracted via LCWR are better than those of local tangent space alignment (LTSA), locally linear embedding (LLE) and principal component analysis (PCA). The accuracy reaches the highest 96.43 % using SVM to identify LCWR based global low-dimensional coordinates, and its effectiveness is testified in diagnosis of rolling bearing fault.

  • A manifold learning named Local coordinate weight reconstruction (LCWR) is proposed for nonlinear dimensionality reduction.
  • LCWR targets to eradicate disturbance and extract sensitive features from the high dimensionality data.
  • The intraclass aggregation and interclass differences of global low-dimensional coordinates via LCWR are better than other methods.
  • The accuracy reaches the highest 96.43% and testifies the effectiveness of LCWR in diagnosis of rolling bearing fault.

Keywords: manifold learning, nonlinear dimensionality reduction, feature extraction, rolling bearing, fault diagnosis.

1. Introduction

Rolling bearing plays a key role in rotating machinery. It is necessary to monitor rolling bearing condition and identify its fault to avoid accident [1-3]. The state information of rolling bearing is usually described with high dimensionality data consisting of multiple characteristics in time and frequency domain [4-6], which contains redundancy and interference, and exists nonlinearity. Therefore, several works have been explored to remove such disturbance and obtain low-dimensional sensitivity features for better accuracy and efficiency of rolling bearing fault [7, 8].

As a type of classic manifold learning for dimensionality reduction [9, 10], local tangent space alignment (LTSA) completing nonlinear dimensionality reduction through finding out neighborhoods of high dimensionality samples, carrying out local dimensionality reduction, and realigning all neighborhoods’ low-dimensional coordinates to construct global low-dimensional coordinates [11], recently has been used for fault diagnosis besides its earlier successful applications to image processing, data mining, machine learning, etc. [12, 13]. For instance, Zhang et al. proposed supervised locally tangent space alignment (S-LTSA) to optimize the neighborhood selection of LTSA based on the training samples’ categories, so that the neighborhood includes the same samples as possible to accurately reflect the local structures of different types of bearing fault signals [14]. Li et al. improved the accuracy of bearing fault diagnosis using LLTSA dimensionality reduction [15]. Kumar A and Kumar R utilized Linear Local Tangent Space Alignment (LLTSA) to suppress noise and retain characteristic defect frequencies of rolling bearing with inner and ball fault [16]. Su et al. proposed orthogonal supervised linear local tangent space alignment (OSLLTSA) to make the neighborhood selection of LLTSA better by introducing sample’s label information, which removed interference and redundancy in high dimensionality fault data and extracted low-dimensional sensitivity fault features [17]. Wang et al. proposed supervised incremental local tangent space alignment (SILTSA) through embedding supervised learning into the incremental local tangent space alignment to extract bearing fault characteristics, process new samples and classify [18]. Su et al. stated supervised extended local tangent space alignment (SE-LTSA) to enhance intraclass aggregation and interclass differences of nonlinear high dimensionality samples by defining distance between samples and optimizing neighborhood choice based on class label [19]. In summary, the current interests focused on neighborhood optimization options and local tangent space estimation of LTSA for dimensionality reduction and low-dimensional sensitivity fault characteristics extraction.

Different from above methods, local coordinate weight reconstruction (LCWR) manifold learning is proposed to reconstruct local coordinates by weight coefficient, so as to extract global low-dimensional sensitivity features for improving fault diagnosis capability of rolling bearing. The following is as below: Section 2 proposes LCWR for coordinate reconstruction, Section 3 validates LCWR and Section 4 is conclusions.


LCWR has two major tasks. First, the projection coordinates of k nearest neighbors of each sample on the tangent space of the neighborhood are calculated to build the local low-dimensional manifold using LTSA. Next, the global low-dimensional coordinates are obtained by aligning the low-dimensional coordinates of all neighborhoods according to weight coefficient, which is a main innovation of LCWR.

2.1. Local coordinate computation

Let sample matrix X=x1,x2,,xNRD×N, D is the number of sample dimensions and N is the number of samples. Sample xi and its k nearest samples (including xi) constitute local neighborhood matrix Xi=(xi1 ,xi2 ,,xik )RD×k , each neighborhood exists a local tangent space QiRD×d(d<D) consisting of standard orthogonal basis vectors, and Θi=θi1,θi2,,θikRd×k is the projection of Xi onto Qi. For the main geometric structure information within neighborhood, minimize the sum of square of distance between Xi and Θi:

Min  Xi-Xi¯-QiΘiF2,S.t.  QiTQi=Id,

where Xi¯=1/kXieeT is a mean matrix of Xi, e is a unit column matrix of length k, and Id is a d by d unit matrix.

Apply singular value decomposition to Xi-Xi¯:


QiΘi, Qi, Θi are computed as below:

QiΘi=UdΣdVdT,Qi=Ud,Θi=Q iTXi-Xi¯,

where URD×D is a left singular vector set, VRk×k is a right singular vector set, RD×k is a singular value diagonal matrix, dRd×d is a diagonal matrix with d maximum singular values in descending order, UdRD×d is the corresponding left singular vector set and VdRk×d is right singular vector set.

Thus, Θi holds the most important geometric structure information in Xi.

2.2. Global coordinate construction

Suppose Ti=τi1,τi2,,τikRd×k is the local low-dimensional coordinate of Xi. Build the affine transformation between Ti and Θi:


The local reconstruction error Ei is written as:


where LiRd×d is a local affine matrix, Ti¯=1/kTieeT is a mean matrix of Ti, EiRd×k is a local reconstruction error matrix.

However, owing to the different contribution of samples to their manifold structure, a novel LCWR based on weight coefficient is proposed to reduce permutation error and reconstruct local coordinates more accurately. According to LCWR, the closer a sample is to its manifold, the larger its weight coefficient is. Likewise, the farther a sample is from its manifold, the smaller its weight coefficient is. So an exponential function of geodesic distance between a sample and the center point of its neighborhood, reflecting the proximity to its manifold structure, is defined as weight coefficient, namely:


where ωij denotes the weight coefficients of the jth nearest neighbor xij in Xi, Gij and σi denote the geodesic distance from xij to the center of Xi and the mean square error of Xi respectively, and β is adjustment parameter.

Then Ei is rewritten as:


where Wi=diag(ωi1,ωi2,,ωik)Rk×kis a weight coefficient matrix, Hi=(I-1/keeT)WiRk×k.

Fix Ti and minimize Ei to preserve as much local information as possible, namely:

Li =TiHiΦiT×invΦiΦiT,

where LiRd×d, Φi=ΘiWiRd×k.

Substitute Eq. (9) into Eq. (7) and obtain Ei:


where Bi=Hi-HiΦiT×invΦiΦiTΦiRk×k.

Minimize the sum of all neighborhoods reconstruction errors to obtain the global low-dimensional coordinate T:


where T=τ1,τ2,,τNRd×N is the global low-dimensional coordinate of X, Ti=TRi, Ri=R:,IiRN×kis a selection matrix, R=diagone1,N, Ii=i1,i1,,ik is the subscript of k nearest neighbors of xi.

It is equal to solve differential equation:

i=1NT(RiBiBiTRiT)T+TR i BiBiTRiT-2λT=0,

where M=MT=i=1NRiBiBiTRiTRN×N.

Therefore, the optimal solution of TT is composed of eigenvectors corresponding to the 2nd to the (d+1)th smallest eigenvalues of M. TT=u2,u3,,ud+1 or T=u2,u3,,ud+1TRd×N is an orthogonal global low-dimensional coordinate mapping matrix of nonlinear manifold in X, where ujR1×N is an eigenvector corresponding to the jth eigenvalue of M.

LCWR is summarized as follows:

(1) Look for neighborhood. Local neighborhood Xi of each sample xi (i=1,2,,N) is found by k nearest neighbors.

(2) Extract local coordinates. Projection matrix Qi and local low-dimensional coordinate matrix Θi of each neighborhood Xi are obtained according to Eq. (2) and Eq. (3), respectively.

(3) Construct global coordinates. Global low-dimensional coordinate matrix T is conducted from Θi reconstruction by weight coefficient matrix Wi and expressed as Eq. (14).

The flow chart of LCWR is shown in Fig. 1.

3. Verification and analysis

Experimental data is from the bearing data center of Case Western Reserve University. Four types of bearing sates under speed of 1750 rpm and load of 1470 W, i.e. healthy state, outer race fault, inner race fault and ball fault with defective size of 0.3556 mm are considered. There are 48 samples for each state and total 192 samples for all states. Each sample acquired at 48 kHz includes 2048 points in length.

3.1. High-dimensional feature construction

As shown in Table 1, 12 time-domain statistical indicators and 8 frequency-domain statistical indicators are selected to constitute a 20-dimensional sample to characterize the bearing state. An original sample signal is decomposed into eight sub-band signals by three-layer db8 wavelet packet decomposition, and the ratio of the energy of each sub-band to the total energy of all sub-bands is taken as the frequency domain indicator. That is, e j =Ej/E, E=j=18Ej, Ej is the energy of sub-band signals. Thus, a high-dimensional feature matrix XR192×20 is created.

Fig. 1. Procedure of LCWR

Procedure of LCWR

Fig. 2. LCWR based fault diagnosis

LCWR based fault diagnosis

Table 1. Basic size and style requirements

Standard deviation
Root mean square
Crest factor
Energy ratio ej

3.2. Low-dimensional feature extraction

According to Fig. 2, some of the low-dimensional feature samples extracted by LCWR are used as training samples to train support vector machine (SVM) while the others as test samples to be recognized by trained SVM. When using LCWR to extract bearing state characteristics, three parameters such as neighbor number k, dimension d and adjustment parameter β need to be optimized. Because the recognition rate can be regarded as a function of three parameters k, d and β, these parameters interact to determine the recognition rate. By changing these parameters in a certain range and the corresponding recognition rate obtained, it is proved that there exist optimization values of parameters with the peak recognition rate. The trend of recognition rate with respect to a single parameter variable while the other two parameters fixed is shown in Fig. 3-5, respectively. Because of different parameters, the trend of recognition rate is also different from each other.

From the recognition rate about the nearest neighbor number k in Fig. 3, there is the maximum rate 96.43 % at k= 8. The role of k on recognition rate is carried out by influencing the intrinsic geometry structure of high-dimensional samples, the close relationship between similar samples and the nonlinear dimensionality reduction ability of LCWR. If k is too small, LCWR can’t maintain the intrinsic geometry of high-dimensional samples and close association between similar samples. If k is too large, it weakens the nonlinear dimensionality reduction capability of LCWR. Hence, the low-dimensional manifold structure hiding in the high-dimensional samples can be found to the greatest extent and achieve the maximum rate at the optimum value k= 8. However, due to the comprehensive effect of different factors affected by k, the recognition rate fluctuates and there are multiple turnover points rather than a monotonous trend.

Dimension d affects the recognition rate by mining the sensitive features of high-dimensional samples in the neighborhood and eliminating redundant and interference components. From Fig. 4, it can be seen that the maximum recognition rate is 96.43 % at d= 3, because an appropriate d makes similar samples have approximate low-dimensional features, leading to better clustering effect and improvement in recognition rate. Otherwise, LCWR can’t fully mine the sensitive features from the high-dimensional samples in neighborhood if d is too small or the low-dimensional features contain redundancy and interference if d is too large. Likewise, due to different factors, the recognition rate has multiple turnover points.

Adjustment parameter β affects the recognition rate by changing the degree of clustering and global geometry retention. In relationship between β and recognition rate in Fig. 5, if β is too small, the proximity of samples is low and the clustering is obvious, but the degree of retention of the global geometric structure is poor. If β is too large, the global geometric structure can be improved but the clustering reduced. These factors cause reduction in recognition rate. As a result, there is an optimal β= 0.1 where the recognition rate reaches the maximum 96.43 %.

Fig. 3. Recognition rate with neighbor k

Recognition rate with neighbor k

Fig. 4. Recognition rate with dimension d

Recognition rate with dimension d

Fig. 5. Recognition rate with parameter β

Recognition rate with parameter β

The three-dimensional sensitive features of bearing high-dimensional characteristic samples reduced by LCWR according to the optimized values of k= 8, d= 3 and β= 0.1 are shown in Fig. 6. After dimension reduction, four kinds of bearing samples have no intersection and overlap. Each of them has its own clustering center. The clustering effect and global geometric structure are objective, displaying a distinct manifold structure.

Fig. 6. Embedded 3 dimensions by LCWR (k= 8, β= 0.1)

Embedded 3 dimensions by LCWR (k= 8, β= 0.1)

3.3. Dimensionality reduction effect analysis

LCWR is compared with LTSA, locally linear embedding (LLE) and principal component analysis (PCA) to verify its dimensionality reduction effect. The dimensionality reduction results of LTSA, LLE and PCA are shown in Fig. 7, Fig. 8, and Fig. 9, respectively. Generally speaking, the reduced dimensionality samples via these methods have different degrees of intersection and overlap, poor clustering within class, lack of clustering centers. It is difficult to mine the essential characteristics of the bearing state and the differences between classes. Although LTSA and LLE find the manifold structure of high-dimensional samples, they are unable to expand the gaps between dissimilar samples in neighborhood. PCA belongs to one of linear statistical distributions without considering the local structure of the samples. It makes the intraclass aggregation poor and the differences between classes unclear, which fails to reveal the non-linear manifold structure of the bearing state as shown in Fig. 9.

Fig. 7. Embedded 3 dimensions by LTSA (k= 9)

Embedded 3 dimensions by LTSA (k= 9)

Fig. 8. Embedded 3 dimensions by LLE (k= 5)

Embedded 3 dimensions by LLE (k= 5)

Combining its weight coefficient with local coordinate’s permutation, LCWR enhances the intraclass aggregation and the differences between classes, overcomes the shortcomings that LTSA and LLE can’t enlarge the gaps between dissimilar samples in neighborhood, simplifies the dimension while retaining the low-dimensional principal characteristics of high-dimensional samples, accurately reflects the relationship between signal characteristics and the bearing state, and effectively distinguishes four kinds of bearing states. As shown in Table 2, the features extracted by various methods are sent to SVM and the recognition rate of LCWR reaches the highest 96.43 % despite a little more time to run LCWR than LTSA, LLE and PCA as shown in Table 3. Therefore, in contrast to other dimensionality reduction methods, LCWR can achieve higher accuracy and prove its effectiveness.

Fig. 9. Embedded 3 dimensions by PCA

Embedded 3 dimensions by PCA

Table 2. Bearing condition recognition rate with various methods (%)

(k= 8, d= 3, β= 0.1)
(k= 9, d= 3)
(k= 5, d= 3)
(d= 3)
(d= 20)
H vs. (O+I+B)
O vs. (I+B)
B vs. (O+I)
I vs. (O+B)
I vs. B
O vs. I
O vs. B
Average rate
H-healthy, O-outer race defect, I-inner race defect, B-ball defect, NDR-non dimensionality reduction

Table 3. Running time of SVM combined with various methods (s)

(k= 8, d= 3, β= 0.1)
(k= 9, d= 3)
(k= 5, d= 3)
(d= 3)
(d= 20)
H vs. (O+I+B)
O vs. (I+B)
B vs. (O+I)
I vs. (O+B)
I vs. B
O vs. I
O vs. B
Average time

Besides, it can be founded that recognition rate of manifold dimensionality reduction using LCWR, LTSA, LLE and PCA (all greater than 90 %) is higher than that of the non-dimensionality reduction method (only 84.72 %). It is further proved that these manifold learning methods can filter redundancy and interference of the high-dimensional features and extract the intrinsic low-dimensional manifold characteristics, which can significantly improve the recognition rate of the bearing state as shown in Table 2. Meanwhile, these manifold learning methods except LCWR consume less time and get better recognition efficiency.

4. Conclusions

LCWR manifold learning is proposed to remove redundancy and noise in bearing high-dimensional fault features and perform non-linear dimensionality reduction for improvement in fault diagnosis capability. Geodesic distance based weight function is used to realign local coordinates to eliminate redundancy and interference in high-dimensional feature samples and extract low-dimensional sensitive fault features. Experiments demonstrate that the intrinsic manifold structure of high-dimensional feature samples can be well preserved after dimensionality reduction by LCWR, and the extracted low-dimensional feature samples can truly represent the non-linear characteristics of different bearing states and the gaps between them. The low-dimensional feature samples are then identified by SVM, which results in a higher recognition rate than other methods. Thus, the effectiveness of LCWR is validated. In addition, LCWR is worth further studying to save running time.


The work was supported by A Project Supported by Scientific Research Fund of Hunan Provincial Education Department (20A107), Project of National Natural Science Foundation of China (51875193) and Xiangtan guiding science and technology plan project (ZDX-CG2019004). The authors thank all reviewers for their valuable comments and constructive advice on this paper.


  1. Randall R. B., Antoni J. Rolling element bearing diagnostics – a tutorial. Mechanical Systems and Signal Processing, Vol. 25, Issue 2, 2011, p. 485-520. [Publisher]
  2. Dolenc B., Boškoski P., Juričić D. Distributed bearing fault diagnosis based on vibration analysis. Mechanical Systems and Signal Processing, Vols. 66-67, 2016, p. 521-532. [Publisher]
  3. Wu C. X., Chen T. F., Jiang R., Ning L. W., Jiang Z. A novel approach to wavelet selection and tree kernel construction for diagnosis of rolling element bearing fault. Journal of Intelligent Manufacturing, Vol. 28, Issue 8, 2017, p. 1847-1858. [Publisher]
  4. Kumar R., Singh M. Outer race defect width measurement in taper roller bearing using discrete wavelet transform of vibration signal. Measurement, Vol. 46, Issue 1, 2013, p. 537-545. [Publisher]
  5. Kankar P. K., Sharma S. C., Harsha S. P. Rolling element bearing fault diagnosis using autocorrelation and continuous wavelet transform. Journal of Vibration and Control, Vol. 17, Issue 14, 2011, p. 2081-2094. [Publisher]
  6. Patel V. N., Tandon N., Pandey R. K. Defect detection in deep groove ball bearing in presence of external vibration using envelope analysis and Duffing oscillator. Measurement, Vol. 45, Issue 5, 2012, p. 960-970. [Publisher]
  7. Wang Y., Xu G. H., Liang L., Jiang K. S. Detection of weak transient signals based on wavelet packet transform and manifold learning for rolling element bearing fault diagnosis. Mechanical Systems and Signal Processing, Vol. 54, Issue 55, 2015, p. 259-276. [Publisher]
  8. Wang J., He Q. B., Kong F. R. Multiscale envelope manifold for enhanced fault diagnosis of rotating machines. Mechanical Systems and Signal Processing, Vol. 52, Issue 53, 2015, p. 376-392. [Search CrossRef]
  9. Jiang Q. S., Jia M. P., Hu J. Z., Xu F. Y. Machinery fault diagnosis using supervised manifold learning. Mechanical Systems and Signal Processing, Vol. 23, Issue 7, 2009, p. 2301-2311. [Publisher]
  10. Wang C., Gan M., Zhu C. G. Non-negative EMD manifold for feature extraction in machinery fault diagnosis. Measurement, Vol. 70, 2015, p. 188-202. [Publisher]
  11. Zhang Z. Y., Zha H. Y. Principal manifolds and nonlinear dimension reduction via local tangent space alignment. Journal of Shanghai University, Vol. 8, Issue 4, 2004, p. 406-424. [Publisher]
  12. Wang Q., Wang W. G., Nian R., He B., Shen Y., Björk K. M., Lendasse A. Manifold learning in local tangent space via extreme learning machine. Neurocomputing, Vol. 174, 2016, p. 18-30. [Publisher]
  13. Zhang P., Qiao H., Zhang B. An improved local tangent space alignment method for manifold learning. Pattern Recognition Letters, Vol. 32, Issue 2, 2011, p. 181-189. [Publisher]
  14. Zhang Y., Li B. W., Wang W., Sun T., Yang X. Y., Wang L. Supervised locally tangent space alignment for machine fault diagnosis. Journal of Mechanical Science and Technology, Vol. 28, Issue 8, 2014, p. 2971-2977. [Publisher]
  15. Li F., Tang B. P., Yang R. S. Rotating machine fault diagnosis using dimension reduction with linear local tangent space alignment. Measurement, Vol. 46, Issue 8, 2013, p. 2525-2539. [Publisher]
  16. Kumar A., Kumar R. Manifold learning using linear local tangent space alignment (LLTSA) Algorithm for noise removal in wavelet filtered vibration signal. Journal of Nondestructive Evaluation, Vol. 35, Issue 3, 2016, p. 50. [Publisher]
  17. Su Z. Q., Tang B. P., Liu Z. R., Qin Y. Multi-fault diagnosis for rotating machinery based on orthogonal supervised linear local tangent space alignment and least square support vector machine. Neurocomputing, Vol. 157, 2015, p. 208-222. [Publisher]
  18. Wang G. B., Zhao X. Q., He Y. H. Fault diagnosis method based on supervised incremental local tangent space alignment and SVM. Applied Mechanics and Materials, Vols. 34-35, 2010, p. 1233-1237. [Publisher]
  19. Su Z. Q., Tang B. P., Deng L., Liu Z. R. Fault diagnosis method using supervised extended local tangent space alignment for dimension reduction. Measurement, Vol. 62, 2015, p. 1-14. [Publisher]
JVE Journals is rebranding to Extrica

Inspired by innovations from the previous century and the rapid growth during the last years, we are improving for excellence in your publishing experience

Read to know more
JVE Journals