Utilizing TabNet Deep Learning for Elephant Flow Detection by Analyzing Information in First Packet Headers

Entropy (Basel). 2024 Jun 22;26(7):537. doi: 10.3390/e26070537.

Abstract

Rapid and precise detection of significant data streams within a network is crucial for efficient traffic management. This study leverages the TabNet deep learning architecture to identify large-scale flows, known as elephant flows, by analyzing the information in the 5-tuple fields of the initial packet header. The results demonstrate that employing a TabNet model can accurately identify elephant flows right at the start of the flow and makes it possible to reduce the number of flow table entries by up to 20 times while still effectively managing 80% of the network traffic through individual flow entries. The model was trained and tested on a comprehensive dataset from a campus network, demonstrating its robustness and potential applicability to varied network environments.

Keywords: TabNet; elephant; feature importance; flow table; flows; input information; machine learning; mice; traffic engineering.

Grants and funding

This work was supported by the Polish Ministry of Science and Higher Education with the subvention funds of the Faculty of Computer Science, Electronics and Telecommunications of AGH University.