category: literaturenote citekey: dengrealtimenetworktraffic2025 title: "Real-Time Network Traffic Forecasting with Missing Data: A Generative Model Approach" authors: "Deng, Lei; Xu, Wenhan; Li, Jingwei; Tsang, Danny H. K." year: 2025 date: 2025-00-00 2025 url: "https://arxiv.org/abs/2506.09647v1" zotero_key: 3ZT8EJ7T zotero_storage: ECPNBMXB collections: transpordi_planeerimine folder: 001_artiklid firstAuthor: "Deng, Lei"
LEI DENG, WENHAN XU, JINGWEI LI, and DANNY H.K. TSANG∗ , The Hong Kong University of Science and Technology (Guangzhou), China
Real-time network traffic forecasting is crucial for network management and early resource allocation. Existing network traffic forecasting approaches operate under the assumption that the network traffic data is fully observed. However, in practical scenarios, the collected data are often incomplete due to various human and natural factors. In this paper, we propose a generative model approach for real-time network traffic forecasting with missing data. Firstly, we model the network traffic forecasting task as a tensor completion problem. Secondly, we incorporate a pre-trained generative model to achieve the low-rank structure commonly associated with tensor completion. The generative model effectively captures the intrinsic low-rank structure of network traffic data during pre-training and enables the mapping from a compact latent representation to the tensor space. Thirdly, rather than directly optimizing the high-dimensional tensor, we optimize its latent representation, which simplifies the optimization process and enables real-time forecasting. We also establish a theoretical recovery guarantee that quantifies the error bound of the proposed approach. Experiments on real-world datasets demonstrate that our approach achieves accurate network traffic forecasting within 100 ms, with a mean absolute error (MAE) below 0.002, as validated on the Abilene dataset.
CCS Concepts: • Networks → Network performance evaluation.
Additional Key Words and Phrases: Network traffic forecasting, gradient descent, tensor completion, generative model
Lei Deng, Wenhan XU, Jingwei Li, and Danny H.K. Tsang. 2025. Real-Time Network Traffic Forecasting with Missing Data: A Generative Model Approach. J. ACM 37, 4, Article 111 (August 2025), 14 pages.https://doi.org/XXXXXXX.XXXXXXX
The widespread adoption of smart devices has significantly accelerated the global expansion of the Internet, leading to a sharp increase in both network traffic and application demands. This rapid growth has amplified the complexity of the network and the volume of data that require continuous monitoring and management [10, 43]. In the progress toward intelligent networks, one of the crucial challenges is real-time network traffic forecasting. This task involves estimating future traffic volumes based on historical data to prevent congestion proactively and maintain high network performance [4]. Real-time forecasting plays a vital role in enabling the network operator to promptly assess user demands, identify anomalous or malicious activities, and implement dynamic resource allocation strategies.
Network traffic forecasting heavily depends on historical data, which provides essential spatiotemporal information for accurate predictions. The quality of this historical data significantly
Authors' Contact Information: Lei Deng, leideng@hkust-gz.edu.cn; Wenhan XU, wxube@connect.ust.hk; Jingwei Li, jingweili@hkust-gz.edu.cn; Danny H.K. Tsang, eetsang@ust.hk, The Hong Kong University of Science and Technology (Guangzhou), Guangzhou, Guangdong, China.
Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from permissions@acm.org.
© 2025 Copyright held by the owner/author(s). Publication rights licensed to ACM.
ACM 1557-735X/2025/8-ART111
https://doi.org/XXXXXXX.XXXXXXX
111:2 Deng et al.
influences the forecasting performance. Unfortunately, in real-world scenarios, such data are often incomplete due to various human and natural factors: (i) to reduce measurement costs, normally only a subset of origin-destination (OD) pairs would be selected for taking traffic measurements [10, 49], resulting in partial observations; (ii) due to system-related issues [44] such as network congestion, node misbehavior, transmission interference, and monitor failure can lead to inevitable data loss [36, 37, 44]. Therefore, achieving real-time and accurate network traffic forecasting in the presence of missing data has become a pressing challenge.
Existing research on network traffic forecasting mainly includes two categories: traditional statistical techniques and deep learning (DL)-based approaches. The majority of statistical models rely on linear methods, such as Autoregressive (AR) and Auto-Regressive Integrated Moving Average (ARIMA) models [29]. While ARIMA is capable of modeling linear and short-range dependencies (SRD), it fails to capture long-range dependencies (LRD), often leading to poor performance in network traffic forecasting [13]. DL methods typically incorporate advanced architectures, including graph neural networks and Long Short-Term Memory (LSTM) models, to effectively learn spatialtemporal correlations within network traffic data, enabling more precise forecasting [31, 33, 35, 45]. However, most of these methods assume the availability of complete historical data, which is rarely the case in real-world deployments. As a result, the above approaches cannot be directly applied for network traffic forecasting with missing data. Although this problem can be addressed by applying data imputation techniques prior to forecasting, this two-step process can distort key latent structures in the data and may result in cumulative forecasting errors. It is desirable to develop a new approach for real-time network traffic forecasting with missing data.
Real-time network traffic forecasting becomes challenging in the presence of missing data. Specifically, network traffic data often exhibits a low-rank structure, reflecting its underlying spatiotemporal correlations [10, 43]. However, missing entries can severely distort this inherent structure [9, 44], making it difficult to effectively leverage low-rank properties during forecasting. This disruption not only complicates the exploitation of spatial-temporal patterns but also slows down the optimization process, thereby hindering the ability to perform real-time forecasting.
With the rapid advancement of generative artificial intelligence [14, 22, 47], generative models have emerged as potential tools for enforcing structural constraints in network traffic data. Notably, pre-trained generative models have demonstrated strong capabilities in recovering sparse signals from limited measurements in the context of compressive sensing [6, 18, 42]. Inspired by these successes, we consider a pre-trained generative model to capture the intrinsic low-rank structure of network traffic data, which can generate a complete traffic tensor from its latent representation. With the help of generative models, it becomes feasible to effectively exploit the underlying low-rank structure of traffic data even in the presence of missing entries, thereby facilitating accurate and real-time traffic forecasting.
In this paper, we propose a Generative Model approach for real-time network traffic forecasting with missing data (GMF). Firstly, we model the task of network traffic forecasting with missing data as a tensor completion (TC) problem [12, 48]. Secondly, we propose a GMF scheme for real-time network traffic forecasting, which employs a pre-trained generative model to relax the low-rank constraint typically associated with traffic tensor completion. The pre-trained generative model allows GMF to learn the low-rank patterns of traffic tensors in the pre-training stage and can map a condensed latent representation to a complete traffic tensor. GMF circumvents the requirement of low-rank tensor decomposition and simplifies the optimization process in TC, leading to fast forecasting.
Our main contributions are summarized as follows:
Existing works on network traffic forecasting mainly include traditional statistical methods and deep learning (DL)-based approaches.
Statistical approaches predominantly utilize linear models for time series forecasting tasks. A widely adopted example is the ARIMA model [29], which has been extensively applied in network traffic prediction. Xu et al. [46] developed a prediction framework based on the Auto-Regressive Moving Average (ARMA) model, which integrates third-party monitoring systems to enhance prediction efficiency while minimizing network resource consumption. Despite their practicality, these linear models are often insufficient for capturing the intricate and nonlinear dynamics of real-world traffic patterns [4]. Furthermore, their reliance on complete historical datasets renders them ineffective in scenarios involving missing or incomplete data.
With the rapid advancement of artificial intelligence, DL has garnered significant attention in the field of network traffic forecasting. Numerous DL-based models have been introduced to achieve efficient and accurate network traffic forecasting[1, 15, 40]. Andreoletti et al. [2] applied a Diffusion Convolutional Recurrent Neural Network (DCRNN) to predict link-level traffic loads within a real-world backbone network. Xie et al. [45] introduced a multi-range, multi-level spatio-temporal learning approach comprising three dedicated aggregation modules, each tailored to capture distinct temporal patterns (recent, daily, and weekly). Despite their promising performance, these DL models are generally built on the assumption of fully available historical traffic data. This reliance on complete datasets significantly limits their applicability in practical scenarios, where network traffic data is normally sparse or incomplete.
Different from existing approaches to network traffic forecasting, our proposed GMF scheme employs a low-rank tensor framework [11, 48] to achieve accurate forecasting in the presence of missing data. At the core of GMF is a pre-trained generative model that enforces the low-rank constraint required for TC. This design eliminates the need for computationally intensive tensor decompositions, thereby supporting real-time forecasting.
111:4 Deng et al.
Input: T ∈ R
𝑛1×𝑛2×𝑛3 .
Output: U, S, V.
1 Te= (T, [ ], 3).;
2 for 𝑘 = 1 : 𝑛3 do
3 [U, S,V] = svd(eT
(𝑘)
), eU
(𝑘) = U,eS
(𝑘) = S, eV
(𝑘) = V.;
4 end
5 U = i(Uf, [ ], 3), S = i(Se, [ ], 3), V = i(Vf, [ ], 3).
We begin by defining the notations and preliminary concepts, and then proceed to describe the system model and problem formulation.
Definition 3.1. (t-product [20]) The t-product between A ∈ R 𝑛1×𝑛2×𝑛3 and B ∈ R 𝑛2×𝑛4×𝑛3 is defined as C ∈ R 𝑛1×𝑛4×𝑛3 = A ∗ B, where C (𝑖, 𝑗, :) = Í𝑛2 ℓ=1 A (𝑖, ℓ, :) ⊙ B (ℓ, 𝑗, :), and ⊙ represent the circular convolution.
Note that the t-product operation is similar to the matrix multiplication where the element-wise multiplication operation is replaced by the circular convolution of two tubes [12]. The t-product can also be regarded as the matrix multiplication in the Fourier domain, i.e., C = A ∗ B is equal to Ce = A△e Be, where △ is the frontal-slice-wise product [24].
Definition 3.2. (Tensor transpose [20]) The transpose of an 𝑛1 × 𝑛2 × 𝑛3 tensor T is a tensor of size 𝑛2 ×𝑛1 ×𝑛3, denoted as T ⊤, which is obtained by transposing every frontal slice of the tensor and then sorting the transposed frontal slices 2 through 𝑛3 in reversed order, i.e., T ⊤(:, :, 1) = (T (: , :, 1))𝐻 and T ⊤(:, :, 𝑛3 + 𝑘 − 2) = (T (:, :, 𝑘))𝐻 for 2 ≤ 𝑘 ≤ 𝑛3.
Now we introduce tensor singular value decomposition (t-SVD) and present its details in Alg. 1.
Definition 3.3. (t-SVD and tensor tubal-rank [20]) The t-SVD of a third-order tensor T ∈ R 𝑛1×𝑛2×𝑛3 is given by T = U∗S∗V⊤. U and V are orthogonal tensors with size 𝑛1 × 𝑛1 × 𝑛3 and 𝑛2 × 𝑛2 × 𝑛3. The tubal-rank of tensor T is defined as the number of non-zero singular tubes of S, while S is an orthogonal tensor.
Consider a network comprising 𝑛 nodes. The end-to-end network traffic volume at a given time interval can be represented by an 𝑛×𝑛 matrix. As the spatial traffic matrix evolves over time, the sequence of traffic matrices forms a third-order tensor, denoted by T ∈ R 𝑛×𝑛×𝑛3 , where 𝑛3 represents the number of time intervals. This traffic tensor captures three dimensions: source nodes, destination nodes, and temporal intervals, as illustrated in Fig. 1. Each entry T𝑖 𝑗𝑘 (1 ≤ 𝑖, 𝑗 ≤ 𝑛, 1 ≤ 𝑘 ≤ 𝑛3) corresponds to the traffic volume between origin node 𝑖 and destination 𝑗 during the 𝑘-th time interval.
Our objective is to achieve real-time and accurate network traffic forecasting in the presence of missing data. Specifically, consider a network consisting of 𝑛 nodes. Given a sequence of historical traffic matrices with missing entries, the task is to accurately predict future traffic matrices in realtime. Let 𝑇ℎ denote the length of the historical sequence (historical length) and 𝑇𝑝 represent the length of the forecasting horizon (forecasting length). By stacking these traffic matrices together along the temporal dimension, we construct a third-order tensor T ∈ R 𝑛×𝑛×𝑛3 , where 𝑛3 = 𝑇ℎ +𝑇𝑝 .
Accordingly, the forecasting task can be formulated as a tensor completion problem, where the goal is to predict the frontal slices (T (𝑇ℎ+1) , T (𝑇ℎ+2) , . . . , T (𝑇ℎ+𝑇𝑝 ) ) in real-time based on incomplete historical observations (T , T , . . . , T (𝑇ℎ ) ). An illustration of this process is provided in Fig. 1.
The low-rank TC problem is normally formulated as [7, 25]:
$$\hat{\mathcal{T}} = \arg\min_{X \in \mathbb{R}^{n \times n \times n3}} |\mathcal{M} - \mathcal{P}{\Omega}(X)|F^2 + \tau |X|{\text{TNN}},\tag{1}$$
Note that kXkTNN represents the tensor nuclear norm (TNN) of X, which constrains that the recovered tensor is low-tuba l-rank. kXkTNN is defined as the sum of the singular values of all the frontal slices of Xe, i.e., ||X||TNN= Í𝑛3 𝑘=1 ||X ( e :, :, 𝑘)||∗. Tˆ is the recovered tensor, and M is measurement tensor (incomplete tensor) and 𝜏 > 0. We use Ω to denote the set of observed elements, |Ω| to denote the number of observed elements, and PΩ to denote the sampling operator, then we can define PΩ(X) = Φ ⊙ X, where ⊙ is the element-wise product (or Hadamard product), Φ is the indicator: Φ(𝑖, 𝑗,𝑘) = ( 1, if (𝑖, 𝑗,𝑘) ∈ Ω, 0, otherwise. and we have M = PΩ(T ).
The solution to (1) can be obtained through tensor singular value thresholding (t-SVT)[7, 25], though this approach requires computationally expensive low-rank decompositions to compute TNN and suffers from a long runtime.
Recent work [6] introduces compressed sensing using generative models (CSGM), which relaxes the traditional sparsity constraint in compressed sensing by leveraging a generative model as a structural prior. This approach motivates our key research question: Can a generative model achieve the tensor low-rank constraint in (1), enabling real-time network traffic tensor completion? To address this question, we have conducted empirical studies to evaluate the capability of generative models in producing low-tubal-rank tensor data (see Appendix A). Now we explain our main idea.
We begin by introducing our main idea of low-tubal-rank TC framework using generative models. Suppose that a tubal-rank 𝑟 tensor T ∈ R 𝑛1×𝑛2×𝑛3 can be represented by a low-dimensional latent representation zˆ ∈ R 𝑙 , 𝑙 ≪ 𝑛1𝑛2𝑛3. A pre-trained generative model (i.e., a generator) 𝐺𝜃 : R 𝑙 →
111:6 Deng et al.
Fig. 1. Tensor model for network traffic data.
Fig. 2. Illustration of the low-tubal-rank TC framework based on genitive models.
R 𝑛1×𝑛2×𝑛3 is then used to map this latent representation zˆ to its corresponding tensor T, such that:
$$\mathcal{T} = G_{\theta}(\hat{\mathbf{z}}). \tag{2}$$
We now provide an intuitive explanation of how Equation (2) operates. Let T = U ∗ S ∗ V⊤ be the t-SVD of T and define the singular vector v = [S (1, 1, :), · · · ,S (𝑟, 𝑟, :)]. This vector v is constructed by stacking all the singular tubes from the diagonal of S into a single long vector.
It is known that these singular tubes encapsulate essential structural information of the tensor and have a significant influence on the performance of TC. Therefore, the singular vector v can be naturally interpreted as a latent representation of T. By integrating the t-SVD formulation with (2), we obtain:
$$\mathcal{T} = \mathcal{U} * \mathcal{S} * \mathcal{V}^{\top} = G_{\theta}(\mathbf{v}), \tag{3}$$
and zˆ ∈ R 𝑙 = v ∈ R 𝑟𝑛3 . In this setting, 𝑙 = 𝑟𝑛3 and the generator implicitly learns the orthogonal tensor components U and V during the pre-training phase, enabling it to map the singular vector v back into the original tensor space. During inference, the latent representation zˆ is optimized to approximate the ground truth singular vector v. An illustration of this process is provided in Fig. 2.
The low-tubal-rank TC framework via generative models can be depicted by the following minimization function
$$\hat{\mathbf{z}} = \arg\min{\mathbf{z}} E{\theta}(\mathcal{M}, \mathbf{z}),$$ (4)
where
$$E{\theta}(\mathcal{M}, \mathbf{z}) = |\mathcal{M} - \mathcal{P}{\Omega}(G{\theta}(\mathbf{z}))|{F}^{2} + \gamma |G{\theta}(\mathbf{z})|{TNN}.$$ (5)
(5) is aligned with (1). Given that TNN is defined as the sum of singular values in the spectral domain, and z represents the potential singular vector of 𝐺𝜃 (z), we can approximate k𝐺𝜃 (z) kTNN using kezk1 in practice, whereez is the spectral version of z, defined byez = [fft(z(1, · · · , 𝑛3)), · · · , fft(z( (𝑟− 1)𝑛3+1, · · · , 𝑟𝑛3))]. The parameter 𝛾 serves to regulate the influence of the low-rank regularization.
(4) can be addressed using a naive gradient descent approach, starting from an initial point z1 ∼ 𝑝z (z):
$$\mathbf{z}_{t+1} \leftarrow \mathbf{z}t - \rho \frac{\partial E{\theta}(\mathcal{M}, \mathbf{z})}{\partial \mathbf{z}} \bigg|_{\mathbf{z} = \mathbf{z}_t},$$ (6)
where 𝜌 is the learning rate.
Unlike the conventional approach that directly optimizes the tensor T in the low-rank completion problem (1), the optimization in (4) is carried out in the latent space of the variable z. We also provide relevant theoretical analysis to reveal the error bound of the proposed framework in Appendix B.
Fig. 3. The framework of the GMF scheme
We now elaborate the technical details of our GMF scheme for real-time network traffic forecasting with missing data, which comprises two key phases: a pre-training phase and an online inference phase.
As presented in Section 4, a pre-trained generative model𝐺𝜃 is required for TC and network traffic forecasting. The quality of𝐺𝜃 highly impacts the forecasting accuracy. Here we first introduce the design and training of the generator 𝐺𝜃 , and then we present a DL-based update rule to speed up the optimization of z.
5.1.1 Architecture Design for the Generative Model. The architecture of the generator 𝐺𝜃 requires deliberate design to capture the low-tubal-rank structures of network traffic tensors. According to (3), we can observe that 𝐺𝜃 needs to learn the orthogonal tensor components U and V with t-product operations. In this setting, tensor layers (TL) [23, 30] become natural choices. TL is a neural network that developed based on t-product. Its forward propagation is defined by A𝑗+1 = 𝜎(W𝑗 ∗ A𝑗 + B𝑗), ∀𝑗 = 1, · · · , 𝑁 − 1, where 𝜎 is an element-wise activation function, 𝑁 is the number of layers in the network, A𝑗 ∈ R ℓ𝑗 ×𝑏×𝑎 is a feature tensor, W𝑗 ∈ R ℓ𝑗+1×ℓ𝑗 ×𝑎 is a trainable weight tensor and B𝑗 ∈ R ℓ𝑗+1×𝑏×𝑎 is a bias tensor.
In a TL, ℓ𝑗 implicitly acts as the tensor tubal-rank. By setting ℓ𝑗 to an appropriate small value, the output of a tensor layer is a low-tubal-rank tensor. Meanwhile, the t-product operation preserves and exploits multidimensional correlations within the data. This is in contrast to standard neural networks, which typically flatten tensors into vectors or matrices, destroying the multi-mode structure and losing access to low-tubal-rank information.
Therefore, we structure the generator 𝐺𝜃 with tensor layers
$$G_{\theta}(\cdot) = \sigma(\text{TL}(\sigma(\text{FC}(\cdot)))),$$ (7)
where FC(·) represents the fully connected layer. We will show that this structure is more powerful than conventional neural networks like pure FC layers in the experiments.
5.1.2 Training for the Generative Model. We now describe the pre-training procedure for the generator 𝐺𝜃 . The goal is to guide the generator to map z into a low-tubal-rank tensor space that approximates the actual network traffic tensor. Since the latent variable z to closely approximate the true singular vector v (as depicted in (3)), a straightforward strategy for training 𝐺𝜃 is to use singular vectors as inputs and the corresponding complete traffic tensors as label, as outlined in Alg. 2. The training loss in Alg. 2 consists of two components: (i) the first term strengthens the
111:8 Deng et al.
Algorithm 2: Pre-Training of 𝐺𝜃
Input: Complete network traffic tensor
T ∈ R
𝑛×𝑛×𝑛3 , max epoch 𝑀𝑎𝑥𝐸𝑝𝑜𝑐ℎ,
parameter 𝛾0
Output: Trained parameters 𝜃 of 𝐺𝜃
1 Initialize: Parameters 𝜃, 𝑒𝑝 = 1;
2 while NOT converged or 𝑒𝑝 ≤ 𝑀𝑎𝑥𝐸𝑝𝑜𝑐ℎ do
3 [U, S, V] = t-SVD(T);
4 v = [S (1, 1, :), · · · , S (𝑛, 𝑛, :) ];
5 loss = k𝐺𝜃 (v) − T k2
𝐹
+ 𝛾0 k𝐺𝜃 (v) kTNN;
6 Update 𝜃 to minimize loss, 𝑒𝑝 = 𝑒𝑝 + 1;
7 end
8 return 𝜃
Algorithm 3: Training of 𝑓𝜙
Input: Complete tensor T, measurements
M = PΩ (T ), latent z, 𝑀𝑎𝑥𝐸𝑝𝑜𝑐ℎ, steps
𝐾, generator 𝐺𝜃
Output: Optimizer parameters 𝜙
1 Initialize: 𝜙, z1 ∼ N (0, 1), 𝜌 = 0.01, 𝑒𝑝 = 1;
2 while NOT converged or 𝑒𝑝 ≤ 𝑀𝑎𝑥𝐸𝑝𝑜𝑐ℎ do
3 for 𝑡 = 1 : 𝐾 do Update z𝑡 by (8);
4 zˆ = z𝐾+1;
5 Compute singular vector v of T via t-SVD;
6 Update 𝜙 to minimize loss in (9), 𝑒𝑝 = 𝑒𝑝 + 1;
7 end
8 return 𝜙
model's capacity to accurately translate singular vectors into the tensor domain; and (ii) the second term enforces a low-tubal-rank structure on the generated tensors by minimizing TNN. While the training routine in Alg. 2 does not employ adversarial learning, it well captures the mapping from singular vectors to the tensor domain.
5.1.3 Latent Optimization Acceleration by DL. Given a well-trained generator 𝐺𝜃 , we can solve (4) by gradient descent (6) for TC and network traffic forecasting. However, standard gradientbased optimization typically requires executing a large number of iterations - often ranging from hundreds to thousands - along with multiple random restarts [5, 6] to obtain a sufficiently good zˆ
Recent studies [3, 8] have demonstrated that the gradient descent process can be facilitated using DL. Motivated by this, we further adopt a DL-based optimizer 𝑓𝜙 , parameterized by 𝜙, to replace the conventional update rule in (6). This substitution avoids multiple iterations and facilitates the optimization. The update of z is then performed as follows:
$$\mathbf{z}{t+1} = f{\phi}(\mathcal{M}, \mathbf{z}_t), \tag{8}$$
where 𝑓𝜙 (M, z𝑡 ) = 𝜎(FC(𝜎(FC(𝜎(FC(Cat(vec(M), z𝑡))))))) is a lightweight neural network. Here, vec(·) is vectorization operation and Cat(·) denotes the vector concatenation operation.
The following loss function is adopted to train 𝑓𝜙 :
$$\operatorname{loss}{f} = \alpha |\hat{\mathbf{z}} - \mathbf{v}|{F}^{2} + \beta |G{\theta}(\hat{\mathbf{z}}) - \mathcal{T}|{F}^{2}, \tag{9}$$
where 𝛼 and 𝛽 are weighting hyperparameters. The first term encourages the optimized latent vector zˆ to approximate the true singular value vector v, while the second term ensures that the generator 𝐺𝜃 reconstructs the target traffic tensor T accurately. The overall training procedure for 𝑓𝜙 is outlined in Alg. 3.
.
With the pre-trained generative model 𝐺𝜃 and the optimizer 𝑓𝜙 in place, GMF can be deployed for real-time network traffic forecasting. The detailed computational procedure of GMF is outlined in Alg. 4. In summary: (i) GMF leverages generative models to capture the low-tubal-rank structures inherent in network traffic data. By directly mapping a latent representation to a complete network traffic tensor, GMF can achieve real-time forecasting. (ii) GMF adopts a tensor-layer-based architecture, enabling efficient learning for low-tubal-rank structures. (iii) To accelerate inference, GMF replaces standard gradient descent with a DL-based update mechanism 𝑓𝜙 .
Input: Observed tensor M ∈ R
𝑛×𝑛×𝑛3 , latent z ∈ R
𝑛𝑛3 , steps 𝐾, generator 𝐺𝜃 , optimizer 𝑓𝜙
Output: Forecast matrices Xˆ
1 Initialize: z1 ∼ N (0, 1);
2 for 𝑡 = 1 : 𝐾 do z𝑡+1 = 𝑓𝜙 (M, z𝑡 ) ; // Latent update via (8)
3 Tˆ = 𝐺𝜃 (z𝐾+1 );
4 return Tˆ (𝑇ℎ+1)
, . . . , Tˆ (𝑛3 )
Remark: Fast inference for larger networks. GMF can provide faster inference compared with the vanilla method, which directly solves (1) by t-SVT [7, 25]. Specifically, given an incomplete network traffic tensor with size 𝑛 ×𝑛 ×𝑛3, t-SVT requires computation cost O (𝑛 2𝑛3 +𝑛 3𝑛3) for TC [26], while our GMF scheme in Alg. 4 requires O (𝑛 2𝑛3 +𝑛𝑛2 3 +𝑛 2𝑛3). Our approach is more scalable for larger networks (i.e., with a larger number of nodes 𝑛).
We test the performance of the proposed method through extensive simulations.
Datasets and preprocessing: We perform experiments to evaluate the efficacy of the proposed GMF in network traffic forecasting using two real-world IP network traffic datasets: the Abilene dataset [28] and the GEANT dataset [38]. The Abilene dataset captures end-to-end tra ´ ffic patterns among 12 network nodes, recorded at 5-minute intervals over 168 days, resulting in 48,384 traffic matrices. An anomalously large value in the Abilene dataset (identified as an outlier) has been replaced with zero to ensure data consistency. The GEANT dataset contains traffic measurements ´ across 23 nodes, collected every 15 minutes over 120 days, resulting in 10,772 traffic matrices.
For each dataset, the initial 80% of consecutive traffic matrices are used for training, while the remaining 20% are reserved for testing. We configure the forecasting task with a historical sequence length of𝑇ℎ = 10 and a forecasting horizon of𝑇𝑝 = 1, resulting in a tensor length of 𝑛3 = 𝑇ℎ+𝑇𝑝 = 11. By applying a sliding window of size 11, we generate multiple third-order tensors T ∈ R 𝑛×𝑛×11 for both training and evaluation, where 𝑛 = 12 for the Abilene dataset and 𝑛 = 23 for the GEANT ´ dataset. To simulate real-world scenarios with missing data, we randomly mask elements in the historical portion of each tensor (i.e., the first 10 frontal slices of size 𝑛×𝑛×10), with missing rates ranging from 0.1 to 0.9. For each generated tensor, we evaluate both the forecasting accuracy and computation time of the final traffic matrix T (𝑛3 )
Baselines: The following forecasting methods are used as baselines for performance comparison. Multi-range Multi-level Spatial-Temporal Learning (M2STL) [45], Uncertainty-Aware Inductive Graph Neural Network (UIGNN) [27], Transformer [39], and LSTM [16]. Since M2STL, Transformer, and LSTM require complete historical data for effective forecasting, we first apply the TNN-ADMM tensor completion algorithm [48] to impute missing values, followed by forecasting. To evaluate forecasting performance, we consider the following standard metrics: Mean Absolute Error (MAE) and Normalized Root Mean Squared Error (NRMSE), which are widely adopted in recent work [45]. All neural network models are implemented using the PyTorch framework, and experiments are conducted on a computer equipped with an NVIDIA GeForce RTX 4090 GPU.
We begin by evaluating the effectiveness of the tensor-layer architecture used in the generator. To this end, we define a variant of GMF composed solely of FC layers, expressed as 𝐺𝜃 (·) =
111:10 Deng et al.
Table 1. Forecasting Performance Comparison (MAE and NRMSE)
| Method | Missing Rate | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Abilene | ´ GEANT |
|||||||||
| 0.1 | 0.3 | 0.5 | 0.7 | 0.9 | 0.1 | 0.3 | 0.5 | 0.7 | 0.9 | |
| MAE | ||||||||||
| GMFTL (Ours) |
0.0014 | 0.0015 0.0015 0.0016 0.00160.0015 0.0015 0.0015 0.0016 0.0016 | ||||||||
| GMFFC | 0.0020 | 0.0020 | 0.0021 | 0.0021 | 0.0022 | 0.0113 | 0.0116 | 0.0122 | 0.0125 | 0.0128 |
| M2STL | 0.0015 | 0.0018 | 0.0019 | 0.0023 | 0.0026 | 0.0016 | 0.0017 | 0.0018 | 0.0019 | 0.0020 |
| UIGNN | 0.0017 | 0.0020 | 0.0023 | 0.0028 | 0.0031 | 0.0065 | 0.0070 | 0.0077 | 0.0089 | 0.0134 |
| LSTM | 0.0013 0.0015 | 0.0016 | 0.0016 | 0.0017 | 0.0024 | 0.0024 | 0.0026 | 0.0027 | 0.0029 | |
| Transformer | 0.0022 | 0.0023 | 0.0025 | 0.0027 | 0.0028 | 0.0039 | 0.0040 | 0.0043 | 0.0044 | 0.0046 |
| NRMSE | ||||||||||
| GMFTL | (Ours)0.0122 0.0123 0.0124 0.0128 | 0.0135 | 0.0045 0.0045 0.0046 0.0047 0.0047 | |||||||
| GMFFC | 0.0129 | 0.0132 | 0.0133 | 0.0136 | 0.0137 | 0.0051 | 0.0052 | 0.0052 | 0.0053 | 0.0054 |
| M2STL | 0.0516 | 0.0529 | 0.0553 | 0.0632 | 0.0783 | 0.0781 | 0.0782 | 0.0783 | 0.0787 | 0.0792 |
| UIGNN | 0.0629 | 0.0713 | 0.0802 | 0.0933 | 0.1130 | 0.1209 | 0.1752 | 0.2221 | 0.2469 | 0.2732 |
| LSTM | 0.0125 | 0.0127 | 0.0128 | 0.0130 | 0.0133 | 0.0046 | 0.0049 | 0.0050 | 0.0053 | 0.0056 |
| Transformer | 0.0135 | 0.0136 | 0.0136 | 0.0137 | 0.0138 | 0.0054 | 0.0056 | 0.0057 | 0.0057 | 0.0059 |
𝛿(FC(𝛿(FC(·)))), and refer to it as GMFFC. The original GMF model utilizing tensor layers is denoted as GMFTL. We compare the forecasting accuracy of GMFTL and GMFFC to assess the impact of the tensor-based design.
Table 1 reports the forecasting accuracy (MAE and NRMSE) across different data missing rates in historical data. It can be observed that GMFTL consistently outperforms GMFFC, which relies solely on FC layers. This performance gap highlights the advantage of incorporating tensor layers in effectively capturing the low-rank structure inherent in network traffic data.
Comparing forecasting errors in Table 1, we observe: (i) GMF consistently outperforms competitors across most missing rates. (ii) On Abilene, LSTM occasionally matches or exceeds GMF, benefiting from TNN-ADMM's effective imputation of complete input data. However, TNN-ADMM's weaker recovery on GEANT limits LSTM's performance there. (iii) UIGNN, designed f ´ or transportation traffic, shows suboptimal performance without domain adaptation. Overall, GMF demonstrates superior robustness for network traffic forecasting.
We evaluate the real-time performance of different methods in Table 2. UIGNN, GMFTL, and GMFFC achieve <100 ms inference times, enabling real-time use. However, UIGNN and GMFFC sacrifice too much accuracy for speed. Traditional methods (M2 STL, LSTM, Transformer) require additional imputation steps (e.g., TNN-ADMM), increasing computation time significantly. GMF provides the optimal balance between speed and accuracy for real-time traffic forecasting.
In this work, we proposed a Generative Model approach for real-time network traffic forecasting with missing data (GMF). GMF formulates the forecasting problem as a tensor completion task and
Dataset [48]+M2STL [48]+Transformer [48]+LSTM UIGNN GMFFC GMFTL Abilene > 2 s > 1 s > 1 s < 100 ms < 100 ms < 100 ms GEANT ´ > 5 s > 3 s > 3 s < 100 ms < 100 ms < 100 ms
Table 2. Computational Time Comparison.
leverages a generative model as a structural surrogate for conventional tensor low-rank assumptions. The model enables the generation of complete network traffic tensors from compact latent representations - thereby streamlining the forecasting process. Additionally, we adopted a tensorlayer architecture in the generative model, which better captures the low-tubal-rank structures in network traffic. Extensive evaluations on real-world network traffic datasets demonstrate that GMF reduces estimation time while preserving high accuracy.
111:12 Deng et al.
1–6.
[18] Ajil Jalal, Liu Liu, Alexandros G Dimakis, and Constantine Caramanis. 2020. Robust compressed sensing using generative models. Advances in Neural Information Processing Systems 33 (2020), 713–727.
[40] Ravi Vinayakumar, KP Soman, and Prabaharan Poornachandran. 2017. Applying deep learning approaches for network traffic prediction. In 2017 International Conference on Advances in Computing, Communications and Informatics (ICACCI). IEEE, 2353–2358.
[41] Lingzhen Wang, Lei Deng, Xinxin Feng, and Haifeng Zheng. 2021. Robust Low-Tubal-Rank Tensor Completion for Network Latency Estimation and Anomaly Detection. In 2021 13th International Conference on Wireless Communications and Signal Processing (WCSP). IEEE, 1–5.
[42] Yan Wu, Mihaela Rosca, and Timothy Lillicrap. 2019. Deep compressed sensing. In International Conference on Machine Learning. PMLR, 6850–6860.
[43] Kun Xie, Yudian Ouyang, Xin Wang, Gaogang Xie, Kenli Li, Wei Liang, Jiannong Cao, and Jigang Wen. 2023. Deep Adversarial Tensor Completion for Accurate Network Traffic Measurement. IEEE/ACM Transactions on Networking (2023).
[44] Kun Xie, Lele Wang, Xin Wang, Gaogang Xie, Jigang Wen, Guangxing Zhang, Jiannong Cao, and Dafang Zhang. 2018. Accurate recovery of Internet traffic data: A sequential tensor completion approach. IEEE/ACM Transactions on Networking 26, 2 (2018), 793–806.
[45] Ruotian Xie, Jigang Wen, Xiaodi Chen, Kun Xie, Wei Liang, Naixue Xiong, Dafang Zhang, Gaogang Xie, and Kenli Li. 2024. M2STL: Multi-Range Multi-Level Spatial-Temporal Learning Model for Network Traffic Prediction. IEEE Transactions on Network Science and Engineering (2024).
[46] Shuona Xu and Biqing Zeng. 2014. Network traffic prediction model based on auto-regressive moving average. Journal of Networks 9, 3 (2014), 653.
[47] Shuo Yang, Minjing Dong, Yunhe Wang, and Chang Xu. 2023. Adversarial Recurrent Time Series Imputation. IEEE Transactions on Neural Networks and Learning Systems 34, 4 (2023), 1639–1650. doi:10.1109/TNNLS.2020.3010524
[48] Zemin Zhang and Shuchin Aeron. 2016. Exact tensor completion using t-SVD. IEEE Transactions on Signal Processing 65, 6 (2016), 1511–1526.
[49] Rui Zhu, Bang Liu, Di Niu, Zongpeng Li, and Hong Vicky Zhao. 2016. Network latency estimation for personal devices: A matrix completion approach. IEEE/ACM Transactions on Networking 25, 2 (2016), 724–737.
The tensor low-rank properties of network traffic data have been widely discussed [10, 44]. To verify if the generative models can capture the tensor low-rank constraint in network traffic tensors, we investigate the low-tubal-rank properties of outputs produced by such models.
Generative artificial intelligence represents a transformative advancement in modern machine learning and has attracted widespread academic and industrial attention [8, 14, 32]. Recent investigations have shown the potential of generative models to effectively handle non-convex sparse vector structures in the context of compressed sensing [6, 42]. Inspired by these developments, we examine whether generative models can also capture the non-convex low-rank structures that are commonly found in tensors.
We begin by evaluating the low-tubal-rank properties of tensors produced by well-established generative models. Specifically, we focus on two representative architectures: (i) Deep Convolutional Generative Adversarial Networks (DCGAN) [34], which take a Gaussian noise vector z ∈ R 120 and generate an image (tensor) of dimensions 64 × 64 × 3. (ii) Progressive Growing of GAN (PGAN) [19], which takes a Gaussian noise vector z ∈ R 512 and generates an image (tensor) of dimensions 256 × 256 × 3.
For each generative architecture, we generate input samples from a Gaussian distribution and use the output tensors for rank analysis. This experiment is repeated 10 times to obtain a collection of 10 tensors, from which we compute the t-SVD and the cumulative distribution functions (CDFs) of their singular tube energy. The energy of each singular tube in S bt t-SVD is defined as its ℓ2 norm. The results are presented in Fig. 4, which displays the energy distribution of singular tubes for both DCGAN- and PGAN-generated tensors. Remarkably, both models exhibit strong low-tubal-rank tendencies. For instance, in the case of DCGAN, the top two singular values alone account for over 94% of the total energy.
These findings clearly demonstrate the inherent ability of generative models to represent and preserve tensor low-tubal-rank structures. Encouraged by this observation, we are motivated to apply such generative models to the task of low-tubal-rank TC and network traffic forecasting.
111:14 Deng et al.
Fig. 4. Energy CDFs of singular tube of tensors generated by generative models.
We present our main theoretical results for the generative model-based low-tubal-rank TC framework (i.e., Section 4), which are developed using compressive sensing techniques. Before proceeding, we introduce several essential definitions and notational conventions. Following the terminology in [6, 18], for two functions 𝑓1 (𝑛) and 𝑓2 (𝑛), we write 𝑓1 (𝑛) = Υ(𝑓2 (𝑛)) to indicate the existence of a positive constant𝑐3 and a natural number 𝑛1 such that for all 𝑛 ≥ 𝑛1, the inequality |𝑓1 (𝑛)| ≥ 𝑐3 𝑓1 (𝑛) holds. We use 𝑎 ∼ N (𝜇, 𝜎2 ) to denote that the variable 𝑎 satisfies a Gaussian distribution. For convenient discussion, we assume 𝑛1 = 𝑛2 = 𝑛. We also define a 𝑙 as the vectorization of the lateral slice of tensor A ∈ R 𝑛×𝑛×𝑛3 , i.e., a 𝑙 = [A(1) (:,𝑙), · · · , A(𝑛3 ) (:,𝑙)]⊤, 𝑙 ∈ [𝑛].
Lemma B.1. Let T ∈ R 𝑛×𝑛×𝑛3 be a tensor with tubal-rank 𝑟. Then, there exists a matrix A such that the sampled tensor M = PΩ(T ) can be expressed in vectorized form as:
$$\mathbf{m}^l = \mathbf{A}\mathbf{g}^l, \quad \forall \ l \in [n],$$
where each vector g 𝑙 is 𝑟𝑛3-sparse1
The proof of Lemma B.1 can be derived by extending Theorem 2 in [10] to the t-product case.
Definition B.2. (Sampling Variance Condition) Let T ∈ R 𝑛×𝑛×𝑛3 be a tensor, and let Ω denote the set of observed entries. The sampling operation is given byM = PΩ(T ), which can be expressed in vectorized form as: m𝑙 = Ag𝑙 , ∀ 𝑙 ∈ [𝑛], for some matrix A. If the entries of A are independently sampled from a Gaussian distribution, i.e., A(𝑖, 𝑗) ∼ N (0, 𝜎2 ), then we say that the measurement tensor M satisfies the sampling variance condition with parameter 𝜎
Given Definition B.2, we provide the following lemma:
Lemma B.3. Let T ∈ R 𝑛×𝑛×𝑛3 be a rank-𝑟 tensor. Suppose that 𝐺𝜃 : R 𝑟𝑛3 → R 𝑛×𝑛×𝑛3 is a generative model from a 𝑑-layer neural network using ReLU activations, and we observe a subset of elements (denoted by Ω) of T uniformly at random. Let |Ω| = O (𝑟𝑛3𝑑 log(𝑛 2𝑛3)) and let the corresponding measurement tensor satisfy the variance condition with parameter 1 |Ω| . For any observations M = PΩ(T ), there exists a 𝛾 ≥ 0 such that if zˆ minimizes (5) to within additive 𝜖 of the optimum, then with 1 − 𝑒 −Υ( |Ω| ) probability,
$$|G{\theta}(\hat{\mathbf{z}}) - \mathcal{T}|{F} \le 6 \min_{\mathbf{z}^* \in \mathbb{R}^{rn3}} |G{\theta}(\mathbf{z}^*) - \mathcal{T}|_{F} + 2\epsilon.$$
$$\tag{10}$$
We omit the proof of Lemma B.3 here, which can be derived by extending Theorem 1.1 in [6] from the vector to the tensor space.
1A vector is said to be 𝑠-sparse if it contains at most 𝑠 nonzero entries.