# A Novel Prediction Model for Car Body Vibration Acceleration Based on Correlation Analysis and Neural Networks

Authors: Shubin Zheng; Qianwen Zhong; Xiaodong Chai; Xingjie Chen; Lele Peng
Journal: Journal of Advanced Transportation
 (2018)

---

## Abstract

This paper aims to create a prediction model for car body vibration acceleration that is reliable, effective, and close to real-world conditions. Therefore, a huge amount of data on railway parameters were collected by multiple sensors, and different correlation coefficients were selected to screen out the parameters closely correlated to car body vibration acceleration. Taking the selected parameters and previous car body vibration acceleration as the inputs, a prediction model for car body vibration acceleration was established based on several training algorithms and neural network structures. Then, the model was successfully applied to predict the car body vibration acceleration of test datasets on different segments of the same railway. The results show that the proposed method overcomes the complexity and uncertainty of the multiparameter coupling analysis in traditional theoretical models. The research findings boast a great potential for application.


---

## Body

## 1. Introduction

Passenger comfort is an important indicator of the operation quality of passenger trains. Previous studies [1, 2] have shown that passenger comfort can be estimated indirectly by parameters like vibration acceleration of the car body. Based on the estimated passenger comfort, it is possible to identify the warning signals or system statuses needed to ensure the smooth operation of the train.Much research has been done to forecast the vibration acceleration of trains. For instance, Shafiullah et al. [3] predicted the forward and backward vertical acceleration conditions by popular regression algorithms. Zhai et al. [4] created a comprehensive train-track dynamics model to predict the ground vibrations of high-speed trains. Inspired by the dynamics model, Czop et al. [5] proposed a rail irregularity detection method based on the bearing box acceleration during train operation and successfully applied the method to recognize the rail regularities of a typical railway in Poland. Qian et al. [6] established a model to predict the vibration acceleration of high-speed trains based on nonlinear autoregressive neural network with exogenous inputs (NARX NN) and multibody dynamic model and proved the prediction accuracy of the model through experimental analysis.In addition, some scholars have attempted to infer important parameters of railways from vibration acceleration of the car body. For example, Connolly et al. [7] assessed the effects of vibration acceleration on passenger comfort and track performance. Koo et al. [8] put forward theoretical derailment coefficients for single wheel pairs, considering the impacts from lateral vibration acceleration and gyroscopic factors as well as flange angle, friction coefficient, wheel unloading, wheel radius, gauge, and bearing position. Navik et al. [9] developed a new sensor system that captures the dynamic behaviour of high-speed rail with several sensors placed at an interval of 150m and predicted the maximum vertical displacement, train speed, dynamic behaviour, and quantification modal parameters with vibration acceleration time series.In general, the previous research into vibration acceleration had concentrated on the traditional multibody dynamics modelling, and the research results were mainly derived through simulation. In actual operation, the train is faced with a complex environment and uncertain track conditions. Thus, there is always some gap between the simulated state and the actual state of railway track and train operation. This calls for a new theoretical model that can accurately reflect the actual conditions of the train and the track.In light of the above, this paper aims to propose a prediction model for car body vibration acceleration that is reliable, effective, and close to real-world conditions. Therefore, a huge amount of data on railway parameters were collected by multiple sensors, and different correlation coefficients were selected to screen out the parameters closely correlated to car body vibration acceleration. Taking the selected parameters and previous car body vibration acceleration as the inputs, a prediction model for car body vibration acceleration was established based on several training algorithms and neural network structures. Then, the model was successfully applied to predict the car body vibration acceleration of test datasets on different segments of the same railway. The results show that the proposed method overcomes the complexity and uncertainty of the multiparameter coupling analysis in traditional theoretical models. The research findings boast a great potential for application.The remainder of this paper is organized as follows: Section2 introduces the data preprocessing and feature selection methods; Section 3 describes the structures of the neural networks and several popular training algorithms; Section 4 verifies the effect of the proposed model on different datasets, and the results under different structures are discussed and compared; Section 5 wraps up this paper with some meaningful conclusions.


## 2. Data Preprocessing and Feature Selection

### 2.1. Data Preprocessing

The research data are a collection of useful data from actual railways. The sensors were subjected to noise reduction and antijamming processing, aiming to enhance the readability and usability of the collected data. In addition, the data underwent a multistep preprocessing.Firstly, time synchronization was performed on the huge amount of data captured by multiple sensors to remove time points with missing or abnormal values and eliminate the variables of constant values. In this way, the data containing useful information were screened out.Secondly, the modelling variables were determined, excluding those rarely used, irrelevant to mechanics, or difficult to measure in actual conditions.Thirdly, the influencing factors of the relevant variables in the transfer part were minimized, e.g., the angular acceleration at different positions of car body, as the prediction variables were expected to consider such parameters as train structure, track state, and operation state. Note that the minimization only treats the transfer process as a black box, rather than overlooking the impacts of the influencing factors. The treatment simplifies the modelling process.Finally, the preprocessed data were normalized for further use.


### 2.2. Variable Selection

#### 2.2.1. Linear Correlation

The linear correlation of two random variables can be measured by the Pearson’s correlation coefficient (P). If each variable hasN scalar observations, then the Pearson’s correlation coefficient ρP can be defined as [10–12](1)ρPM1,M2=1N-1∑i=1NM1i-μM1-σM1M2i-μM2σM2where μM1 and σM1 are the mean and standard deviations of M1, respectively, and μM2 and σM2 are the mean and standard deviations of M2, respectively.The Pearson’s correlation coefficient can also be described based on the covariance ofM1 and M2 as follows.(2)ρPM1,M2=cov⁡M1,M2σM1σM2


#### 2.2.2. Nonlinear Correlation

The linear correlation coefficient cannot fully reflect the relationship between variables, owing to the possible existence of nonlinear correlations. Thus, the Spearman’s rank correlation coefficient (S) [13] was employed to analyze the nonlinear correlations between variables. This coefficient ρS can be defined as follows.(3)ρSM1,M2=∑iM1i-M1-M2i-M2-∑iM1i-M1-2∑iM2i-M2-2By this definition, M1 and M2 are fully correlated as long as they share a monotonic functional relationship. This is different from Pearson’s correlation, in which only linearly correlated variables are considered as relevant to each other.Then, the correlation coefficient matrixCf of n random variables is a matrix of correlation coefficients for each pair of variable combinations.(4)Cf=ρM1,M1⋯ρM1,Mn⋮⋱⋮ρMn,M1⋯ρMn,MnSince M1 and M2 are always directly correlated, the diagonal entries are 1, that is,(5)Cf=1⋯ρM1,Mn⋮⋱⋮ρMn,M1⋯1.Through the above calculation, the variables with the greater values under the two types of correlation coefficients,CfP and CfS, can be selected as predictor variables.


## 2.1. Data Preprocessing

The research data are a collection of useful data from actual railways. The sensors were subjected to noise reduction and antijamming processing, aiming to enhance the readability and usability of the collected data. In addition, the data underwent a multistep preprocessing.Firstly, time synchronization was performed on the huge amount of data captured by multiple sensors to remove time points with missing or abnormal values and eliminate the variables of constant values. In this way, the data containing useful information were screened out.Secondly, the modelling variables were determined, excluding those rarely used, irrelevant to mechanics, or difficult to measure in actual conditions.Thirdly, the influencing factors of the relevant variables in the transfer part were minimized, e.g., the angular acceleration at different positions of car body, as the prediction variables were expected to consider such parameters as train structure, track state, and operation state. Note that the minimization only treats the transfer process as a black box, rather than overlooking the impacts of the influencing factors. The treatment simplifies the modelling process.Finally, the preprocessed data were normalized for further use.


## 2.2. Variable Selection

### 2.2.1. Linear Correlation

The linear correlation of two random variables can be measured by the Pearson’s correlation coefficient (P). If each variable hasN scalar observations, then the Pearson’s correlation coefficient ρP can be defined as [10–12](1)ρPM1,M2=1N-1∑i=1NM1i-μM1-σM1M2i-μM2σM2where μM1 and σM1 are the mean and standard deviations of M1, respectively, and μM2 and σM2 are the mean and standard deviations of M2, respectively.The Pearson’s correlation coefficient can also be described based on the covariance ofM1 and M2 as follows.(2)ρPM1,M2=cov⁡M1,M2σM1σM2


