FR-MTL: Traffic Flow Prediction Using Fused Ridge Denoising and Multi-Task Learning

抄録

Traffic flow prediction is one of the fundamental components in Intelligent Transportation Systems (ITS). Many traffic flow prediction models have been developed, but with limitation of noise sensitivity, which will result in poor generalization. Fused Lasso, also known as total variation denoising, penalizes L₁-norm on the model coefficients and pairwise differences between neighboring coefficients, has been widely used to analyze highly correlated features with a natural order, as is the case with traffic flow. It denoises data by encouraging both sparsity of coefficients and their differences, and estimates the coefficients of highly correlated variables to be equal to each other. However, for traffic data, the same coefficients will lead to overexpression of features, and losing the trend of time series of traffic flow. In this work, we propose a Fused Ridge multi-task learning (FR-MTL) model for multi-road traffic flow prediction. It introduces Fused Ridge for traffic data denoising, imposes penalty on L₂-norm of the coefficients and their differences. The penalty of L₂-norm proportionally shrinks coefficients, and generates smooth coefficient vectors with non-sparsity. It has both capability of trend preservation and denoising. In addition, we jointly consider multi-task learning (MTL) for training shared spatiotemporal information among traffic roads. The experiments on real traffic data show the advantages of the proposed model over other four regularized baseline models, and on traffic data with Gaussian noise and missing data, the FR-MTL model demonstrates potential and promising capability with satisfying accuracy and effectiveness.