Traffic Flow Anomaly Detection Based on Robust Ridge Regression with Particle Swarm Optimization Algorithm

Abstract EN

Traffic flow anomaly detection is helpful to improve the efficiency and reliability of detecting fault behavior and the overall effectiveness of the traffic operation.

The data detected by the traffic flow sensor contains a lot of noise due to equipment failure, environmental interference, and other factors.

In the case of large traffic flow data noises, a traffic flow anomaly detection method based on robust ridge regression with particle swarm optimization (PSO) algorithm is proposed.

Feature sets containing historical characteristics with a strong linear correlation and statistical characteristics using the optimal sliding window are constructed.

Then by providing the feature sets inputs to the PSO-Huber-Ridge model and the model outputs the traffic flow.

The Huber loss function is recommended to reduce noise interference in the traffic flow.

The L2 regular term of the ridge regression is employed to reduce the degree of overfitting of the model training.

A fitness function is constructed, which can balance the relative size between the k-fold cross-validation root mean square error and the k-fold cross-validation average absolute error with the control parameter η to improve the optimization efficiency of the optimization algorithm and the generalization ability of the proposed model.

The hyperparameters of the robust ridge regression forecast model are optimized by the PSO algorithm to obtain the optimal hyperparameters.

The traffic flow data set is used to train and validate the proposed model.

Compared with other optimization methods, the proposed model has the lowest RMSE, MAE, and MAPE.

Finally, the traffic flow that forecasted by the proposed model is used to perform anomaly detection.

The abnormality of the error between the forecasted value and the actual value is detected by the abnormal traffic flow threshold based on the sliding window.

The experimental results verify the validity of the proposed anomaly detection model.