Deep Learning and Statistical Testing Approaches for Cyberattack Classification in Imbalanced Datasets
摘要
This study aims to address the critical challenge of class imbalance in cyberattack data using advanced deep learning techniques. Imbalance in datasets can lead to biased model predictions, disproportionately favoring the majority class while underperforming on minority attack types. To counter this, we applied Bayesian Neural Networks (BNN), Naive Bayes, and Long Short-Term Memory (LSTM) models, achieving an accuracy of 88% with the LSTM model, the highest out of 10 iterations. Both BNN and LSTM demonstrated their efficacy in cyberattack classification. To mitigate the effects of class imbalance, we applied mathematical sampling techniques, ensuring a more equitable representation of classes. The Kolmogorov-Smirnov (KS) test was employed to evaluate the models’ performance and understand the data distribution. Visual representations of each class versus others, with p-values from the KS test, were used to highlight distributional disparities, supporting the statistical analysis and enhancing dataset understanding.