The invention provides a network traffic matrix prediction method based on multi-scale
convolution and attention collaborative enhancement LSTM (
Long Short Term Memory), and solves the problems that an existing traffic matrix prediction model is insufficient in multi-scale feature capture, weak in space-
time correlation modeling and poor in generalization. A dynamic multi-scale
convolution module (adopting a 3 * 3 / 5 * 5 / 7 * 7 two-dimensional
convolution kernel) is used to extract local fine-grained association and global coarse-grained trend of a two-dimensional traffic matrix, deep fusion of LSTM
time sequence features and two-dimensional multi-scale spatial features is realized in combination with a cross attention mechanism, and time-space cross-dimension association is enhanced. And meanwhile, the problem of gradient disappearance of the deep network is relieved by relying on residual connection, and topology independence is realized based on a flow-by-flow method so as to adapt to traffic matrixes of different dimensions. The method specifically comprises the following steps: performing
stream-by-
stream division and normalization preprocessing on an original two-dimensional traffic matrix, and after
processing by modules such as dynamic multi-scale convolution, cross attention enhancement LSTM, residual block and gating screening and the like, outputting a single OD
stream predicted value and reconstructing a complete traffic matrix; and integrating objective functions of the
encoder and the decoder, taking the MSE as a
loss function, and adopting an Adam optimizer to iteratively update
model parameters until the model converges. Experiments show that the MAE and the MSE of the method on real data sets such as Ailene, GEANT and the like are better than those of mainstream models such as SVR, LSTM, MTGNN and the like, the reconstruction precision of a high-dimensional traffic matrix is particularly remarkable, and a reliable basis can be provided for
network congestion control and resource optimization.