Abstract:Malicious traffic detection is important to defend network attacks. Traditional machine learning-based malicious traffic detection methods aim to improve detection performance and ignore resource consumption. For Internet of Things devices with constrained resources, reducing resource consumption in traffic detection while ensuring the detection accuracy is challenging. In order to solve this problem, this paper proposed an effective feature selection algorithm based on chi-squared test (EFS-CST), which selects the most relevant features to train machine learning (ML) models like random forest, naive bayes, decision tree, and convolutional neural networks. The proposed algorithm was evaluated in terms of accuracy, precision, F1 score, AUROC (area under the receiver operating characteristic curve), AUC-PR (area under the precision-recall curve), model size, training time, and dataset size. Results proved that ML models with EFS-CST could obtain similar detection performance compared to ML models with all features on both UNSW-NBls and TII-SSRC-23 datasets. The detection performance of some ML models with EFS-CST could even outperform that of ML models without feature selection. The dataset sizes for TII-SSRC-23 and UNSW-NB15 were reduced by 48.8% and 47.8%, respectively. In addition, the model training time for TII-SSRC-23 and UNSW-NB15 datasets can be reduced by up to 86.1 % and 48.9%, respectively. Finally, the model sizes for TII-SSRC-23 and UNSW-NB15 datasets could be reduced by up to 60.0% and 65.6 %” respectively.

Shafiq, M and Yu, X and Bashir, AK and Chaudhry, HN and Wang, D (2018)A Machine Learning Approach for Feature Selection Traffic Classification Using Security

A Machine Learning Approach for Feature Selection Traffic Classification Using Security Analysis

Feature selection for optimizing traffic classification

Effective Feature Selection for 5G IM Applications Traffic Classification.

Robust Feature Selection for IM Applications at Early Stage Traffic Classification Using Machine Learning Algorithms

A New Feature Selection Method For Internet Traffic Classification Using Ml

An Efficient Traffic Classification Scheme Using Embedded Feature Selection and LightGBM

Optimizing Traffic Classification Using Hybrid Feature Selection

Efficient and Robust Feature Extraction and Selection for Traffic Classification.

A Systematic Framework for Malicious Traffic Detection Based on Feature Repository.

An Effective Feature Selection Algorithm for Machine Learning-based Malicious Traffic Detection

Optimizing the Performance of Machine Learning Based Traffic Classification

Machine Learned Real-Time Traffic Classifiers

A Hybrid Feature Selection Algorithm Based on Mutual Information and Memetic Framework to Optimize Traffic Classification

IoT Malicious Traffic Identification Using Wrapper-Based Feature Selection Mechanisms

Optimizing Feature Selection for Efficient Encrypted Traffic Classification: A Systematic Approach

Balanced Feature Selection Method for Internet Traffic Classification

A Statistical-Feature-based Approach to Internet Traffic Classification Using Machine Learning.

Network Intrusion Detection Through Discriminative Feature Selection by Using Sparse Logistic Regression

Network Traffic Classification Techniques and Comparative Analysis Using Machine Learning Algorithms