An Analysis of the Application of Machine Learning in Network Security
DOI:
https://doi.org/10.5281/zenodo.10775504Keywords:
Data Imbalance, SMOTE, GA-XGBoost, Network Data ClassificationAbstract
In order to deal with the problem of imbalance and complex feature relationship in network data classification, this study proposes a machine learning classification method, combined with improved SMOTE technology and genetic algorithm optimization for XGBoost (GA-XGBoost). Sample balance is achieved by introducing local outlier factors in the SMOTE process, oversampling the minority classes, and randomly undersampling the majority classes. Meanwhile, the evolutionary iterative advantage of the genetic algorithm was used to optimize the XGBoost model parameters and improve the model fit. The experimental results on the UNSW _ NB 15 dataset show that the classification prediction accuracy reached 97.40%, average recall of 70.2% and average F1-score of 68.8%, showing a higher performance compared to the traditional machine learning algorithm and SMOTE + XGBoost method. In addition, the tests on the data collected by the industrial information security platform also confirmed the effectiveness of the proposed method, with a classification accuracy of 99%.
Downloads
Metrics
References
Liu, T., Xu, C., Qiao, Y., Jiang, C., & Chen, W. (2024). News Recommendation with Attention Mechanism. Journal of Industrial Engineering and Applied Science, 2(1), 21-26.
Liu, T., Xu, C., Qiao, Y., Jiang, C., & Yu, J. (2024). Particle Filter SLAM for Vehicle Localization. Journal of Industrial Engineering and Applied Science, 2(1), 27-31.
Xu, C., Qiao, Y., Zhou, Z., Ni, F., & Xiong, J. (2024). Accelerating Semi-Asynchronous Federated Learning. arXiv preprint arXiv:2402.10991.
Zhou, Z. (2024, February). ADVANCES IN ARTIFICIAL INTELLIGENCE-DRIVEN COMPUTER VISION: COMPARISON AND ANALYSIS OF SEVERAL VISUALIZATION TOOLS. In The 8th International scientific and practical conference “Priority areas of research in the scientific activity of teachers”(February 27–March 01, 2024) Zagreb, Croatia. International Science Group. 2024. 298 p. (p. 224).
Xu, C., Yu, J., Chen, W., & Xiong, J. (2024, January). DEEP LEARNING IN PHOTOVOLTAIC POWER GENERATION FORECASTING: CNN-LSTM HYBRID NEURAL NETWORK EXPLORATION AND RESEARCH. In The 3rd International scientific and practical conference “Technologies in education in schools and universities”(January 23-26, 2024) Athens, Greece. International Science Group. 2024. 363 p. (p. 295).
Su, J., Jiang, C., Jin, X., Qiao, Y., Xiao, T., Ma, H., ... & Lin, J. (2024). Large Language Models for Forecasting and Anomaly Detection: A Systematic Literature Review. arXiv preprint arXiv:2402.10350.
Qiao, Y., Jin, J., Ni, F., Yu, J., & Chen, W. (2023). Application of machine learning in financial risk early warning and regional prevention and control: A systematic analysis based on shap. WORLD TRENDS, REALITIES AND ACCOMPANYING PROBLEMS OF DEVELOPMENT, 331.
Qiao, Y., Ni, F., Xia, T., Chen, W., & Xiong, J. (2024, January). AUTOMATIC RECOGNITION OF STATIC PHENOMENA IN RETOUCHED IMAGES: A NOVEL APPROACH. In The 1st International scientific and practical conference “Advanced technologies for the implementation of new ideas”(January 09-12, 2024) Brussels, Belgium. International Science Group. 2024. 349 p. (p. 287).
Ni, F., Zang, H., & Qiao, Y. (2024, January). SMARTFIX: LEVERAGING MACHINE LEARNING FOR PROACTIVE EQUIPMENT MAINTENANCE IN INDUSTRY 4.0. In The 2nd International scientific and practical conference “Innovations in education: prospects and challenges of today”(January 16-19, 2024) Sofia, Bulgaria. International Science Group. 2024. 389 p. (p. 313).
Su, J., Nair, S., & Popokh, L. (2022, November). Optimal Resource Allocation in SDN/NFV-Enabled Networks via Deep Reinforcement Learning. In 2022 IEEE Ninth International Conference on Communications and Networking (ComNet) (pp. 1-7). IEEE.
Zhao, Z., Zhang, N., Xiong, J., Feng, M., Jiang, C., & Wang, X. (2024). Enhancing E-commerce Recommendations: Unveiling Insights from Customer Reviews with BERTFusionDNN. Journal of Theory and Practice of Engineering Science, 4(02), 38-44.
Xiong, J., Feng, M., Wang, X., Jiang, C., Zhang, N., & Zhao, Z. (2024). Decoding sentiments: Enhancing covid-19 tweet analysis through bert-rcnn fusion. Journal of Theory and Practice of Engineering Science, 4(01), 86-93.
Bao, W., Che, H., & Zhang, J. (2020, December). Will_Go at SemEval-2020 Task 3: An accurate model for predicting the (graded) effect of context in word similarity based on BERT. In Proceedings of the Fourteenth Workshop on Semantic Evaluation (pp. 301-306).
Su, J., Nair, S., & Popokh, L. (2023, February). EdgeGym: A Reinforcement Learning Environment for Constraint-Aware NFV Resource Allocation. In 2023 IEEE 2nd International Conference on AI in Cybersecurity (ICAIC) (pp. 1-7). IEEE.
Su, J., Nair, S., & Popokh, L. (2022, November). Optimal Resource Allocation in SDN/NFV-Enabled Networks via Deep Reinforcement Learning. In 2022 IEEE Ninth International Conference on Communications and Networking (ComNet) (pp. 1-7). IEEE.
Xiao, X. Y., An, S., Guo, F., Li, X., & Liang, Z. (2023, September). How Optimization Algorithm Can Contribute to Sustainability Development-A Case of Nepal. In Proceedings of the 2023 6th International Conference on Big Data Technologies (pp. 189-194).
Xiao, X. Y., An, S., Guo, F., Li, X., & Lyu, C. (2023, September). Assessing the Impact of Olympic Games: A Study Based on Big Data Mining and Analysis. In Proceedings of the 2023 6th International Conference on Big Data Technologies (pp. 141-145).
Guo, F. (2023, July). A Study of Smart Grid Program Optimization Based on K-Mean Algorithm. In 2023 3rd International Conference on Electrical Engineering and Mechatronics Technology (ICEEMT) (pp. 711-714). IEEE.
Guo, F., Wu, J. Z., & Pan, L. (2023, July). An Empirical Study of AI Model’s Performance for Electricity Load Forecasting with Extreme Weather Conditions. In International Conference on Science of Cyber Security (pp. 193-204). Cham: Springer Nature Switzerland.
Bian, J., Wang, L., Yang, K., Shen, C., & Xu, J. (2023). Accelerating Hybrid Federated Learning Convergence under Partial Participation. arXiv preprint arXiv:2304.05397.
Wang, H., Bian, J., & Xu, J. (2023). On the Local Cache Update Rules in Streaming Federated Learning. arXiv preprint arXiv:2303.16340.
Bian, J., Shen, C., & Xu, J. (2023, March). Federated Learning via Indirect Server-Client Communications. In 2023 57th Annual Conference on Information Sciences and Systems (CISS) (pp. 1-5). IEEE.
Bian, J., & Xu, J. (2022). Mobility improves the convergence of asynchronous federated learning. arXiv preprint arXiv:2206.04742.
Bian, J., Fu, Z., & Xu, J. (2021). FedSEAL: Semi-supervised federated learning with self-ensemble learning and negative learning. arXiv preprint arXiv:2110.07829.
Xu, J., & Sen, S. (2021). Decision Intelligence for Nationwide Ventilator Allocation During the COVID-19 Pandemic. SN Computer Science, 2(6), 423.
Deng, Y., Kesselman, C., Sen, S., & Xu, J. (2019, December). Computational operations research exchange (core): A cyber-infrastructure for analytics. In 2019 Winter Simulation Conference (WSC) (pp. 3447-3456). IEEE.
Xiaobin, L., Changlin, Y., & Wanting, W. (2023). CARASSISTNET: DESIGN AND IMPLEMENTATION OF A DRIVING ASSISTANCE SYSTEM USING COMPUTER VISION. Ответственный редактор, 28.
Yuan, C., Liu, X., & Zhang, Z. (2021, May). The Current Status and progress of Adversarial Examples Attacks. In 2021 International Conference on Communications, Information System and Computer Engineering (CISCE) (pp. 707-711). IEEE.
Xiao, T., Zeng, L., Shi, X., Zhu, X., & Wu, G. (2022, September). Dual-Graph Learning Convolutional Networks for Interpretable Alzheimer’s Disease Diagnosis. In International Conference on Medical Image Computing and Computer-Assisted Intervention (pp. 406-415). Cham: Springer Nature Switzerland.
Jin, X., & Wang, Y. (2023). Understand Legal Documents with Contextualized Large Language Models. arXiv preprint arXiv:2303.12135.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 Journal of Industrial Engineering and Applied Science
This work is licensed under a Creative Commons Attribution 4.0 International License.