English

Tiny Neural Networks for Session-Level Traffic Classification

Networking and Internet Architecture 2025-05-13 v2

Abstract

This paper presents a system for session-level traffic classification on endpoint devices, developed using a Hardware-aware Neural Architecture Search (HW-NAS) framework. HW-NAS optimizes Convolutional Neural Network (CNN) architectures by integrating hardware constraints, ensuring efficient deployment on resource-constrained devices. Tested on the ISCX VPN-nonVPN dataset, the method achieves 97.06% accuracy while reducing parameters by over 200 times and FLOPs by nearly 4 times compared to leading models. The proposed model requires up to 15.5 times less RAM and 26.4 times fewer FLOPs than the most hardware-demanding models. This system enhances compatibility across network architectures and ensures efficient deployment on diverse hardware, making it suitable for applications like firewall policy enforcement and traffic monitoring.

Keywords

Cite

@article{arxiv.2504.04008,
  title  = {Tiny Neural Networks for Session-Level Traffic Classification},
  author = {Adel Chehade and Edoardo Ragusa and Paolo Gastaldo and Rodolfo Zunino},
  journal= {arXiv preprint arXiv:2504.04008},
  year   = {2025}
}
R2 v1 2026-06-28T22:47:52.124Z