计算机与现代化 ›› 2021, Vol. 0 ›› Issue (01): 65-69.

• 信息安全 • 上一篇    下一篇

基于Focal Loss和卷积神经网络的入侵检测

  

  1. (广东工业大学计算机学院,广东广州510006)
  • 出版日期:2021-01-28 发布日期:2021-01-29
  • 作者简介:闫芮铵(1994—),男,河南开封人,硕士研究生,研究方向:信息物理融合系统,机器学习,E-mail: 15736959790@163.com; 张立臣(1962—),男,教授,博士,研究方向:信息物理融合系统,数据库,分布式数据库,Web数据库,移动数据库,实时数据库。
  • 基金资助:
    国家自然科学基金资助项目(61873068)

Intrusion Detection Based on Focal Loss and Convolutional Neural Network

  1. (School of Computers, Guangdong University of Technology, Guangzhou 510006, China)
  • Online:2021-01-28 Published:2021-01-29

摘要: 入侵检测是信息安全防护领域中的一个重要环节。随着网络技术的发展,主动防御网络入侵变得越来越重要,同时入侵数据变得更加海量、复杂和不平衡,这导致传统的入侵检测技术的检测性能比较低,因此如何提高入侵检测系统的性能对于不平衡数据集的检测性能是一项巨大的挑战。传统的CNN模型对于处理复杂的数据具有很好的性能,但是在处理不平衡数据集上的效果不是很好。为了解决这个问题,提出一种基于Focal Loss和卷积神经网络的入侵检测方法。与传统的卷积神经网络不同,该模型利用Focal Loss损失函数来解决数据不平衡问题,并在卷积层加入正则化方法(DropBlock)用来提高模型的泛化能力。采用KDD 99数据集进行的实验表明,该模型入侵检测的准确率和精确率比传统的入侵检测模型有所提高。

关键词: 入侵检测, 网络安全, 卷积神经网络, 正则化, 不平衡数据集, Focal Loss

Abstract: Intrusion detection is an important link in the field of information security protection. With the development of network technology, the active defense against network intrusions becomes more and more important, and intrusion data becomes more massive, complex, and unbalanced, which leads to the detection performance of the traditional intrusion detection technology is relatively low, so how to improve the detection performance of the intrusion detection system for unbalanced data sets is a huge challenge. The traditional CNN model has a good performance for processing complex data, but its effect of dealing with imbalanced data set is not very good. In order to solve this problem, an intrusion detection method based on Focal Loss and convolutional neural network is proposed. Different from traditional convolutional neural network, this model uses the Focal Loss function to solve the data imbalance problem, and in the convolutional layer, a regularization method (DropBlock) is added to improve the generalization ability of the model. Experiments of using KDD 99 data set show that the accuracy and precision of intrusion detection of this model are improved compared with the traditional intrusion detection model.

Key words: intrusion detection, network security, convolutional neural network, regularization, imbalanced data set, Focal Loss