### 2.2.2. Nonlinear Correlation

The linear correlation coefficient cannot fully reflect the relationship between variables, owing to the possible existence of nonlinear correlations. Thus, the Spearman’s rank correlation coefficient (S) [13] was employed to analyze the nonlinear correlations between variables. This coefficient ρS can be defined as follows.(3)ρSM1,M2=∑iM1i-M1-M2i-M2-∑iM1i-M1-2∑iM2i-M2-2By this definition, M1 and M2 are fully correlated as long as they share a monotonic functional relationship. This is different from Pearson’s correlation, in which only linearly correlated variables are considered as relevant to each other.Then, the correlation coefficient matrixCf of n random variables is a matrix of correlation coefficients for each pair of variable combinations.(4)Cf=ρM1,M1⋯ρM1,Mn⋮⋱⋮ρMn,M1⋯ρMn,MnSince M1 and M2 are always directly correlated, the diagonal entries are 1, that is,(5)Cf=1⋯ρM1,Mn⋮⋱⋮ρMn,M1⋯1.Through the above calculation, the variables with the greater values under the two types of correlation coefficients,CfP and CfS, can be selected as predictor variables.


## 2.2.1. Linear Correlation

The linear correlation of two random variables can be measured by the Pearson’s correlation coefficient (P). If each variable hasN scalar observations, then the Pearson’s correlation coefficient ρP can be defined as [10–12](1)ρPM1,M2=1N-1∑i=1NM1i-μM1-σM1M2i-μM2σM2where μM1 and σM1 are the mean and standard deviations of M1, respectively, and μM2 and σM2 are the mean and standard deviations of M2, respectively.The Pearson’s correlation coefficient can also be described based on the covariance ofM1 and M2 as follows.(2)ρPM1,M2=cov⁡M1,M2σM1σM2


## 2.2.2. Nonlinear Correlation

The linear correlation coefficient cannot fully reflect the relationship between variables, owing to the possible existence of nonlinear correlations. Thus, the Spearman’s rank correlation coefficient (S) [13] was employed to analyze the nonlinear correlations between variables. This coefficient ρS can be defined as follows.(3)ρSM1,M2=∑iM1i-M1-M2i-M2-∑iM1i-M1-2∑iM2i-M2-2By this definition, M1 and M2 are fully correlated as long as they share a monotonic functional relationship. This is different from Pearson’s correlation, in which only linearly correlated variables are considered as relevant to each other.Then, the correlation coefficient matrixCf of n random variables is a matrix of correlation coefficients for each pair of variable combinations.(4)Cf=ρM1,M1⋯ρM1,Mn⋮⋱⋮ρMn,M1⋯ρMn,MnSince M1 and M2 are always directly correlated, the diagonal entries are 1, that is,(5)Cf=1⋯ρM1,Mn⋮⋱⋮ρMn,M1⋯1.Through the above calculation, the variables with the greater values under the two types of correlation coefficients,CfP and CfS, can be selected as predictor variables.


## 3. Method of Prediction Model

### 3.1. Training Algorithms

The training algorithms pursue the minimum gap between the predicted value and the measured value. In most cases, the minimization is achieved by adjusting the weights of each layer in the neural networks. Below is a brief introduction to the training algorithms adopted for our research.(1) Broyden–Fletcher–Goldfarb–Shanno (BFGS) Quasi-Newton Backpropagation (BQ).The BQ is an alternative to the conjugate gradient methods for fast optimization. The basic formula [14] is as follows:(6)dk+1=dk-Hk-1gkwhere dk+1 is the step distance of k+1; dk is the step distance of k; Hk-1 is the Hessian matrix (second derivatives) of performance index at the current weights and biases; and gk is the gradient of step k.(2) Conjugate Gradient Backpropagation with Powell-Beale Restarts (CGB). For all conjugate gradient algorithms, the search direction is periodically reset to the negative of the gradient. The reset happens whenever there is too little orthogonality left between the current and the previous gradients. This condition is tested with the following inequality [15].(7)gk-1Tgk≥0.2gk2(3) Conjugate Gradient Backpropagation with Fletcher-Reeves Updates (CGF). The optimal distances to move along the current search direction, the new search direction, and the conjugate weight adjustment coefficient are, respectively, computed by the following equations [16]:(8)dk+1=dkαkpk(9)pk=-gk+βkpk-1(10)βk=gkTgkgk-1Tgk-1where αk is a variable to minimize the performance along the current search direction; pk is the next search direction, which is conjugate to the previous search direction; and βk is a constant that adjusts the conjugate weights. Most conjugate gradient algorithms differ only in the calculation of the constant βk.(4) Conjugate Gradient Backpropagation with Polak-Ribiére Updates (CGP). For this algorithm, the search direction in each iteration is the same as the new search direction in the CGF algorithm [16]. The constant βk can be obtained by(11)βk=∆gk-1Tgkgk-1Tgk-1where ∆gk-1T=gk-gk-1T is the change transposing form in the gradient from the previous iteration.(5) One-Step Secant Backpropagation (OSS). The OSS algorithm is an approximate secant method with relatively small storage and computing load [17]. By this method, the weights can be adjusted in the following manner:(12)pk+1=-gk+aksk+bk∆gkwhere ak is the weight adjustment coefficient; sk is the change in the weights of the previous iteration; and bk is the gradient adjustment coefficient.(6) Resilient Backpropagation (RB). The RB is a local learning algorithm that is easy to implement and compute. In this algorithm, the weights are updated according to the behaviour of the sign sequence for the partial derivatives in each dimension of the weight space [18]: (13)∆wk=-ϵ∂Ek∂wk+μ∆wk-1(14)∂Ek-1∂wk-1∙∂Ek∂wk=0where μ is a parameter to scale the influence of the previous iteration and ϵ is the learning rate.(7) Scaled Conjugate Gradient Backpropagation (SCG). The SCG is a step-size scaling algorithm [19, 20] created to expand the applicable scope of conjugate gradient (CG) algorithm from the functions with positive definite Hessian matrices. The SCG works faster than other second-order algorithms, as it prevents the time-consuming search in each iteration.(8) Levenberg-Marquardt Backpropagation (LM). The LM algorithm uses the approximate Hessian matrix in the following Newton-like update [21, 22]:(15)H=JTJ(16)dk+1=dk-JTJ+μI-1JTewhere H is the Hessian matrix; J is the Jacobian matrix containing the first-order derivatives of network errors with respect to the weights and biases; and e is a vector of network errors. When the scalar μ is zero, the LM algorithm is essentially a Newton’s method using the approximate Hessian matrix.(9) Bayesian Regularization Backpropagation (BR). Besides reducing the sum of squared errors, ED, the regularization adds an additional term. Thus, the objective function E(w) can be expressed as(17)Ew=βED+αEWwhere EW is the sum of squares of network weights and α and β are two parameters of the objective function. Under the Bayesian framework [23, 24], this method can optimize the regularization parameters.With different weight adjustment mechanisms, the above algorithms differ in training accuracy, storage, and running time. Their performance will be compared in the following section. In addition, the functions of the output layer and the hidden layer are as follows.(18)fHx=tansigx=21+exp⁡-2∗x-1(19)fOx=purelinx=x


### 3.2. Structures of Neural Networks

#### 3.2.1. Feedforward Neural Networks (FFNN)

The FFNN is one of the most popular neural networks. The networks have multiple layers, including an input layer, several hidden layers, and an output layer. Layers are connected to each other by nodes or neurons. The input layer is connected to the inputs, while the output layer exports the predicted results. Each hidden layer treats the output of the previous layer as its input.


#### 3.2.2. Time-Series Neural Network without Feedback Time Delays (TSNF)

The parameters like vibration and body attitude are often affected by relevant factors and the existing states. This type of variable often uses more accurate time-series neural network prediction methods. A typical time-series neural network structure is as shown in Figure1, where the input vectors are formed by the input variables and their delays.Figure 1
Structure of the TSNF.Other than the input layer, the other parts of the TSNF are similar to those of the FFNN. In other words, the TSNF also has multiple hidden layers and one output layer. Each hidden layer contains a certain number of neurons.


#### 3.2.3. Time-Series Neural Networks with Feedback Time Delays (TSF)

