Cybernetics and Information Technologies · 2024 · 20 citations · 29 references
ComputingComputer ScienceCybersecurityArtificial IntelligenceMachine LearningNetwork SecurityAnomaly DetectionExplainable Machine LearningFeature SelectionIntrusion DetectionOptimization For Machine LearningRepresentation LearningClass ImbalanceExplainable Artificial IntelligenceFeature Selection For Intrusion DetectionGradient BoostingHybrid Feature SelectionLight Gradient Boosting MachineMachine Learning For Intrusion DetectionShapley Value Explanations
Abstract An Intrusion Detection System (IDS) is essential to identify cyber-attacks and implement appropriate measures for each risk. The efficiency of the Machine Learning (ML) techniques is compromised in the presence of irrelevant features and class imbalance. In this research, an efficient data pre-processing strategy was proposed to enhance the model’s generalizability. The class dissimilarity is addressed using k-Means SMOTE. After this, we furnish a hybrid feature selection method that combines filters and wrappers. Further, a hyperparameter-tuned Light Gradient Boosting Machine (LGBM) is analyzed by varying the optimal feature subsets. The experiments used the datasets – UNSW-NB15 and CICIDS-2017, yielding an accuracy of 90.71% and 99.98%, respectively. As the transparency and generalizability of the model depend significantly on understanding each component of the prediction, we employed the eXplainable Artificial Intelligence (XAI) method, SHapley Additive exPlanation (SHAP), to improve the comprehension of forecasted results.
29
LightGBM: A Highly Efficient Gradient Boosting Decision Tree
Guolin Ke, Qi Meng, Thomas Finley et al. · HAL (Le Centre pour la Communication Scientifique Directe) · 2017 · 9.5K citations · Full text
Floating search methods in feature selection
Pavel Pudil, Jana Novovičová, Josef Kittler · Pattern Recognition Letters · 1994 · 3.1K citations