cs.NIAug 22, 2026

Pruned Traffic Trees: Native Semantic Compression with a Protocol-Structured Model Family for Encrypted Traffic Classification

Authors: Yuantu Luo, Jun Tao, Xiangyu Xu, Linxiao Yu, Kangying Li

Organizations: School of Cyber Science and Engineering, Southeast University, Nanjing, China · School of Computer Science and Engineering, Southeast University, Nanjing, China

Abstract

Deep learning has achieved strong performance in encrypted traffic classification (ETC), yet its computational cost limits deployment on resource-constrained network devices such as routers and middleboxes. Existing compression methods mainly operate on weights, channels, hidden representations, or predictions, but do not explicitly determine which protocol fields and structural contexts should remain. We propose Pruned Traffic Trees (PTT), a three-level protocol-structured model family that treats native protocol structures as compression units. PTT-Full learns protocol-structured representations and field salience from complete Protocol Tree Graphs (PTGs), with flow-level self-supervised learning and protocol-presence-aware sparse execution. The learned salience and TopK+kk closure construct Distilled PTGs (PTG-Ds) for PTT-Distilled, while PTT-Lite inherits this topology and reduces width through structure-aligned transfer and flow-level logits distillation. Under flow-disjoint and Strong Information Information (SII)-masked settings, PTT-Full achieves Macro-F1 scores of 0.9519 and 0.9416 on CSTNET-TLS1.3 and CipherSpectrum, while PTT-Lite retains 0.9325 and 0.9136 with 80.3% and 61.3% fewer parameters, 98.85% and 98.78% lower effective GFLOPs, and 8.75×\times and 8.46×\times CPU inference speedups. These results demonstrate that treating protocol structure itself as the compression object enables effective performance-efficiency trade-offs for lightweight ETC.

Figures & tables

Explore similar work

CardsList
  1. PACC: Protocol-Aware Cross-Layer Compression for Compact Network Traffic Representation

    Feb 9, 2026Zhaochen Guo, Honghao Wang, Tianyufei Zhou +2Representation LearningEncrypted Network Traffic Analysis

  2. Treat Traffic Like Trees: A Semantic-Preserving Hierarchical Graph-Based Expert Framework for Encrypted Traffic Analysis

    Jun 3, 2026Yuantu Luo, Jun Tao, Linxiao Yu +1Graph Neural NetworksHierarchical GNNs

  3. TrafficMoE: Heterogeneity-aware Mixture of Experts for Encrypted Traffic Classification

    Mar 31, 2026Qing He, Xiaowei Fu, Lei ZhangEncrypted Traffic ClassificationEncrypted Network Traffic Analysis