Research on Network Traffic Classification Method Based on Dual-branch Spatial-Temporal Attention
<p>Qian Wang, Tianbo Xu</p> · Academic Journal of Computing & Information Science · 2024
Network traffic classification plays an important role in network monitoring and network management. Deep learning methods have become an effective traffic classification method due to its ability to automatically extract features. In the article, a two-branch spatio-temporal attention-based model is designed to represent bi-directional flows using multiple dimensional uniform packets as samples, 2DCNN with channel and spatial attention to extract spatial features, and LSTM with many-to-one attention mechanism to extract temporal features. The article conducts experiments on the public traffic dataset USTC-TFC2016. The results show that the classification performance of the model is better compared to the ablation experiments.