[发明专利]一种基于GAN的不平衡网络入侵数据检测系统及其方法

说明书

本发明公开了一种基于GAN的不平衡网络入侵数据检测系统系统及其方法，该系统包括：GAN‑DAE模型、误差计算模块、结果输出模块和评价模块；GAN‑DAE模型包括生成器和鉴别器；该方法为：将待检测数据输入至训练好的GAN‑DAE模型中，GAN‑DAE模型根据误差计算模块得到的误差值的大小来识别待检测数据中是否存在异常；若误差值小于预设阈值，则输出不存在异常结果，若误差值大于预设阈值，则输出存在异常结果；本发明通过DAE捕获高维数据的潜在表示，以突破维度瓶颈；仅使用正常样本训练网络，通过计算测试样本的损失来识别异常。

技术领域

本发明涉及异常检测技术领域，更具体的说是涉及一种基于GAN的不平衡网络入侵数据检测系统及其方法。

背景技术

入侵检测技术概念于1980年由美国专家James.P.Anderson首次提出，Heady将入侵行为看作故意破坏系统资源行为的集合。随着计算机技术的迅速发展以及网络应用的广泛使用，网络威胁的规模和数量也在不断地增加，而且比以往任何时候都更严重，这可能会导致计算机中的重要信息被窃取和篡改，会给用户带来巨大的损失。因此，研究入侵检测对网络是否能安全、稳定地运行具有十分重要的意义。

然而，通常网络都是在正常情况下运行，出现被入侵或异常的情况很少，庞大的数据集中往往会出现异常类别的样本远远小于正常类别的样本。目前遇到的挑战是如何在样本维度高、样本类别不平衡数据中准确地识别出网络流量的异常行为。

现有技术中上述问题的解决办法大致分为传统的机器学习方法和深度学习方法。传统的机器学习方法如OC-SVM，IForest等，将网络入侵检测视为一种分类问题。这些方法通常使用数据训练出分类器，将其分为正常类和异常类。但是对于样本类别不平衡问题，它们往往倾向于确保数量多的类别的准确率，导致不能有效地识别出网络流量中的异常行为。由于网络产生的数据的维度越来越高，数据量也越来越大，在其他领域的出色表现以及其具有强大的数据学习和表达能力的深度学习被应用到了网络入侵检测中。这些方法包括CNN，RNN，DNN和DAGMM等，通过神经网络对数据进行多次抽象学习，可以较好地解决数据维度对入侵检测效果的限制问题，但是它们同样地不能很好地处理不平衡问题，对异常的识别往往具有倾向性，而生成对抗网络(Generative Adversarial Network,GAN)在图像识别领域可以有效地缓解数据不平衡问题。

因此，如何提供一种准确性高的基于GAN的不平衡网络入侵数据检测系统及其方法是本领域技术人员亟需解决的问题。

发明内容

有鉴于此，本发明提供了一种基于GAN的不平衡网络入侵数据检测系统及其方法，用于缓解网络入侵检测所遇到的数据维度高和样本中存在的类别不平衡的问题。

为了实现上述目的，本发明采用如下技术方案：

一种基于GAN的不平衡网络入侵数据检测系统，包括：GAN-DAE模型、误差计算模块、结果输出模块和评价模块；

所述GAN-DAE模型包括生成器和鉴别器；

所述生成器中的深度自编码器DAE至少依次包括第一编码器、第一解码器和第二编码器；

所述第一编码器，用于接收输入样本X并进行第一次编码得到编码后结果Z，并将Z发送至所述第一解码器；

所述第一解码器，用于接收Z并进行解码得到解码后结果X’，并将X’发送至所述第二编码器和所述鉴别器；

所述第二编码器，用于接收X’并进行第二次编码得到编码后的结果Z’；

所述鉴别器，用于接收X和X’，并判断X’是否为正常样本；

所述训练模块，用于控制生成器和鉴别器进行相互对抗，优化生成器和鉴别器的代价函数，直至生成器损失最小化，即鉴别器损失最大化；