TSF is another common time-series neural network (Figure2). The structure of TSF originates from the NARX. Unlike the TSNF, the TSF contains both input delayed variables and feedback delays.Figure 2
Structure of the TSF.The introduction of delayed feedback is equivalent to taking the states of the target close to the next predicted moment as the input variables. According to the analysis in the previous section, the TSF structure is expected to further improve the prediction accuracy. Hence, the performance of the prediction model can be optimized by this structure. Of course, the other two structures cannot be neglected in actual practice; it is sometimes necessary to make predictions based on predictors with or without feedback.


### 3.3. Model Construction Process

The optimal prediction model can be constructed in two phases, namely, data preprocessing and variable selection, and the model construction based on neural networks. Figure3 is the flow chart of the model construction.Figure 3
Flow chart of model construction.In the data preprocessing and variable selection block, firstly the dataset applied here is from GJ-5 track inspection car. To make the following process more effective, we screen the missing and singular values and delete the corresponding sampling points that may deteriorate the analysis results. Then the target variables are defined as the vibration acceleration of car body from three directions, i.e., horizontal, vertical, and lateral directions. The rest measured parameters (139 other parameters in our dataset) are all considered as the predictors at this step. However, under reasonable deduction, there must be a big mount of the predictors being redundant and almost having no impact on the vibration of car body. To solve this problem, next, the correlation analysis between the predictors and the response parameters is carried out. Specifically, the Pearson’s correlation coefficient and Spearman’s rank correlation coefficient were adopted to select the predictor variables. Since the accuracy of car body vibration forecast model is one of the common key indices, all the predictors with absolute correlation coefficient over 0.1 are taken into consideration as inputs in the following model building process. It should be noted that if the data is very large or selected predictors are still redundant, the chosen bound of the correlation coefficient could be changed to improve the efficiency.Next, in the building the prediction model block, the selected variables were used to generate the models trained by different algorithms under three neural network structures. The algorithms chosen are widely verified effectively in neural network method and briefly introduced in Section3.1, which contains LM, BR, BFG, RP, SCG, CGB, CGF, CGP, and OSS, respectively. The three neural networks structures are FF, TSNF, and TSF, respectively, which are illustrated in Section 3.2. The performance evaluation indices of these models were compared to determine the optimal prediction model for car body vibration acceleration. In this work, the accuracy is defined as the priority pursuing goal of the ranking as we want to improve the suitability of the proposed forecast model, and the real measured values of car body vibration acceleration are very low, normally less than 0.1. Under such consideration, the MSE is set to be the main index of the comparison. If the MSE values are very close, the R and MAE values are compared as the auxiliary indices. Once the data are large and the running times of algorithms are obviously different, a more comprehensive ranking equation should be designed containing both the three indices and the running time with reasonable corresponding weights. Finally, the forecast model of car body vibration is obtained.


## 3.1. Training Algorithms

The training algorithms pursue the minimum gap between the predicted value and the measured value. In most cases, the minimization is achieved by adjusting the weights of each layer in the neural networks. Below is a brief introduction to the training algorithms adopted for our research.(1) Broyden–Fletcher–Goldfarb–Shanno (BFGS) Quasi-Newton Backpropagation (BQ).The BQ is an alternative to the conjugate gradient methods for fast optimization. The basic formula [14] is as follows:(6)dk+1=dk-Hk-1gkwhere dk+1 is the step distance of k+1; dk is the step distance of k; Hk-1 is the Hessian matrix (second derivatives) of performance index at the current weights and biases; and gk is the gradient of step k.(2) Conjugate Gradient Backpropagation with Powell-Beale Restarts (CGB). For all conjugate gradient algorithms, the search direction is periodically reset to the negative of the gradient. The reset happens whenever there is too little orthogonality left between the current and the previous gradients. This condition is tested with the following inequality [15].(7)gk-1Tgk≥0.2gk2(3) Conjugate Gradient Backpropagation with Fletcher-Reeves Updates (CGF). The optimal distances to move along the current search direction, the new search direction, and the conjugate weight adjustment coefficient are, respectively, computed by the following equations [16]:(8)dk+1=dkαkpk(9)pk=-gk+βkpk-1(10)βk=gkTgkgk-1Tgk-1where αk is a variable to minimize the performance along the current search direction; pk is the next search direction, which is conjugate to the previous search direction; and βk is a constant that adjusts the conjugate weights. Most conjugate gradient algorithms differ only in the calculation of the constant βk.(4) Conjugate Gradient Backpropagation with Polak-Ribiére Updates (CGP). For this algorithm, the search direction in each iteration is the same as the new search direction in the CGF algorithm [16]. The constant βk can be obtained by(11)βk=∆gk-1Tgkgk-1Tgk-1where ∆gk-1T=gk-gk-1T is the change transposing form in the gradient from the previous iteration.(5) One-Step Secant Backpropagation (OSS). The OSS algorithm is an approximate secant method with relatively small storage and computing load [17]. By this method, the weights can be adjusted in the following manner:(12)pk+1=-gk+aksk+bk∆gkwhere ak is the weight adjustment coefficient; sk is the change in the weights of the previous iteration; and bk is the gradient adjustment coefficient.(6) Resilient Backpropagation (RB). The RB is a local learning algorithm that is easy to implement and compute. In this algorithm, the weights are updated according to the behaviour of the sign sequence for the partial derivatives in each dimension of the weight space [18]: (13)∆wk=-ϵ∂Ek∂wk+μ∆wk-1(14)∂Ek-1∂wk-1∙∂Ek∂wk=0where μ is a parameter to scale the influence of the previous iteration and ϵ is the learning rate.(7) Scaled Conjugate Gradient Backpropagation (SCG). The SCG is a step-size scaling algorithm [19, 20] created to expand the applicable scope of conjugate gradient (CG) algorithm from the functions with positive definite Hessian matrices. The SCG works faster than other second-order algorithms, as it prevents the time-consuming search in each iteration.(8) Levenberg-Marquardt Backpropagation (LM). The LM algorithm uses the approximate Hessian matrix in the following Newton-like update [21, 22]:(15)H=JTJ(16)dk+1=dk-JTJ+μI-1JTewhere H is the Hessian matrix; J is the Jacobian matrix containing the first-order derivatives of network errors with respect to the weights and biases; and e is a vector of network errors. When the scalar μ is zero, the LM algorithm is essentially a Newton’s method using the approximate Hessian matrix.(9) Bayesian Regularization Backpropagation (BR). Besides reducing the sum of squared errors, ED, the regularization adds an additional term. Thus, the objective function E(w) can be expressed as(17)Ew=βED+αEWwhere EW is the sum of squares of network weights and α and β are two parameters of the objective function. Under the Bayesian framework [23, 24], this method can optimize the regularization parameters.With different weight adjustment mechanisms, the above algorithms differ in training accuracy, storage, and running time. Their performance will be compared in the following section. In addition, the functions of the output layer and the hidden layer are as follows.(18)fHx=tansigx=21+exp⁡-2∗x-1(19)fOx=purelinx=x


## 3.2. Structures of Neural Networks

### 3.2.1. Feedforward Neural Networks (FFNN)

The FFNN is one of the most popular neural networks. The networks have multiple layers, including an input layer, several hidden layers, and an output layer. Layers are connected to each other by nodes or neurons. The input layer is connected to the inputs, while the output layer exports the predicted results. Each hidden layer treats the output of the previous layer as its input.


### 3.2.2. Time-Series Neural Network without Feedback Time Delays (TSNF)

The parameters like vibration and body attitude are often affected by relevant factors and the existing states. This type of variable often uses more accurate time-series neural network prediction methods. A typical time-series neural network structure is as shown in Figure1, where the input vectors are formed by the input variables and their delays.Figure 1
Structure of the TSNF.Other than the input layer, the other parts of the TSNF are similar to those of the FFNN. In other words, the TSNF also has multiple hidden layers and one output layer. Each hidden layer contains a certain number of neurons.


### 3.2.3. Time-Series Neural Networks with Feedback Time Delays (TSF)

TSF is another common time-series neural network (Figure2). The structure of TSF originates from the NARX. Unlike the TSNF, the TSF contains both input delayed variables and feedback delays.Figure 2
Structure of the TSF.The introduction of delayed feedback is equivalent to taking the states of the target close to the next predicted moment as the input variables. According to the analysis in the previous section, the TSF structure is expected to further improve the prediction accuracy. Hence, the performance of the prediction model can be optimized by this structure. Of course, the other two structures cannot be neglected in actual practice; it is sometimes necessary to make predictions based on predictors with or without feedback.


