为物联网设备设计轻量级加密流量分类模型,兼顾精度与低资源消耗。
Hardware-Aware Neural Architecture Search for Encrypted Traffic Classification on Resource-Constrained Devices
- 通过硬件感知的神经架构搜索,定制1D-CNN以适配资源受限设备。
- 仅用88.26K参数达96.60%准确率,内存与计算开销降低超300倍。
- 可在STM32上部署,支持短会话与隐私保护场景,适合嵌入式部署。
本文提出一种面向资源受限物联网与边缘设备的硬件高效深度神经网络(DNN),通过硬件感知神经架构搜索(HW-NAS)优化,实现会话级加密流量分类。基于ISCX VPN-nonVPN数据集,该1D卷积神经网络在严格内存与计算限制下保持高性能,参数仅88.26K,浮点运算量10.08M,最大张量尺寸20.12K,准确率达96.60%。相比现有最优模型,参数、计算量和张量大小分别减少444倍、312倍和15倍。在多个VPN及流量分类任务中最高达99.86%准确率,外部基准测试中在USTC-TFC与QUIC NetFlow上分别达到99.98%。头信息预处理研究显示模型对多种配置具备鲁棒性,会话长度缩短75%可显著提升效率,准确率仅下降1-2%。但不当预处理或过激截断可能导致7%准确率损失。量化模型已在STM32微控制器上部署,验证了短会话带来的实际低延迟推理优势,证明其在受限物联网网络中加密流量分析的可行性。
原文摘要 · Abstract (English)
This paper presents a hardware-efficient deep neural network (DNN), optimized through hardware-aware neural architecture search (HW-NAS); the DNN supports the classification of session-level encrypted traffic on resource-constrained Internet of Things (IoT) and edge devices. Thanks to HW-NAS, a 1D convolutional neural network (CNN) is tailored on the ISCX VPN-nonVPN dataset to meet strict memory and computational limits while achieving robust performance. The optimized model attains 96.60% accuracy with just 88.26K parameters, 10.08M FLOPs, and a maximum tensor size of 20.12K. Compared to state-of-the-art models, it achieves reductions of up to 444-fold, 312-fold, and 15-fold in these metrics, respectively, minimizing memory footprint and runtime requirements. The model also achieves up to 99.86% across multiple VPN and traffic classification (TC) tasks; it further generalizes to external benchmarks with up to 99.98% accuracy on USTC-TFC and QUIC NetFlow. In addition, an in-depth study of header-level preprocessing confirms that the optimized model can provide performance across a wide range of configurations, even in scenarios with stricter privacy considerations. Likewise, a reduction in the length of sessions of up to 75% yields significant improvements in efficiency, while maintaining high accuracy with only a negligible drop of 1-2%. However, the importance of careful preprocessing and session length selection in the classification of raw traffic data is still present, as improper settings or aggressive reductions can cause a 7% reduction in accuracy. The quantized architecture was deployed on STM32 microcontrollers and evaluated across input sizes; results confirm that the efficiency gains from shorter sessions translate to practical, low-latency embedded inference. These findings demonstrate the method's practicality for encrypted traffic analysis in constrained IoT networks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。