A modular machine learning system for flow-level traffic classification in large networks

Yu Jin, Nick Duffield, Jeffrey Erman, Patrick Haffner, Subhabrata Sen, Zhi Li Zhang

Research output: Contribution to journalArticlepeer-review

59 Scopus citations

Abstract

The ability to accurately and scalably classify network traffic is of critical importance to a wide range of management tasks of large networks, such as tier-1 ISP networks and global enterprise networks. Guided by the practical constraints and requirements of traffic classification in large networks, in this article, we explore the design of an accurate and scalable machine learning based flow-level traffic classification system, which is trained on a dataset of flow-level data that has been annotated with application protocol labels by a packet-level classifier. Our system employs a lightweight modular architecture, which combines a series of simple linear binary classifiers, each of which can be efficiently implemented and trained on vast amounts of flow data in parallel, and embraces three key innovative mechanisms, weighted threshold sampling, logistic calibration, and intelligent data partitioning, to achieve scalability while attaining high accuracy. Evaluations using real traffic data from multiple locations in a large ISP show that our system accurately reproduces the labels of the packet level classifier when runs on (unlabeled) flow records, while meeting the scalability and stability requirements of large ISP networks. Using training and test datasets that are two months apart and collected from two different locations, the flow error rates are only 3% for TCP flows and 0.4% for UDP flows. We further show that such error rates can be reduced by combining the information of spatial distributions of flows, or collective traffic statistics, during classification. We propose a novel two-step model, which seamlessly integrates these collective traffic statistics into the existing traffic classification system. Experimental results display performance improvement on all traffic classes and an overall error rate reduction by 15%. In addition to a high accuracy, at runtime, our implementation easily scales to classify traffic on 10Gbps links.

Original languageEnglish (US)
Article number4
JournalACM Transactions on Knowledge Discovery from Data
Volume6
Issue number1
DOIs
StatePublished - Mar 2012

Keywords

  • Algorithms
  • Measurement
  • Theory

Fingerprint Dive into the research topics of 'A modular machine learning system for flow-level traffic classification in large networks'. Together they form a unique fingerprint.

Cite this