## 3.2.1. Feedforward Neural Networks (FFNN)

The FFNN is one of the most popular neural networks. The networks have multiple layers, including an input layer, several hidden layers, and an output layer. Layers are connected to each other by nodes or neurons. The input layer is connected to the inputs, while the output layer exports the predicted results. Each hidden layer treats the output of the previous layer as its input.


## 3.2.2. Time-Series Neural Network without Feedback Time Delays (TSNF)

The parameters like vibration and body attitude are often affected by relevant factors and the existing states. This type of variable often uses more accurate time-series neural network prediction methods. A typical time-series neural network structure is as shown in Figure1, where the input vectors are formed by the input variables and their delays.Figure 1
Structure of the TSNF.Other than the input layer, the other parts of the TSNF are similar to those of the FFNN. In other words, the TSNF also has multiple hidden layers and one output layer. Each hidden layer contains a certain number of neurons.


## 3.2.3. Time-Series Neural Networks with Feedback Time Delays (TSF)

TSF is another common time-series neural network (Figure2). The structure of TSF originates from the NARX. Unlike the TSNF, the TSF contains both input delayed variables and feedback delays.Figure 2
Structure of the TSF.The introduction of delayed feedback is equivalent to taking the states of the target close to the next predicted moment as the input variables. According to the analysis in the previous section, the TSF structure is expected to further improve the prediction accuracy. Hence, the performance of the prediction model can be optimized by this structure. Of course, the other two structures cannot be neglected in actual practice; it is sometimes necessary to make predictions based on predictors with or without feedback.


## 3.3. Model Construction Process

The optimal prediction model can be constructed in two phases, namely, data preprocessing and variable selection, and the model construction based on neural networks. Figure3 is the flow chart of the model construction.Figure 3
Flow chart of model construction.In the data preprocessing and variable selection block, firstly the dataset applied here is from GJ-5 track inspection car. To make the following process more effective, we screen the missing and singular values and delete the corresponding sampling points that may deteriorate the analysis results. Then the target variables are defined as the vibration acceleration of car body from three directions, i.e., horizontal, vertical, and lateral directions. The rest measured parameters (139 other parameters in our dataset) are all considered as the predictors at this step. However, under reasonable deduction, there must be a big mount of the predictors being redundant and almost having no impact on the vibration of car body. To solve this problem, next, the correlation analysis between the predictors and the response parameters is carried out. Specifically, the Pearson’s correlation coefficient and Spearman’s rank correlation coefficient were adopted to select the predictor variables. Since the accuracy of car body vibration forecast model is one of the common key indices, all the predictors with absolute correlation coefficient over 0.1 are taken into consideration as inputs in the following model building process. It should be noted that if the data is very large or selected predictors are still redundant, the chosen bound of the correlation coefficient could be changed to improve the efficiency.Next, in the building the prediction model block, the selected variables were used to generate the models trained by different algorithms under three neural network structures. The algorithms chosen are widely verified effectively in neural network method and briefly introduced in Section3.1, which contains LM, BR, BFG, RP, SCG, CGB, CGF, CGP, and OSS, respectively. The three neural networks structures are FF, TSNF, and TSF, respectively, which are illustrated in Section 3.2. The performance evaluation indices of these models were compared to determine the optimal prediction model for car body vibration acceleration. In this work, the accuracy is defined as the priority pursuing goal of the ranking as we want to improve the suitability of the proposed forecast model, and the real measured values of car body vibration acceleration are very low, normally less than 0.1. Under such consideration, the MSE is set to be the main index of the comparison. If the MSE values are very close, the R and MAE values are compared as the auxiliary indices. Once the data are large and the running times of algorithms are obviously different, a more comprehensive ranking equation should be designed containing both the three indices and the running time with reasonable corresponding weights. Finally, the forecast model of car body vibration is obtained.


## 4. Results and Discussion

### 4.1. Measurement System

The training data were collected by a GJ-5 track inspection vehicle of ImageMap, Inc., between Shenzhen and Guangzhou, two first-tier cities in China. The verification data were acquired on the return section. The main test items include geometry inspection items, on-board dynamics test items, and ground dynamics test items. Specifically, the track geometry inspections include different wavelengths and gauge, track pitch variation rate, level, triangle pit twisting curvature, and curvature change rate; on-board dynamics test items mainly include wheel rail force and left and right wheel vertical forces, lateral force, derailment coefficient, deceleration rate, three-section acceleration reduction rate of wheel load, lateral stability index of the structure, stability and the vehicle body vertical and lateral acceleration of the left and right axle box, frame, and body of the vehicle left and right axle boxes, vertical and lateral acceleration of the frame, car body vertical and lateral acceleration of the left and right axle boxes, frames, and bodywork of the middle car; ground dynamics test items include derailment coefficient, load shedding rate, lateral force, vertical force, and vertical rail displacement horizontal. The main parameters of track inspection and test accuracy of the inspection vehicle are listed in Table1.Table 1
Main parameters of track inspection and test accuracy of the inspection vehicle.
Test items Accuracy Range Gauge ± 0.8 mm 1420~1480 mm Rail direction (left & right,λ≤30m) ± 1.0 mm ± 100 mm Rail direction (left & right,λ≤50m) ± 2.0 mm ± 100 mm Height (left & right,λ≤30m) ± 1.0 mm ± 100 mm Height (left & right,λ≤50m) ± 2.0 mm ± 100 mm Horizon ± 1.5 mm ± 50 mm Super elevation ± 1.5 mm ± 50 mm Triangle pit ± 1.5 mm ± 220 mm Curve 1.2 × 10 - 4 m-1 −− Body acceleration ± 0.01 g ± 1 g Rail bottom slope ± 0.25 deg −− Rail cross section (left & right) ± 0.5 mm −−Through the correlation analysis, the predictor variables for car body vibration acceleration were selected by the absolute values of coefficients falling between 0.1 and 0.9. This interval was chosen to exclude parameters loosely correlated with or similar to the target variable, making it possible to obtain a practical analysis of the impact of each track factor on the target variable. The selected variables are listed in Table2.Table 2
The prediction variables selected through correlation analysis.
Car body_HACCEL (g) Car body_VACCEL (g) Car body_LACCEL (g) ρ P ρ S  ρ P ρ S  ρ P ρ S Superelev (mm) -0.50 -0.25 L Surf 1 (mm) -0.23 -0.20 Mileage (km) -0.49 -0.49 Curve (rad/km) -0.14 -0.18 R Surf 1 (mm) -0.27 -0.22 Gauge (mm) 0.16 0.11 C_IRREG_LEFT (mm) 0.31 0.19 L Surf 2 (mm) -0.29 -0.24 Superelev (mm) 0.26 0.24 C_IRREG_RIGHT (mm) 0.30 0.18 R Surf 2 (mm) -0.32 -0.26 Curve (rad/km) 0.29 0.26 CURVE_RATE (rad/km/m) -0.15 -0.11 L Surf 3 (mm) -0.31 -0.25 Speed(kph) -0.48 -0.60 LOFFSET (mm) -0.22 0.05 R Surf 3 (mm) -0.35 -0.29 CURVE_RATE (rad/km/m) -0.11 -0.09 ROFFSET (mm) -0.27 0.04 L Surf 4 (mm) -0.17 -0.12 LOFFSET (mm) 0.25 0.20 LSURFACE (mm) -0.43 -0.31 R Surf 4 (mm) -0.21 -0.16 ROFFSET (mm) 0.23 0.20 RSURFACE (mm) 0.39 0.26 - - - LSURFACE (mm) -0.18 -0.15 YAW (deg/s) 0.12 0.14 - - - YAW (deg/s) -0.21 -0.19 - - - - - - BOGIE_FRAME_HACCEL (g) 0.43 0.44 - - - - - - BOGIE_FRAME_VACCEL (g) 0.41 0.42 Total number 10 Total number 8 Total number 12As shown in Table2, it is clear that the vertical vibration acceleration of the car body directly hinges on the surface conditions of the track. The horizontal and lateral acceleration are affected by relatively more factors, owing to the track state and train operation. Moreover, there are some repeated variables, which is not out of expectation. Through the above processes, the main influencing factors of car body vibration acceleration were all identified, laying the basis for subsequent improvement of train structure and passenger comfort. Finally, the repeated variables were eliminated, leaving a total of 23 predictor variables.


