English

Data Augmentation for Traffic Classification

Machine Learning 2024-01-24 v2 Networking and Internet Architecture

Abstract

Data Augmentation (DA) -- enriching training data by adding synthetic samples -- is a technique widely adopted in Computer Vision (CV) and Natural Language Processing (NLP) tasks to improve models performance. Yet, DA has struggled to gain traction in networking contexts, particularly in Traffic Classification (TC) tasks. In this work, we fulfill this gap by benchmarking 18 augmentation functions applied to 3 TC datasets using packet time series as input representation and considering a variety of training conditions. Our results show that (i) DA can reap benefits previously unexplored, (ii) augmentations acting on time series sequence order and masking are better suited for TC than amplitude augmentations and (iii) basic models latent space analysis can help understanding the positive/negative effects of augmentations on classification performance.

Keywords

Cite

@article{arxiv.2401.10754,
  title  = {Data Augmentation for Traffic Classification},
  author = {Chao Wang and Alessandro Finamore and Pietro Michiardi and Massimo Gallo and Dario Rossi},
  journal= {arXiv preprint arXiv:2401.10754},
  year   = {2024}
}

Comments

to appear at Passive and Active Measurements (PAM), 2024