Lightweight GenAI for Network Traffic Generation: Fidelity, Augmentation, and Classification
Organizations: DIETI, University of Naples Federico II, Italy. · Huawei Technologies France SASU.
Abstract
Network Traffic Classification (NTC) increasingly relies on data-driven models, yet its practical deployment is often constrained by limited labeled data, strict privacy requirements, and the cost of collecting representative traffic traces. While Network Traffic Generation (NTG) provides an effective means to mitigate data scarcity, conventional generative methods struggle to model the complex temporal dynamics of modern traffic and often incur high computational costs. In this article, we investigate lightweight Generative Artificial Intelligence (GenAI) architectures for practical NTG. Rather than generating raw packet bytes or relying on large foundation models, we synthesize compact flow-level traffic representations derived from early packet-header information, enabling transformer-based, state-space, and diffusion models with only a few million parameters. We present a modular GenAI pipeline for NTG and evaluate it along four complementary axes: (i) synthetic traffic fidelity, (ii) synthetic-only training for privacy-preserving NTC, (iii) data augmentation under low-data regimes, and (iv) computational efficiency. Experiments on two heterogeneous datasets show that lightweight transformer-based and state-space models preserve both static and temporal traffic characteristics, while providing useful synthetic data for downstream NTC. Among them, transformer-based models offer the best fidelity-efficiency trade-off, combining high-quality traffic generation with moderate computational overhead.
Figures & tables
| Model | HS | IS | #L | #AH | - | - - | Repo |
| 500/250 | 20 | 6 | – | 1.01M + 0 | 1.09M + 0 | – | |
| 32 | – | 4 | 4 | 1.02M + 5k | 1.03M + 10k | ||
| 72 | 144 | 4 | – | 966k + 427k | 966k + 432k | ||
| 160 | 320 | 4 | 8 | 1.03M + 948k | 1.03M + 961k |
| GenAI Approach | - | - - | ||
| RF | 2D-CNN | RF | 2D-CNN | |
| 66.90\text{\tiny\pm0.35} | 45.17\text{\tiny\pm1.73} | 75.27\text{\tiny\pm0.19} | 63.58\text{\tiny\pm0.24} | |
| 46.57\text{\tiny\pm0.60} | 43.35\text{\tiny\pm1.68} | 61.68\text{\tiny\pm0.23} | 63.36\text{\tiny\pm0.37} | |
| 76.80\text{\tiny\pm0.14} | 60.60\text{\tiny\pm1.38} | 85.77\text{\tiny\pm0.16} | 76.45\text{\tiny\pm0.32} | |
| 78.83 0.19 | 62.70 1.73 | 87.18 0.21 | 78.42 0.41 | |
| w/ rect | 77.14\text{\tiny\pm0.17} | 66.36\text{\tiny\pm0.42} | 85.75\text{\tiny\pm0.10} | 78.44\text{\tiny\pm0.29} |
| Model | Training (GPU) | Generation (RPi 4B) | Model Size | ||||
| Time s/epoch | GPU | Mem MB | Time s/sample | CPU | Mem MB | on Disk MB | |
| 22.57 | 18 | 379 | 0.07 | 30 | 402 | 3.9 | |
| 117.03 | 21 | 387 | 105.67 | 61 | 339 | 4.2 | |
| 108.48 | 15 | 377 | 10.18 | 47 | 534 | 15.5 | |
| 36.81 | 20 | 393 | 2.83 | 44 | 499 | 7.9 | |
| same as | 3.48 | 45 | 509 | 3.5 | |||
| Giampaolo Bovenzi ([email protected]) is an Assistant Professor at the University of Napoli Federico II. His research concerns (anonymized and encrypted) traffic classification and network security. |
| Domenico Ciuonzo [SM] ([email protected]) is an Associate Professor at the University of Napoli Federico II. His research concerns data fusion, network analytics, IoT, signal processing, and AI. |
| Jonatan Krolikowski ([email protected]) is a senior research engineer at the DataCom Lab of Huawei’s Paris Research Center. His research interests include ML- and operations research-driven optimization of real-world networks and the modeling and analysis of network-related problems, with a recent focus on time series modeling. |
| Antonio Montieri ([email protected]) is an Assistant Professor at the University of Napoli Federico II. His research concerns network measurements, traffic classification, modeling and prediction, and AI for networks. |
| Alfredo Nascita ([email protected]) is an Assistant Professor at the University of Napoli Federico II. His research interests include Internet network traffic analysis, machine and deep learning, and explainable AI. |
| Antonio Pescapé [SM] ([email protected]) is a Full Professor at the University of Napoli Federico II. His work focuses on measurement, monitoring, and analysis of the Internet. |
| Dario Rossi [SM] ([email protected]) is network AI CTO and director of the DataCom Lab at Huawei Technologies, France. He has coauthored 15+ patents and over 200+ papers in leading conferences and journals, and has received 9 best paper awards, a Google Faculty Research Award (2015), and an IRTF Applied Network Research Prize (2016). |