# Dataset Optimization Strategies for MalwareTraffic Detection

arXiv: https://arxiv.org/abs/2009.11347 · HTML: https://mlsanity.com/paper/2009.11347/

## Key facts

- Cited by 4 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 23 Sep 2020.

## Built on by (most cited first, 4 of 4)

- [A Stock Trading System for a Medium Volatile Asset using Multi Layer Perceptron](https://mlsanity.com/paper/2201.12286/) (2022)
- [DNN-ForwardTesting: A New Trading Strategy Validation using Statistical Timeseries Analysis and Deep Neural Networks](https://mlsanity.com/paper/2210.11532/) (2022)
- [A Comparative Analysis of Statistical and Machine Learning Models for Outlier Detection in Bitcoin Limit Order Books](https://mlsanity.com/paper/2507.14960/) (2025)
- [A Framework for Predictive Directional Trading Based on Volatility and Causal Inference](https://mlsanity.com/paper/2507.09347/) (2025)

## Abstract (arXiv)

Machine learning is rapidly becoming one of the most important technology for malware traffic detection, since the continuous evolution of malware requires a constant adaptation and the ability to generalize. However, network traffic datasets are usually oversized and contain redundant and irrelevant information, and this may dramatically increase the computational cost and decrease the accuracy of most classifiers, with the risk to introduce further noise. We propose two novel dataset optimization strategies which exploit and combine several state-of-the-art approaches in order to achieve an effective optimization of the network traffic datasets used to train malware detectors. The first approach is a feature selection technique based on mutual information measures and sensibility enhancement. The second is a dimensional reduction technique based autoencoders. Both these approaches have been experimentally applied on the MTA-KDD'19 dataset, and the optimized results evaluated and compared using a Multi Layer Perceptron as machine learning model for malware detection.