### 4.2. Performance Evaluation Indices

Three indices were selected to evaluate the performance of our prediction model: the mean square error (MSE), mean absolute error (MAE), and regression coefficient (R). The MSE served as the main index and the other two as the auxiliary indices.The MSE can be calculated by the equation below:(20)MSE=1N∑i=1Nei2=1N∑i=1Ny^i-yi2where y^i is a vector of N predictions; yi is the vector of observed values corresponding to the inputs; and ei is the square of the errors. When the MSE is close to zero, it means the model is suitable for prediction when it is not overfitted.The MAE is a yardstick of the accuracy of evaluation and prediction [25, 26]. The definition of the MAE is as follows.(21)MAE=1N∑i=1Nyi-y^iThe R indicates the amount of variance explained by the prediction model. This index can be expressed as(22)R=∑i=1Nyi^-y^i-yi-y-i∑i=1Nyi^-y^i-2∑i=1Nyi-y-i2where y-i is the mean value of the measured data; y^i- is the mean value of the predicted data; ∑i=1N(yi-y^i)2is the residual sum of squares; and ∑i=1N(yi-y-i)2 is the explained sum of squares. The value of R falls between 0 and 1. If the R is close to one, it means the model has explained the majority of the variance [27, 28].


### 4.3. Model Performance

Our prediction models were built on the Matlab software with some codes in the neural network toolbox. The program runs on a Lenovo workstation (CPU: Intel® Xeon® Processor E5-2623 v3; 3.00GHz; RMB: 64GB). To compare the models based on the said three neural network structures, the number of hidden layers and the number of hidden layer nodes were set to 1 and 15, respectively. The number of hidden layer nodes was determined because the FFNN structure can achieve the optimal performance with 15 hidden layer nodes through 10~20 traversal iterations.


#### 4.3.1. Training Performance

As mentioned before, the modelling data were captured at the interval of 0.25m by a GJ-5 track inspection vehicle moving from Shenzhen to Guangzhou. In total, 50,000 sample points which are continuous in time series were selected for modelling.eX replaces 10X. It can be seen that the models trained with LM and BR outperformed those trained by other algorithms, but the run times of the two models were relatively long.For better understanding, the predicted values of the FFNN-structure model trained by BR were compared with the measured values in Figure4, where the horizontal axis is the total number of samples and the vertical axis is the target variable (i.e., the car body vibration acceleration in three directions).Figure 4
Comparison between the predicted values of the FFNN-structure prediction model trained by BR with the measured values.As shown in Figure4, the predicted acceleration in all three directions basically conformed to the trend of the target variables. However, only the predicted acceleration in the horizontal direction was entirely consistent with the measured value, while that in the other two directions merely approximated the mean value of the target variables. The magnitude and range of the measured values were not well reflected.The performance of the TSNF-structure prediction model is presented in Table4. It can be seen that the three indices were all improved from the levels in Table 3, indicating that it is meaningful to consider the time delays. This is because the car body has some time delays in its response to the relevant factors, such as track surface and driver’s operation. Nonetheless, it is also learned that the TSNF increased the run time from that of the FFNN.Table 3
Performance of the FFNN-structure prediction model.
Method Set MSE R MAE Running time (s) BQ Training 1.67e-04 0.21 0.0095 4.04 Validation 1.88e-04 -0.10 0.0105 Test 5.43e-04 0.21 0.0191 CGB Training 5.29e-05 0.75 0.0053 2.21 Validation 4.81e-05 0.58 0.0052 Test 1.15e-04 0.71 0.0085 CGF Training 1.42e-04 0.25 0.0089 1.52 Validation 1.94e-04 0.24 0.0112 Test 3.71e-04 0.38 0.0163 CGP Training 1.01e-04 0.52 0.0077 1.81 Validation 8.24e-05 0.34 0.0068 Test 3.38e-04 0.33 0.0136 OSS Training 5.79e-04 -0.22 0.0196 1.68 Validation 4.04e-04 -0.44 0.0174 Test 5.77e-04 -0.43 0.0208 RB Training 2.91e-05 0.89 0.0040 1.77 Validation 2.52e-05 0.80 0.0038 Test 7.86e-05 0.71 0.0057 SCG Training 1.32e-04 0.43 0.0088 1.09 Validation 1.31e-04 0.53 0.0094 Test 7.86e-05 0.30 0.0213 LM Training 2.61e-05 0.92 0.0038 16.20 Validation 4.91e-05 0.87 0.0053 Test 2.21e-04 0.86 0.0134 BR Training 1.77e-05 0.94 0.0031 289.73 Test 2.38e-05 0.78 0.0033Table 4
Performances of the TSNF-structure prediction model.
Method MSE R MAE Running time (s) BQ 1.51e-05 0.94 0.0029 2.01e+03 CGB 1.40e-05 0.95 0.0028 244.87 CGF 1.44e-05 0.94 0.0028 252.16 CGP 1.41e-05 0.95 0.0028 237.03 OSS 1.81e-05 0.92 0.0032 299.56 RB 2.03e-05 0.91 0.0034 27.15 SCG 1.41e-05 0.95 0.0028 119.79 LM 1.19e-05 0.96 0.0026 5.64e+04 BR 1.20e-05 0.96 0.0026 8.76e+03The predicted values of the TSNF-structure model trained by BR were compared with the measured values in Figure5. It is clear that the TSNF-structure model reflected the magnitude and range of the measured values more accurately than the FFNN-structure model.Figure 5
Comparison between the predicted values of the TSNF-structure prediction model trained by BR with the measured values.Finally, the performance of the TSF-structure model (Table5) shows that the model performance changed little with the training algorithms. The order of magnitude of the MSE was always at 10-6; the R value remained near 0.99. Therefore, the TSF structure can bless the model with a good performance with a short run time.Table 5
Performance of the TSF-structure prediction model.
Method MSE R MAE Running time (s) CGB 2.27e-06 0.9898 0.0012 257.82 CGF 2.29e-06 0.9897 0.0012 263.01 CGP 2.29e-06 0.9897 0.0012 258.24 OSS 3.18e-06 0.9876 0.0014 321.94 BQ 2.54e-06 0.9890 0.0012 1.91e+03 RB 7.67e-06 0.9618 0.0021 36.37 SCG 2.39e-06 0.9895 0.0012 81.06 LM 2.02e-06 0.9910 0.0011 5.26e+04 BR 2.05e-06 0.9908 0.0011 1.14e+04The predicted values of the TSF-structure model trained by LM were compared with the measured values in Figure6. It can be seen that the predicted values were highly consistent with the measured values. Compared to the other two structures, the TSF managed to reflect the actual magnitude and range of the measured values.Figure 6
Comparison between the predicted values of the TSNF-structure prediction model trained by LM with the measured values.


#### 4.3.2. Additional Dataset Verification

