计算机应用 ›› 0, Vol. ›› Issue (): 0-0.DOI: 10.11772/j.issn.1001-9081.2018102053

• 人工智能与仿真 •    下一篇

基于BN-DBN的网络安全态势要素获取机制

朱江1,王婷婷2   

  1. 1. 移动通信技术重庆市重点实验室(重庆邮电大学),重庆 400065
    2. 重庆邮电大学通信与信息工程学院
  • 收稿日期:2018-10-11 修回日期:2018-12-06 发布日期:2020-01-17 出版日期:2020-01-17
  • 通讯作者: 王婷婷

The Mechanism of Security Situational Element Acquisition Based on the BN-DBN

  • Received:2018-10-11 Revised:2018-12-06 Online:2020-01-17 Published:2020-01-17

摘要: 为了加快深度信念网络(Deep Belief Network,DBN)的收敛速度以及提高小样本条件下的态势要素的获取精度,提出了一种基于批量归一化(Batch Normalization,BN)的深度信念网络安全态势要素获取机制。一方面在深度神经网络中加入BN以解决梯度消失问题;另一方面再通过深度神经网络输出层提出了一种改进的主动学习(Improved Active Learning,IAL)算法反向微调深度信念网络,该算法通过在每次迭代中主动选择训练样本来平衡样本种类。理论分析和实例数据仿真结果表明该机制能够解决深度神经网络收敛速度过慢或梯度消失,以及样本的准确分类问题。其获取精度、收敛速度、算法复杂度优于该文中所列的其他态势要素获取机制。

Abstract: In order to speed up the convergence rate of the depth belief network (Deep belief NETWORK,DBN) and to improve the acquisition accuracy of the situation elements under small sample conditions, a new method based on batch normalization (Batch normalization, BN), the security situation element acquisition mechanism of the depth belief network. On the one hand, BN is added to the deep neural network to solve the problem of gradient disappearance. On the other hand, an improved active learning (improved active learning, IAL) algorithm is proposed by using the depth neural network output layer to reverse fine-tune the depth belief network, which balances the sample types by actively selecting training samples in each iteration. By selecting the training samples actively in each iteration, the algorithm can balance the sample types to improve the accuracy of the mechanism in the small sample condition. Theoretical analysis and case data simulation results show that the proposed mechanism can solve the problem of low convergence speed or gradient disappearance of the depth neural network and accurate classification of samples. The acquisition accuracy, convergence speed and algorithm complexity are better than other situation elements in this paper.

中图分类号: