Deep Learning-Based Network Traffic Anomaly Detection: A Study in IoT Environments

Deep Learning-Based Network Traffic Anomaly Detection: A Study in IoT Environments

Authors

  • Lin Li Electrical and Computer Engineering, Carnegie Mellon University, PA, USA
  • Yitian Zhang Accounting, UW-Madison, WI, USA
  • Jiayi Wang Computer engineering, Illinois institute of technology, IL, USA
  • Ke Xiong Computer Science, University of Southern California, CA, USA

DOI:

https://doi.org/10.53469/wjimt.2024.07(06).03

Keywords:

Internet of Things (IoT), Network Anomaly Detection, Deep Learning, LSTM

Abstract

This study presents a deep learning technique to detect vulnerabilities in the Internet of Things (IoT) environment. The proposed method combines the manual design with the learning function of autoencoders, together with the deep neural network architecture associated with the Long Short-Term Memory (LSTM) layer. Experiments performed on the IoT-23 dataset show that our method outperforms traditional machine learning and state-of-the-art deep learning methods, achieving an accuracy of 99.2%, an F1-score of 0.987, and an AUC-ROC of 0.998. The framework addresses critical issues in IoT security, including device diversity, vehicle model diversity, and real-time research needs. The ablation studies show the importance of combining manual and autoencoder-based feature extraction. Grad-CAM visualisations improve the model definition by identifying the essential features for classifying bad vehicles and not good. The model's ability to capture the time-dependent nature of network flows makes it helpful in investigating complex, time-dependent variables. Although there are limitations in computing needs and general capabilities across IoT ecosystems, the proposed system shows significant potential for practical use in IoT security systems. This research contributes to advancing IoT security by providing a powerful, efficient, and easily interpretable system to discover the system's capabilities in terms of strengths and weaknesses in the IoT network environment.

References

Iqbal, W., Abbas, H., Daneshmand, M., Rauf, B., & Bangash, Y. A. (2020). An in-depth analysis of IoT security requirements, challenges, and their countermeasures via software-defined security. IEEE Internet of Things Journal, 7(10), 10250-10276.

Ratre, A., & Pankajakshan, V. (2022, May). Deep imbalanced data learning approach for video anomaly detection. In 2022 National Conference on Communications (NCC) (pp. 391-396). IEEE.

Dawoud, A., Shahristani, S., & Raun, C. (2018, December). Deep learning for network anomaly detection. In 2018 International Conference on Machine Learning and Data Engineering (iCMLDE) (pp. 149-153). IEEE.

Kumar, J., & Ramesh, P. R. (2018, February). Low-cost energy efficient, intelligent security system with information stamping for IoT networks. In 2018 3rd International Conference On Internet of Things: Smart Innovation and Usages (IoT-SIU) (pp. 1-5). IEEE.

Sharma, R. K., & Pippal, R. S. (2020, September). Malicious Attack and Intrusion Prevention in IoT Network using Blockchain-based Security Analysis. In 2020 12th International Conference on Computational Intelligence and Communication Networks (CICN) (pp. 380-385). IEEE.

Shen, Q., Wen, X., Xia, S., Zhou, S., & Zhang, H. (2024). AI-Based Analysis and Prediction of Synergistic Development Trends in US Photovoltaic and Energy Storage Systems. International Journal of Innovative Research in Computer Science & Technology, 12(5), 36-46.

Zhu, Y., Yu, K., Wei, M., Pu, Y., & Wang, Z. (2024). AI-Enhanced Administrative Prosecutorial Supervision in Financial Big Data: New Concepts and Functions for the Digital Era. Social Science Journal for Advanced Research, 4(5), 40-54.

Li, H., Zhou, S., Yuan, B., & Zhang, M. (2024). OPTIMIZING INTELLIGENT EDGE COMPUTING RESOURCE SCHEDULING BASED ON FEDERATED LEARNING. Journal of Knowledge Learning and Science Technology ISSN: 2959-6386 (online), 3(3), 235-260.

Pu, Y., Zhu, Y., Xu, H., Wang, Z., & Wei, M. (2024). LSTM-Based Financial Statement Fraud Prediction Model for Listed Companies. Academic Journal of Sociology and Management, 2(5), 20-31.

Liu, Y., Tan, H., Cao, G., & Xu, Y. (2024). Enhancing User Engagement through Adaptive UI/UX Design: A Study on Personalized Mobile App Interfaces.

Huang, D., Yang, M., Wen, X., Xia, S., & Yuan, B. (2024). AI-Driven Drug Discovery: Accelerating the Development of Novel Therapeutics in Biopharmaceuticals. Journal of Knowledge Learning and Science Technology ISSN: 2959-6386 (online), 3(3), 206-224.

Lei, H., Wang, B., Shui, Z., Yang, P., & Liang, P. (2024). Automated Lane Change Behavior Prediction and Environmental Perception Based on SLAM Technology. arXiv preprint arXiv:2404.04492.

Wang, B., Zheng, H., Qian, K., Zhan, X., & Wang, J. (2024). Edge computing and AI-driven intelligent traffic monitoring and optimization. Applied and Computational Engineering, 77, 225-230.

Wang, Shikai, Kangming Xu, and Zhipeng Ling. "Deep Learning-Based Chip Power Prediction and Optimization: An Intelligent EDA Approach." International Journal of Innovative Research in Computer Science & Technology 12.4 (2024): 77-87.

Xu, K., Zhou, H., Zheng, H., Zhu, M., & Xin, Q. (2024). Intelligent Classification and Personalized Recommendation of E-commerce Products Based on Machine Learning. arXiv preprint arXiv:2403.19345.

Xu, K., Zheng, H., Zhan, X., Zhou, S., & Niu, K. (2024). Evaluation and Optimization of Intelligent Recommendation System Performance with Cloud Resource Automation Compatibility.

Zheng, H., Xu, K., Zhou, H., Wang, Y., & Su, G. (2024). Medication Recommendation System Based on Natural Language Processing for Patient Emotion Analysis. Academic Journal of Science and Technology, 10(1), 62-68.

Zheng, H.; Wu, J.; Song, R.; Guo, L.; Xu, Z. Predicting Financial Enterprise Stocks and Economic Data Trends Using Machine Learning Time Series Analysis. Applied and Computational Engineering 2024, 87, 26–32,

Liu, B., & Zhang, Y. (2023). Implementation of seamless assistance with Google Assistant leveraging cloud computing. Journal of Cloud Computing, 12(4), 1-15.

Zhang, M., Yuan, B., Li, H., & Xu, K. (2024). LLM-Cloud Complete: Leveraging Cloud Computing for Efficient Large Language Model-based Code Completion. Journal of Artificial Intelligence General science (JAIGS) ISSN: 3006-4023, 5(1), 295-326.

Li, P., Hua, Y., Cao, Q., & Zhang, M. (2020, December). Improving the Restore Performance via Physical-Locality Middleware for Backup Systems. In Proceedings of the 21st International Middleware Conference (pp. 341-355).

Zhou, S., Yuan, B., Xu, K., Zhang, M., & Zheng, W. (2024). THE IMPACT OF PRICING SCHEMES ON CLOUD COMPUTING AND DISTRIBUTED SYSTEMS. Journal of Knowledge Learning and Science Technology ISSN: 2959-6386 (online), 3(3), 193-205.

Shang, F., Zhao, F., Zhang, M., Sun, J., & Shi, J. (2024). Personalized Recommendation Systems Powered By Large Language Models: Integrating Semantic Understanding and User Preferences. International Journal of Innovative Research in Engineering and Management, 11(4), 39-49.

Sun, J., Wen, X., Ping, G., & Zhang, M. (2024). Application of News Analysis Based on Large Language Models in Supply Chain Risk Prediction. Journal of Computer Technology and Applied Mathematics, 1(3), 55-65.

Zhao, F., Zhang, M., Zhou, S., & Lou, Q. (2024). Detection of Network Security Traffic Anomalies Based on Machine Learning KNN Method. Journal of Artificial Intelligence General science (JAIGS) ISSN: 3006-4023, 1(1), 209-218.

Wang, S., Zheng, H., Wen, X., Xu, K., & Tan, H. (2024). Enhancing chip design verification through AI-powered bug detection in RTL code. Applied and Computational Engineering, 92, 27-33.

Yu, K., Bao, Q., Xu, H., Cao, G., & Xia, S. (2024). An Extreme Learning Machine Stock Price Prediction Algorithm Based on the Optimisation of the Crown Porcupine Optimisation Algorithm with an Adaptive Bandwidth Kernel Function Density Estimation Algorithm.

Zhang, X., 2024. Machine learning insights into digital payment behaviors and fraud prediction. Applied and Computational Engineering, 67, pp.61-67.

Zhang, X. (2024). Analyzing Financial Market Trends in Cryptocurrency and Stock Prices Using CNN-LSTM Models.

Che, C., Huang, Z., Li, C., Zheng, H., & Tian, X. (2024). Integrating generative ai into financial market prediction for improved decision making. arXiv preprint arXiv:2404.03523.

Che, C., Zheng, H., Huang, Z., Jiang, W., & Liu, B. (2024). Intelligent robotic control system based on computer vision technology. arXiv preprint arXiv:2404.01116.

Jiang, Y., Tian, Q., Li, J., Zhang, M., & Li, L. (2024). The Application Value of Ultrasound in the Diagnosis of Ovarian Torsion. International Journal of Biology and Life Sciences, 7(1), 59-62.

Li, L., Li, X., Chen, H., Zhang, M., & Sun, L. (2024). Application of AI-assisted Breast Ultrasound Technology in Breast Cancer Screening. International Journal of Biology and Life Sciences, 7(1), 1-4.

Lijie, L., Caiying, P., Liqian, S., Miaomiao, Z., & Yi, J. The application of ultrasound automatic volume imaging in detecting breast tumors.

Wang, S., Zheng, H., Wen, X., & Fu, S. (2024). Distributed high-performance computing methods for accelerating deep learning training. Journal of Knowledge Learning and Science Technology ISSN: 2959-6386 (online), 3(3), 108-126.

Wang, S., Zhu, Y., Lou, Q., & Wei, M. (2024). Utilizing Artificial Intelligence for Financial Risk Monitoring in Asset Management. Academic Journal of Sociology and Management, 2(5), 11-19.

Downloads

Published

2024-11-05
Loading...