To verify the universality of the proposed model, another 40,000 sample points were selected from the data acquired by the GJ-5 track inspection vehicle on the return journey. Table6 compares the performances of the FFNN-, TSNF-, and TSF-structure models on the additional dataset. Figures 7–10 display the best performing model of each structure on the test dataset.Table 6
Performances of the FFNN-, TSNF-, and TSF-structure models on the new dataset.
Method FF TSNF TSF MSE R MAE MSE R MAE MSE R MAE BQ 1.31e-03 -0.37 3.07 6.61e-04 0.2305 0.0213 4.83e-04 0.61 1.86 CGB 1.30e-03 -0.16 2.97 0.0020 0.3609 0.0413 1.06e-04 0.78 0.81 CGF 7.14e-03 -0.25 7.95 0.0029 0.4036 0.0376 9.35e-04 0.95 2.76 CGP 1.01e-03 0.39 2.84 0.0011 0.2284 0.0275 1.53e-04 0.91 1.03 OSS 4.28e-03 0.09 5.63 0.0011 -0.0584 0.0275 1.71e-03 0.80 3.79 RB 2.35e-03 -0.06 3.61 0.0021 0.2364 0.0351 4.50e-04 0.42 1.72 SCG 3.85e-03 0.69 4.36 3.05e-04 0.0131 0.0128 4.06e-04 0.79 1.64 LM 1.02e-03 0.66 2.76 0.0122 -0.3984 0.0864 5.50e-03 0.04 4.52 BR 3.74e-04 0.68 1.27 0.0111 -0.0890 0.0696 1.01e-03 0.84 2.93Figure 7
FFNN-structure model trained by BR.Figure 8
TSNF-structure model trained by SCG.Figure 9
TSF-structure model trained by CGB.Figure 10
TSF-structure model trained by CGP.As shown in Figures7, 8, 9, and 10, the predicted values of FFNN- and TSNF-structure models deviated significantly from the measured results in some intervals. The possible reasons are as follows: First, the target variables may be affected by other implicit variables. Second, the correlation analysis fails to screen out all the representative predictor variables. Before applying the models of these two structures to actual projects, it is necessary to expand the dataset to include various situations and calculate correlation coefficients in a proper manner. These are the necessary steps to acquire the typical influencing factors of the target variables.Besides, the predicted values of the TSF-structure model trained by CGB and CGP were basically consistent with the measured values, which are obviously better than those of the other two structure models. In some stationary phases, however, there was a constant deviation between the predicted and measured values. A possible reason lies in the fact that the predicted values are less affected by other influencing factors in relatively stable phases and are only determined by the impacts from delayed feedback values. By contrast, the constant deviation did not appear in the FFNN-structure model. Thus, the FFNN structure might be helpful to eliminate the deviation in the TSF-structure model. This idea will be examined in future research.Given the accurately predicted vibration acceleration of the car body, the passenger comfort can be derived according to international standards like ISO2631 or UIC513. Thus, the proposed prediction model lays the basis for early warning and fault detection in operation and maintenance processes.


## 4.1. Measurement System

The training data were collected by a GJ-5 track inspection vehicle of ImageMap, Inc., between Shenzhen and Guangzhou, two first-tier cities in China. The verification data were acquired on the return section. The main test items include geometry inspection items, on-board dynamics test items, and ground dynamics test items. Specifically, the track geometry inspections include different wavelengths and gauge, track pitch variation rate, level, triangle pit twisting curvature, and curvature change rate; on-board dynamics test items mainly include wheel rail force and left and right wheel vertical forces, lateral force, derailment coefficient, deceleration rate, three-section acceleration reduction rate of wheel load, lateral stability index of the structure, stability and the vehicle body vertical and lateral acceleration of the left and right axle box, frame, and body of the vehicle left and right axle boxes, vertical and lateral acceleration of the frame, car body vertical and lateral acceleration of the left and right axle boxes, frames, and bodywork of the middle car; ground dynamics test items include derailment coefficient, load shedding rate, lateral force, vertical force, and vertical rail displacement horizontal. The main parameters of track inspection and test accuracy of the inspection vehicle are listed in Table1.Table 1
Main parameters of track inspection and test accuracy of the inspection vehicle.
Test items Accuracy Range Gauge ± 0.8 mm 1420~1480 mm Rail direction (left & right,λ≤30m) ± 1.0 mm ± 100 mm Rail direction (left & right,λ≤50m) ± 2.0 mm ± 100 mm Height (left & right,λ≤30m) ± 1.0 mm ± 100 mm Height (left & right,λ≤50m) ± 2.0 mm ± 100 mm Horizon ± 1.5 mm ± 50 mm Super elevation ± 1.5 mm ± 50 mm Triangle pit ± 1.5 mm ± 220 mm Curve 1.2 × 10 - 4 m-1 −− Body acceleration ± 0.01 g ± 1 g Rail bottom slope ± 0.25 deg −− Rail cross section (left & right) ± 0.5 mm −−Through the correlation analysis, the predictor variables for car body vibration acceleration were selected by the absolute values of coefficients falling between 0.1 and 0.9. This interval was chosen to exclude parameters loosely correlated with or similar to the target variable, making it possible to obtain a practical analysis of the impact of each track factor on the target variable. The selected variables are listed in Table2.Table 2
The prediction variables selected through correlation analysis.
Car body_HACCEL (g) Car body_VACCEL (g) Car body_LACCEL (g) ρ P ρ S  ρ P ρ S  ρ P ρ S Superelev (mm) -0.50 -0.25 L Surf 1 (mm) -0.23 -0.20 Mileage (km) -0.49 -0.49 Curve (rad/km) -0.14 -0.18 R Surf 1 (mm) -0.27 -0.22 Gauge (mm) 0.16 0.11 C_IRREG_LEFT (mm) 0.31 0.19 L Surf 2 (mm) -0.29 -0.24 Superelev (mm) 0.26 0.24 C_IRREG_RIGHT (mm) 0.30 0.18 R Surf 2 (mm) -0.32 -0.26 Curve (rad/km) 0.29 0.26 CURVE_RATE (rad/km/m) -0.15 -0.11 L Surf 3 (mm) -0.31 -0.25 Speed(kph) -0.48 -0.60 LOFFSET (mm) -0.22 0.05 R Surf 3 (mm) -0.35 -0.29 CURVE_RATE (rad/km/m) -0.11 -0.09 ROFFSET (mm) -0.27 0.04 L Surf 4 (mm) -0.17 -0.12 LOFFSET (mm) 0.25 0.20 LSURFACE (mm) -0.43 -0.31 R Surf 4 (mm) -0.21 -0.16 ROFFSET (mm) 0.23 0.20 RSURFACE (mm) 0.39 0.26 - - - LSURFACE (mm) -0.18 -0.15 YAW (deg/s) 0.12 0.14 - - - YAW (deg/s) -0.21 -0.19 - - - - - - BOGIE_FRAME_HACCEL (g) 0.43 0.44 - - - - - - BOGIE_FRAME_VACCEL (g) 0.41 0.42 Total number 10 Total number 8 Total number 12As shown in Table2, it is clear that the vertical vibration acceleration of the car body directly hinges on the surface conditions of the track. The horizontal and lateral acceleration are affected by relatively more factors, owing to the track state and train operation. Moreover, there are some repeated variables, which is not out of expectation. Through the above processes, the main influencing factors of car body vibration acceleration were all identified, laying the basis for subsequent improvement of train structure and passenger comfort. Finally, the repeated variables were eliminated, leaving a total of 23 predictor variables.


## 4.2. Performance Evaluation Indices

Three indices were selected to evaluate the performance of our prediction model: the mean square error (MSE), mean absolute error (MAE), and regression coefficient (R). The MSE served as the main index and the other two as the auxiliary indices.The MSE can be calculated by the equation below:(20)MSE=1N∑i=1Nei2=1N∑i=1Ny^i-yi2where y^i is a vector of N predictions; yi is the vector of observed values corresponding to the inputs; and ei is the square of the errors. When the MSE is close to zero, it means the model is suitable for prediction when it is not overfitted.The MAE is a yardstick of the accuracy of evaluation and prediction [25, 26]. The definition of the MAE is as follows.(21)MAE=1N∑i=1Nyi-y^iThe R indicates the amount of variance explained by the prediction model. This index can be expressed as(22)R=∑i=1Nyi^-y^i-yi-y-i∑i=1Nyi^-y^i-2∑i=1Nyi-y-i2where y-i is the mean value of the measured data; y^i- is the mean value of the predicted data; ∑i=1N(yi-y^i)2is the residual sum of squares; and ∑i=1N(yi-y-i)2 is the explained sum of squares. The value of R falls between 0 and 1. If the R is close to one, it means the model has explained the majority of the variance [27, 28].


## 4.3. Model Performance

Our prediction models were built on the Matlab software with some codes in the neural network toolbox. The program runs on a Lenovo workstation (CPU: Intel® Xeon® Processor E5-2623 v3; 3.00GHz; RMB: 64GB). To compare the models based on the said three neural network structures, the number of hidden layers and the number of hidden layer nodes were set to 1 and 15, respectively. The number of hidden layer nodes was determined because the FFNN structure can achieve the optimal performance with 15 hidden layer nodes through 10~20 traversal iterations.


### 4.3.1. Training Performance

