Self-attentive deep learning method for online traffic classification and its interpretability

Xie, Guorui1,2; Li, Qing2,3; Jiang, Yong1,2

2021-09-04

10.1016/j.comnet.2021.108267

Computer Networks   影响因子和分区

1389-1286

1872-7069

Traffic classification is one of the fundamental tasks in computer networking. This task aims to associate network traffic to a specific class according to the requirements (e.g., QoS provisioning). Online classification, which refers to the situations where flows need to be classified in real time, is an essential technique for this topic. In recent academic research, traffic classification methods based on machine learning (ML) or deep learning (DL) have been proposed. However, most of these methods take flow-level data as input, which requires observing the entire or large portion of a flow and violates the restrictions of online classification. Furthermore, the DL-based methods scarcely discuss the interpretability (e.g., which features are learned by DL, where is the discrimination power from). The lack of interpretability makes people question their reliability and may hinder their further applications. In this paper, we propose a self-attentive method (SAM) for traffic classification. We properly design a neural network whose input can be more fine-grained (i.e., packet-level). This neural network outputs classification results (similar to 2 ms per packet) and consequently satisfies the requirements of online classification. Furthermore, a new technique called self-attention mechanism is utilized for interpretability exploration. By assigning attentive weights to different parts of the input, the self-attention mechanism allows us to understand how the DL model learns discriminative features from the input. According to experimental results, SAM outperforms the current state-of-the-art schemes, improving classification accuracy by similar to 8% (protocol classification), similar to 5% (application classification), and similar to 13% (traffic type classification).

Online traffic classification Deep learning Self-attention mechanism Interpretability

SCI ; EI

英语

通讯

National Key Research and Development Program of China[2020YFB1804704] ; National Natural Science Foundation of China[61972189] ; Shenzhen Key Lab of Software Defined Networking[ZDSYS20140509172959989]

Computer Science ; Engineering ; Telecommunications

Computer Science, Hardware & Architecture ; Computer Science, Information Systems ; Engineering, Electrical & Electronic ; Telecommunications

WOS:000699725100027

ELSEVIER

Web of Science

1.Tsinghua Shenzhen Int Grad Sch, Shenzhen 518055, Peoples R China
2.Peng Cheng Lab, Shenzhen 518040, Peoples R China
3.Southern Univ Sci & Technol, Shenzhen 518055, Peoples R China

Xie, Guorui,Li, Qing,Jiang, Yong. Self-attentive deep learning method for online traffic classification and its interpretability[J]. Computer Networks,2021,196.

Xie, Guorui,Li, Qing,&Jiang, Yong.(2021).Self-attentive deep learning method for online traffic classification and its interpretability.Computer Networks,196.

Xie, Guorui,et al."Self-attentive deep learning method for online traffic classification and its interpretability".Computer Networks 196(2021).