不分词、不预训练,直接用字节流分类网络流量,效果媲美大模型。
MambaNetBurst: Direct Byte-level Network Traffic Classification without Tokenization or Pretraining

- 直接处理原始字节流,跳过分词和预训练步骤
- 在6个公开数据集上表现优于或持平大模型基线
- 适合资源受限场景的高效流量分析,无需复杂预处理
我们提出MambaNetBurst,一种基于Mamba-2骨干的紧凑型无分词字节级序列分类器,用于网络突发流量分类。与多数近期强性能的流量分类与入侵检测方法不同,该方法直接作用于原始包字节,避免分词、分块及复杂的多模态表示,并无需自监督预训练阶段。给定一个数据流,我们从前几包中构建固定长度的突发数据,将字节序列嵌入并附加可学习的CLS标记,通过堆叠残差预归一化Mamba-2模块实现端到端监督分类。在涵盖加密移动应用识别、VPN/Tor流量分类、恶意软件流量分类及物联网攻击流量的六个公开基准上,MambaNetBurst始终表现强劲,与更庞大且常需预训练的基线模型相当甚至超越。消融实验表明,保持字节级时间分辨率至关重要,早期通过步幅下采样始终有害,中等状态大小已足以实现鲁棒泛化。进一步验证了尽管相比Mamba-1结构更受限,但Mamba-2在包字节建模中仍具高有效性,并在训练速度上展现明显效率优势。总体而言,结果表明:使用紧凑选择性状态空间模型进行直接、未稀释的字节到分类学习,是高效、可部署的流量分析新方向,无需绕过高度优化的线性注意力架构的预训练复杂流程。
原文摘要 · Abstract (English)
We present MambaNetBurst, a compact tokenizer-free byte-level sequence classifier for network burst classification based on a Mamba-2 backbone. In contrast to most recent strong traffic-classification and intrusion-detection approaches, our method operates directly on raw packet bytes, avoids tokenization, patching, and heavy engineered multimodal representations, and does not require any self-supervised pre-training stage. Given a packet flow, we form a fixed-length burst from the first few packets, embed the resulting byte sequence appending a learnable CLS token, and process it with a stack of residual pre-normalized Mamba-2 blocks for end-to-end supervised classification. Across six public benchmarks spanning encrypted mobile app identification, VPN/Tor traffic classification, malware traffic classification, and IoT attack traffic, MambaNetBurst achieves consistently strong results and is competitive with, or outperforms, substantially heavier and often pre-trained baselines. Our ablation study shows that preserving byte-level temporal resolution is critical, that early downsampling through striding is consistently harmful, and that moderate state sizes are sufficient for robust generalization. We further show that Mamba-2, despite its more constrained transition structure relative to Mamba-1, remains highly effective for packet-byte modeling while providing clear efficiency advantages, particularly in training speed. Overall, our results demonstrate that direct **undiluted** byte-to-classification learning with compact selective state space models is a practical, effective and novel direction for efficient, deployable traffic analysis that bypasses the complexity of pre-training pipelines even over highly optimized linear attention architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。