As mentioned before, the modelling data were captured at the interval of 0.25m by a GJ-5 track inspection vehicle moving from Shenzhen to Guangzhou. In total, 50,000 sample points which are continuous in time series were selected for modelling.eX replaces 10X. It can be seen that the models trained with LM and BR outperformed those trained by other algorithms, but the run times of the two models were relatively long.For better understanding, the predicted values of the FFNN-structure model trained by BR were compared with the measured values in Figure4, where the horizontal axis is the total number of samples and the vertical axis is the target variable (i.e., the car body vibration acceleration in three directions).Figure 4
Comparison between the predicted values of the FFNN-structure prediction model trained by BR with the measured values.As shown in Figure4, the predicted acceleration in all three directions basically conformed to the trend of the target variables. However, only the predicted acceleration in the horizontal direction was entirely consistent with the measured value, while that in the other two directions merely approximated the mean value of the target variables. The magnitude and range of the measured values were not well reflected.The performance of the TSNF-structure prediction model is presented in Table4. It can be seen that the three indices were all improved from the levels in Table 3, indicating that it is meaningful to consider the time delays. This is because the car body has some time delays in its response to the relevant factors, such as track surface and driver’s operation. Nonetheless, it is also learned that the TSNF increased the run time from that of the FFNN.Table 3
Performance of the FFNN-structure prediction model.
Method Set MSE R MAE Running time (s) BQ Training 1.67e-04 0.21 0.0095 4.04 Validation 1.88e-04 -0.10 0.0105 Test 5.43e-04 0.21 0.0191 CGB Training 5.29e-05 0.75 0.0053 2.21 Validation 4.81e-05 0.58 0.0052 Test 1.15e-04 0.71 0.0085 CGF Training 1.42e-04 0.25 0.0089 1.52 Validation 1.94e-04 0.24 0.0112 Test 3.71e-04 0.38 0.0163 CGP Training 1.01e-04 0.52 0.0077 1.81 Validation 8.24e-05 0.34 0.0068 Test 3.38e-04 0.33 0.0136 OSS Training 5.79e-04 -0.22 0.0196 1.68 Validation 4.04e-04 -0.44 0.0174 Test 5.77e-04 -0.43 0.0208 RB Training 2.91e-05 0.89 0.0040 1.77 Validation 2.52e-05 0.80 0.0038 Test 7.86e-05 0.71 0.0057 SCG Training 1.32e-04 0.43 0.0088 1.09 Validation 1.31e-04 0.53 0.0094 Test 7.86e-05 0.30 0.0213 LM Training 2.61e-05 0.92 0.0038 16.20 Validation 4.91e-05 0.87 0.0053 Test 2.21e-04 0.86 0.0134 BR Training 1.77e-05 0.94 0.0031 289.73 Test 2.38e-05 0.78 0.0033Table 4
Performances of the TSNF-structure prediction model.
Method MSE R MAE Running time (s) BQ 1.51e-05 0.94 0.0029 2.01e+03 CGB 1.40e-05 0.95 0.0028 244.87 CGF 1.44e-05 0.94 0.0028 252.16 CGP 1.41e-05 0.95 0.0028 237.03 OSS 1.81e-05 0.92 0.0032 299.56 RB 2.03e-05 0.91 0.0034 27.15 SCG 1.41e-05 0.95 0.0028 119.79 LM 1.19e-05 0.96 0.0026 5.64e+04 BR 1.20e-05 0.96 0.0026 8.76e+03The predicted values of the TSNF-structure model trained by BR were compared with the measured values in Figure5. It is clear that the TSNF-structure model reflected the magnitude and range of the measured values more accurately than the FFNN-structure model.Figure 5
Comparison between the predicted values of the TSNF-structure prediction model trained by BR with the measured values.Finally, the performance of the TSF-structure model (Table5) shows that the model performance changed little with the training algorithms. The order of magnitude of the MSE was always at 10-6; the R value remained near 0.99. Therefore, the TSF structure can bless the model with a good performance with a short run time.Table 5
Performance of the TSF-structure prediction model.
Method MSE R MAE Running time (s) CGB 2.27e-06 0.9898 0.0012 257.82 CGF 2.29e-06 0.9897 0.0012 263.01 CGP 2.29e-06 0.9897 0.0012 258.24 OSS 3.18e-06 0.9876 0.0014 321.94 BQ 2.54e-06 0.9890 0.0012 1.91e+03 RB 7.67e-06 0.9618 0.0021 36.37 SCG 2.39e-06 0.9895 0.0012 81.06 LM 2.02e-06 0.9910 0.0011 5.26e+04 BR 2.05e-06 0.9908 0.0011 1.14e+04The predicted values of the TSF-structure model trained by LM were compared with the measured values in Figure6. It can be seen that the predicted values were highly consistent with the measured values. Compared to the other two structures, the TSF managed to reflect the actual magnitude and range of the measured values.Figure 6
Comparison between the predicted values of the TSNF-structure prediction model trained by LM with the measured values.


### 4.3.2. Additional Dataset Verification

To verify the universality of the proposed model, another 40,000 sample points were selected from the data acquired by the GJ-5 track inspection vehicle on the return journey. Table6 compares the performances of the FFNN-, TSNF-, and TSF-structure models on the additional dataset. Figures 7–10 display the best performing model of each structure on the test dataset.Table 6
Performances of the FFNN-, TSNF-, and TSF-structure models on the new dataset.
Method FF TSNF TSF MSE R MAE MSE R MAE MSE R MAE BQ 1.31e-03 -0.37 3.07 6.61e-04 0.2305 0.0213 4.83e-04 0.61 1.86 CGB 1.30e-03 -0.16 2.97 0.0020 0.3609 0.0413 1.06e-04 0.78 0.81 CGF 7.14e-03 -0.25 7.95 0.0029 0.4036 0.0376 9.35e-04 0.95 2.76 CGP 1.01e-03 0.39 2.84 0.0011 0.2284 0.0275 1.53e-04 0.91 1.03 OSS 4.28e-03 0.09 5.63 0.0011 -0.0584 0.0275 1.71e-03 0.80 3.79 RB 2.35e-03 -0.06 3.61 0.0021 0.2364 0.0351 4.50e-04 0.42 1.72 SCG 3.85e-03 0.69 4.36 3.05e-04 0.0131 0.0128 4.06e-04 0.79 1.64 LM 1.02e-03 0.66 2.76 0.0122 -0.3984 0.0864 5.50e-03 0.04 4.52 BR 3.74e-04 0.68 1.27 0.0111 -0.0890 0.0696 1.01e-03 0.84 2.93Figure 7
FFNN-structure model trained by BR.Figure 8
TSNF-structure model trained by SCG.Figure 9
TSF-structure model trained by CGB.Figure 10
TSF-structure model trained by CGP.As shown in Figures7, 8, 9, and 10, the predicted values of FFNN- and TSNF-structure models deviated significantly from the measured results in some intervals. The possible reasons are as follows: First, the target variables may be affected by other implicit variables. Second, the correlation analysis fails to screen out all the representative predictor variables. Before applying the models of these two structures to actual projects, it is necessary to expand the dataset to include various situations and calculate correlation coefficients in a proper manner. These are the necessary steps to acquire the typical influencing factors of the target variables.Besides, the predicted values of the TSF-structure model trained by CGB and CGP were basically consistent with the measured values, which are obviously better than those of the other two structure models. In some stationary phases, however, there was a constant deviation between the predicted and measured values. A possible reason lies in the fact that the predicted values are less affected by other influencing factors in relatively stable phases and are only determined by the impacts from delayed feedback values. By contrast, the constant deviation did not appear in the FFNN-structure model. Thus, the FFNN structure might be helpful to eliminate the deviation in the TSF-structure model. This idea will be examined in future research.Given the accurately predicted vibration acceleration of the car body, the passenger comfort can be derived according to international standards like ISO2631 or UIC513. Thus, the proposed prediction model lays the basis for early warning and fault detection in operation and maintenance processes.


## 4.3.1. Training Performance

