PNetGPT: Proprietary Protocol Network Traffic Generation with Pre-trained Transformer
Zedong Li, Shijie Li, Dongliang Fang, Xin Chen, Zhanwei Song, Zhi Li, Shichao Lv, Limin Sun
Abstract
Generative pre-trained transformers are exceedingly effective as generative models and classifiers, widely used in natural language processing and computer vision. This work contributes to the exploration of generative pre-trained transformer-based models in the proprietary protocol network traffic. However, building a pre-trained model for proprietary protocol network traffic is non-trivial due to the heterogeneous unknown formats and the extreme scarcity of proprietary protocol network traffic datasets. In this paper, we present PNetGPT, a pre-trained transformer-based model for generating proprietary protocol network traffic. We have constructed the inaugural dataset of 2 real-world proprietary protocols. After training on this dataset, PNetGPT possesses the capacity to generate high-quality proprietary protocol network traffic to support various applications of proprietary protocols, including reverse analysis, protocol fuzzy testing, intrusion detection, etc. We evaluated PNetGPT with two real proprietary protocols and demonstrated state-of-the-art (SOTA) performance in handling heterogeneous unknown formats. The code and datasets are available at: https://github.com/Snail1502/PNetGPT
BibTeX
@inproceedings{icassp2025_pnetgptproprieta,
title = {PNetGPT: Proprietary Protocol Network Traffic Generation with Pre-trained Transformer},
author = {Zedong Li and Shijie Li and Dongliang Fang and Xin Chen and Zhanwei Song and Zhi Li and Shichao Lv and Limin Sun},
booktitle = {ICASSP 2025},
year = {2025}
}