arrow
Return

TrafficMAE: A network traffic classification model based on masked autoencoder

delete2025-11-13
delete0
delete
OA
AI
W
Wenjian Zhu *
H
Haiming Zhang
C
Changhua Pei
J
J. C. Li
DOI:10.1016/j.icte.2025.11.004delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Network traffic classification is critical for security and performance. Encrypted protocols challenge traditional methods, and supervised approaches suffer from limited labeled data. We propose TrafficMAE, a masked autoencoder using a Session Video representation to encode temporal and hierarchical traffic information. Our self-supervised pre-training combines masked reconstruction with bidirectional packet direction prediction to learn from unlabeled data. Fine-tuning fuses statistical and learned features. Evaluations on five tasks show TrafficMAE outperforms existing methods with superior F1-scores, strong few-shot performance, and better efficiency compared to transformer-based baselines.
Keywords:
Network traffic classification
Masked autoencoder
Encrypted traffic analysis
Deep learning
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

ICT Express cover
ICT Express
IF:
4.2
Papers:
990
Citations:
2.5K

Organization

C
Computer Network Information Center
Scholars:
70
Papers: 34
Citations: 581