As mentioned before, the modelling data were captured at the interval of 0.25m by a GJ-5 track inspection vehicle moving from Shenzhen to Guangzhou. In total, 50,000 sample points which are continuous in time series were selected for modelling.eX replaces 10X. It can be seen that the models trained with LM and BR outperformed those trained by other algorithms, but the run times of the two models were relatively long.For better understanding, the predicted values of the FFNN-structure model trained by BR were compared with the measured values in Figure4, where the horizontal axis is the total number of samples and the vertical axis is the target variable (i.e., the car body vibration acceleration in three directions).Figure 4
Comparison between the predicted values of the FFNN-structure prediction model trained by BR with the measured values.As shown in Figure4, the predicted acceleration in all three directions basically conformed to the trend of the target variables. However, only the predicted acceleration in the horizontal direction was entirely consistent with the measured value, while that in the other two directions merely approximated the mean value of the target variables. The magnitude and range of the measured values were not well reflected.The performance of the TSNF-structure prediction model is presented in Table4. It can be seen that the three indices were all improved from the levels in Table 3, indicating that it is meaningful to consider the time delays. This is because the car body has some time delays in its response to the relevant factors, such as track surface and driver’s operation. Nonetheless, it is also learned that the TSNF increased the run time from that of the FFNN.Table 3
Performance of the FFNN-structure prediction model.
Method Set MSE R MAE Running time (s) BQ Training 1.67e-04 0.21 0.0095 4.04 Validation 1.88e-04 -0.10 0.0105 Test 5.43e-04 0.21 0.0191 CGB Training 5.29e-05 0.75 0.0053 2.21 Validation 4.81e-05 0.58 0.0052 Test 1.15e-04 0.71 0.0085 CGF Training 1.42e-04 0.25 0.0089 1.52 Validation 1.94e-04 0.24 0.0112 Test 3.71e-04 0.38 0.0163 CGP Training 1.01e-04 0.52 0.0077 1.81 Validation 8.24e-05 0.34 0.0068 Test 3.38e-04 0.33 0.0136 OSS Training 5.79e-04 -0.22 0.0196 1.68 Validation 4.04e-04 -0.44 0.0174 Test 5.77e-04 -0.43 0.0208 RB Training 2.91e-05 0.89 0.0040 1.77 Validation 2.52e-05 0.80 0.0038 Test 7.86e-05 0.71 0.0057 SCG Training 1.32e-04 0.43 0.0088 1.09 Validation 1.31e-04 0.53 0.0094 Test 7.86e-05 0.30 0.0213 LM Training 2.61e-05 0.92 0.0038 16.20 Validation 4.91e-05 0.87 0.0053 Test 2.21e-04 0.86 0.0134 BR Training 1.77e-05 0.94 0.0031 289.73 Test 2.38e-05 0.78 0.0033Table 4
Performances of the TSNF-structure prediction model.
Method MSE R MAE Running time (s) BQ 1.51e-05 0.94 0.0029 2.01e+03 CGB 1.40e-05 0.95 0.0028 244.87 CGF 1.44e-05 0.94 0.0028 252.16 CGP 1.41e-05 0.95 0.0028 237.03 OSS 1.81e-05 0.92 0.0032 299.56 RB 2.03e-05 0.91 0.0034 27.15 SCG 1.41e-05 0.95 0.0028 119.79 LM 1.19e-05 0.96 0.0026 5.64e+04 BR 1.20e-05 0.96 0.0026 8.76e+03The predicted values of the TSNF-structure model trained by BR were compared with the measured values in Figure5. It is clear that the TSNF-structure model reflected the magnitude and range of the measured values more accurately than the FFNN-structure model.Figure 5
Comparison between the predicted values of the TSNF-structure prediction model trained by BR with the measured values.Finally, the performance of the TSF-structure model (Table5) shows that the model performance changed little with the training algorithms. The order of magnitude of the MSE was always at 10-6; the R value remained near 0.99. Therefore, the TSF structure can bless the model with a good performance with a short run time.Table 5
Performance of the TSF-structure prediction model.
Method MSE R MAE Running time (s) CGB 2.27e-06 0.9898 0.0012 257.82 CGF 2.29e-06 0.9897 0.0012 263.01 CGP 2.29e-06 0.9897 0.0012 258.24 OSS 3.18e-06 0.9876 0.0014 321.94 BQ 2.54e-06 0.9890 0.0012 1.91e+03 RB 7.67e-06 0.9618 0.0021 36.37 SCG 2.39e-06 0.9895 0.0012 81.06 LM 2.02e-06 0.9910 0.0011 5.26e+04 BR 2.05e-06 0.9908 0.0011 1.14e+04The predicted values of the TSF-structure model trained by LM were compared with the measured values in Figure6. It can be seen that the predicted values were highly consistent with the measured values. Compared to the other two structures, the TSF managed to reflect the actual magnitude and range of the measured values.Figure 6
Comparison between the predicted values of the TSNF-structure prediction model trained by LM with the measured values.


## 4.3.2. Additional Dataset Verification

To verify the universality of the proposed model, another 40,000 sample points were selected from the data acquired by the GJ-5 track inspection vehicle on the return journey. Table6 compares the performances of the FFNN-, TSNF-, and TSF-structure models on the additional dataset. Figures 7–10 display the best performing model of each structure on the test dataset.Table 6
Performances of the FFNN-, TSNF-, and TSF-structure models on the new dataset.
Method FF TSNF TSF MSE R MAE MSE R MAE MSE R MAE BQ 1.31e-03 -0.37 3.07 6.61e-04 0.2305 0.0213 4.83e-04 0.61 1.86 CGB 1.30e-03 -0.16 2.97 0.0020 0.3609 0.0413 1.06e-04 0.78 0.81 CGF 7.14e-03 -0.25 7.95 0.0029 0.4036 0.0376 9.35e-04 0.95 2.76 CGP 1.01e-03 0.39 2.84 0.0011 0.2284 0.0275 1.53e-04 0.91 1.03 OSS 4.28e-03 0.09 5.63 0.0011 -0.0584 0.0275 1.71e-03 0.80 3.79 RB 2.35e-03 -0.06 3.61 0.0021 0.2364 0.0351 4.50e-04 0.42 1.72 SCG 3.85e-03 0.69 4.36 3.05e-04 0.0131 0.0128 4.06e-04 0.79 1.64 LM 1.02e-03 0.66 2.76 0.0122 -0.3984 0.0864 5.50e-03 0.04 4.52 BR 3.74e-04 0.68 1.27 0.0111 -0.0890 0.0696 1.01e-03 0.84 2.93Figure 7
FFNN-structure model trained by BR.Figure 8
TSNF-structure model trained by SCG.Figure 9
TSF-structure model trained by CGB.Figure 10
TSF-structure model trained by CGP.As shown in Figures7, 8, 9, and 10, the predicted values of FFNN- and TSNF-structure models deviated significantly from the measured results in some intervals. The possible reasons are as follows: First, the target variables may be affected by other implicit variables. Second, the correlation analysis fails to screen out all the representative predictor variables. Before applying the models of these two structures to actual projects, it is necessary to expand the dataset to include various situations and calculate correlation coefficients in a proper manner. These are the necessary steps to acquire the typical influencing factors of the target variables.Besides, the predicted values of the TSF-structure model trained by CGB and CGP were basically consistent with the measured values, which are obviously better than those of the other two structure models. In some stationary phases, however, there was a constant deviation between the predicted and measured values. A possible reason lies in the fact that the predicted values are less affected by other influencing factors in relatively stable phases and are only determined by the impacts from delayed feedback values. By contrast, the constant deviation did not appear in the FFNN-structure model. Thus, the FFNN structure might be helpful to eliminate the deviation in the TSF-structure model. This idea will be examined in future research.Given the accurately predicted vibration acceleration of the car body, the passenger comfort can be derived according to international standards like ISO2631 or UIC513. Thus, the proposed prediction model lays the basis for early warning and fault detection in operation and maintenance processes.


## 5. Conclusions

This paper establishes a prediction model for car body vibration acceleration. In the beginning, the various parameters related to the track and the train were filtered by correlation analysis based on both Pearson’ correlation coefficient and Spearman’s rank correlation coefficient. The parameters closely correlated with the target variable were selected as predictor variables. Then, the selected variables were used to construct prediction models with three different neural network structures, namely, the FFNN, the TSNF, and the TSF. To verify the performance, the proposed models were applied to predict the car body vibration acceleration with actual railway datasets. The following phenomena were observed from the predicted results.During the training process and new prediction, according to the obtained values of indices, the BR training algorithm achieved very good performances both on the training dataset and the new dataset under FFNN and TSNF structures, but it consumed too much time. The LM boasted the best performance under the TSF structure but performed poorly on the new test dataset. The TSF-structure models trained by CGB and CGP achieved even more accurate prediction on the new dataset.The future research will further improve the adaptability of the proposed model and apply the predicted values to enhance passenger comfort.



---

*Source: 1752070-2018-12-27.xml*