计算机工程与应用 ›› 2023, Vol. 59 ›› Issue (21): 258-266.DOI: 10.3778/j.issn.1002-8331.2207-0128

• 网络、通信与安全 • 上一篇    下一篇

小样本下基于决策树-SNN的恶意流量检测方法

李道全,李玉秀,任大用   

  1. 青岛理工大学 信息与控制工程学院,山东 青岛 266520
  • 出版日期:2023-11-01 发布日期:2023-11-01

Malicious Traffic Detection Method Based on Decision Tree-SNN Under Small Sample

LI Daoquan, LI Yuxiu , REN Dayong   

  1. School of Information and Control Engineering, Qingdao University of Technology, Qingdao, Shandong 266520, China
  • Online:2023-11-01 Published:2023-11-01

摘要: 针对目前小样本下的恶意流量检测方法存在准确度低、特征提取不足和模型过拟合问题,提出了一种小样本下基于改进决策树-孪生神经网络的恶意流量检测算法。为了降低小样本下多分类任务的难度,利用类间中心距离构建二叉决策树将多分类问题转换为二分类问题。将孪生神经网络的对比分支设计为三支一维卷积神经网络并行的结构来解决小样本下特征提取不足问题。引入了通过池化策略和一维卷积操作优化的SE(squeeze-and-excitation)模块,以减少小样本下模型过拟合问题。通过对比样本的相似度实现了恶意流量检测。实验结果表明,所提方法在小样本下的恶意流量检测问题上具有良好的效果。

关键词: 恶意流量, 决策树, 孪生神经网络, 类间中心距离, 小样本, 通道注意力

Abstract: Aiming at the low accuracy, insufficient feature extraction, and model overfitting problems of the current malicious traffic detection method under small samples, a malicious traffic detection algorithm based on an improved decision tree-siamese neural network(SNN) under small samples is proposed. To reduce the difficulty of multi-classification tasks under small samples, a binary decision tree is constructed using the center distance between classes to convert multi-class problems into binary classification problems. The comparative branch of SNN is designed as a parallel structure of three one-dimensional convolutional neural networks to solve the problem of insufficient feature extraction under small samples. The squeeze-and-excitation module optimized by pooling strategies and one-dimensional convolution operations is introduced to reduce the problem of model overfitting under small samples. Malicious traffic detection is achieved by comparing the similarity of samples. The experimental results show that the proposed method effectively detects malicious traffic under small samples.

Key words: malicious traffic, decision tree, siamese neural network(SNN), center distance between classes, small sample, channel attention