物联网环境下基于多层极限学习机的分布式入侵检测方法与流程

1.本发明属于入侵检测和深度学习领域，具体涉及物联网环境下基于多层极限学习机的分布式入侵检测方法。


背景技术：

2.飞速发展的物联网技术通过日益复杂的传感设备与基于互联网的远程控制设备相连接，正在迅速弥合传统信息服务和周边的物理环境之间的鸿沟。许多潜力巨大的物联网应用服务应运而生，如环境监控、交通监控、健康医疗监视等等，这些应用极大提高了人类与计算设备的交互性。日益增长的物联网应用和物理信息服务对网络安全的要求也日渐增高，物联网下的入侵检测已成为当下物联网技术发展的一个主流趋势。
3.anderson于1980年最早提出了入侵检测的概念，拉开了入侵检测研究的序幕。受益于基于深度学习的分类模型的优秀表现，基于深度学习的入侵检测得到了飞速的发展。由于相关设备的资源受约束的特性，当下的一种流行解决方案是在靠近数据源的边界设备上分配一定的存储和计算能力，也就是“边缘计算”，这可以将特定的应用程序和服务从集中点转移到边缘。同时这些边界设备能够运行预置的分类模型，从而对于实现自动攻击检测的这类计算较重的任务，其核心运算设备能够减少运算和处理的工作量。但是当面对大量的数据时，却缺少足够的存储和运算处理能力来应对，同时受限于硬件性能约束，预置模型也无法进行大幅度的升级。因此我们希望能够将计算密度和存储量大的训练运算分离出去，尽可能让边界设备执行计算量小的工作，在保证准确率的前提下进一步提高效率。


技术实现要素：

4.为了解决现有技术中存在的物联网环境下分布式入侵检测的边界设备无法处理过大的训练数据问题，本发明提供了一种物联网环境下基于多层极限学习机的分布式入侵检测方法。本方法基于多隐藏层极限学习机模型，将计算密度和存储量大的训练运算转移到云服务器中进行，从而让边界设备能够基于云服务器中的预置模型来执行流量分类，分类出是正常流量还是网络攻击。本发明要解决的技术问题通过以下技术方法实现：
5.本发明提出一种物联网环境下基于多层极限学习机的分布式入侵检测方法，包括以下步骤：
6.步骤1：对网络流量数据进行预处理；
7.步骤2：搭建单隐藏层极限学习机分类模型；
8.步骤2.1，单隐藏层极限学习机一共有三层，第一层是输入层，第二层是隐藏层，第三层是输出层；其中输入层和隐藏层的连接权值w和偏置b是随机设定的，特征列数量为隐藏层隐藏节点数，偏置b通过给输入特征矩阵加入一列全1的值实现；
9.步骤2.2，对于m个任意的输入样本，首先经过任意的权重w和偏置b进行线性运算得到特征矩阵，再利用sigmoid激活函数对特征矩阵进行非线性运算，得到非线性特征矩阵h；其中，sigmoid激活函数将特征值映射到[0,1]，其公式如下：
[0034]
……
[0035]
x
n
‑1＝β
n
‑2·
x
n
‑2[0036]
步骤3.5，对于a
n
，即最后一层隐藏层，其输入矩阵为
[0037]
x
n
＝β
n
‑1·
x
n
‑1[0038]
之后代入单隐藏层极限学习机，得到最终的β
n
；
[0039]
步骤4：分别将训练集输入单隐藏层极限学习机和多隐藏层极限学习机，得到输出
[0040]
权重β；
[0041]
步骤5：利用训练好的模型对测试集进行分类。
[0042]
作为优选，所述步骤1包括以下步骤：
[0043]
步骤1.1，将网络流量数据作为数据集，将数据集的字符特征数据转换成数值，再将该特征进行one
‑
hot编码；
[0044]
步骤1.2，分别对每个特征数据进行归一化，公式如下：
[0045][0046]
其中x
min
为该特征的最小值，x
max
为该特征的最大值；
[0047]
步骤1.3，分割数据集，对整个数据集以8:2的比例分成训练集和测试集；
[0048]
步骤1.4，分别分割训练集和数据集中的label字段，并进行判断，若值为0，则标注为
‑
1，否则标注为1，即正常样本流量标注为
‑
1，攻击样本标注为1；
[0049]
步骤1.5，得到训练数据train_x,训练数据标签train_y；测试数据test_x,测试数据标签test_y。
[0050]
作为优选，所述的步骤2.2中的连接权值w为[
‑
1,1]的均匀分布，其分布密度函数如下：
[0051][0052]
其中m是w随机取值范围的最大值,n是w随机取值范围的最小值。
[0053]
作为优选，所述步骤5包括以下步骤：
[0054]
步骤5.1，在已知输出权重β，输入权重w和隐藏层偏置b的情况下，通过以下公式对输入的测试集进行分类，输出判别数组predicts；
[0055][0056]
其中，β
i
为输出权重，x为测试数据，sign()为符号函数，其表达式如下：
[0057][0058]
作为优选，还包括模型评价；具体包括以下步骤：
[0059]
步骤6.1，计算步骤5中得到的判别数组predicts与测试数据真实标签test_y的差异；
[0060]
步骤6.2，根据predicts和test_y画出两个模型的roc曲线；
[0061]
步骤6.3，根据predicts和test_y计算各项评价指标并记录训练和分类消耗时间。
[0062]
本发明相对于背景技术的优点在于：
[0063]
①
利用多隐藏层极限学习机模型，其输入层和隐藏层的连接权值和隐藏层的偏置是随机取的，且取完后不再调整。与bp神经网络相比，不需要不断反向调整权值和偏置，因此效率大大增高。
[0064]
②
分类只需对输入数据进行矩阵运算，计算量较小，其计算密度大的训练过程放在云服务器上进行，提高效率的同时降低对边界设备的性能要求。
[0065]
③
多隐藏层极限学习机通过构建多层隐藏层，相比较单隐藏层极限学习机，能够提取更深层次的数据特征联系，提高分类器的精度。
[0066]
以下将结合附图及实施例对本发明做进一步详细说明。
附图说明
[0067]
图1是本发明的整体架构；
[0068]
图2是本发明的实验流程；
[0069]
图3是本发明中单隐藏层极限学习机网络结构；
[0070]
图4是本发明中多隐藏层极限学习机网络结构；
[0071]
图5是本发明中的roc曲线图。
具体实施方式
[0072]
下面结合实施例来对本发明做进一步的描述，详细说明如下：
[0073]
本发明的入侵检测整体架构如附图图1所示，该架构分为两部分，边界设备与云服务器。边界设备预置了云服务器训练好的模型，其功能为对原始网络数据进行预处理以及分类，若分类结果出现异常，则通知管理员；云服务器接受边界设备传输来的网络数据，利用数据进行模型训练，再将新训练的模型分发给边界设备，更新模型。接下去是实验部分的详细说明，实验流程图可参考附图图2。
[0074]
步骤1，对网络流量数据进行预处理。
[0075]
本发明步骤1包含以下步骤：
[0076]
步骤1.1，将数据集的字符特征数据转换成数值，再将该特征进行one
‑
hot编码。
[0077]
本方法实验使用通信安全机构和加拿大网络安全研究所合作项目收集的网络入侵数据集(cse
‑
cic
‑
ids2017)，将其中的protocol字段下的字符值按照类别映射到数值，再进行one
‑
hot编码。
[0078]
步骤1.2，分别对每个字段特征数据进行归一化，公式如下：
[0079][0080]
其中x
min
为该字段的最小值，x
max
为该字段的最大值。例如某列特征数据的x
max
＝10，x
min
＝0，则该列的某特征值归一化后为0.6，处于[0,1]。
[0081]
步骤1.3，分割数据集，对整个数据集以8:2的比例分成训练集和测试集。
[0082]
在原始数据集中选择100,000条样本数据，标记为dataset，分割数据集，其中训练集80,000条，测试集20,000条。
[0083]
步骤1.4，分别分割训练集和数据集中的label字段，并进行判断，若值为0，则标注
为
‑
1，否则标注为1，即正常样本流量标注为
‑
1，攻击样本标注为1。
[0084]
将label字段下为benign的数据改成
‑
1，为正常网络流量；非benign的数据改成1，为网络攻击流量。
[0085]
步骤1.5，得到训练数据train_x,训练数据标签train_y；测试数据test_x,测试数据标签test_y。
[0086]
train_x数据集形状为[80000,74]，train_y数据集形状为[80000]；test_x数据集形状为[20000,74],test_y数据集形状为[20000]。
[0087]
步骤2，搭建单隐藏层极限学习机分类模型(elm)。(单隐藏层极限学习机网络结构参考附图图3)。
[0088]
本发明步骤2包含以下步骤：
[0089]
步骤2.1，单隐藏层极限学习机一共有三层，第一层是输入层，第二层是隐藏层，第三层是输出层。其中输入层和隐藏层的连接权值w和偏置b是随机设定的，在本方法中，连接权值w为[
‑
1,1]的均匀分布，其特征列数量为隐藏层隐藏节点数，该分布密度函数如下：
[0090][0091]
其中m是w随机取值范围的最大值,n是w随机取值范围的最小值；偏置b通过给输入特征矩阵加入一列全1的值实现；
[0092]
该实验中输入层的特征数据矩阵形状为[80000,74]，隐藏节点数为50，特征数据列数为74，因此权值w的形状为[50,74]，添加偏置b之后的特征数据矩阵形状为[80000,74]。
[0093]
步骤2.2，对于m个任意的输入样本，首先经过任意的权重w和偏置b进行线性运算得到特征矩阵，再利用sigmoid激活函数对特征矩阵进行非线性运算，得到非线性特征矩阵h。其中，sigmoid激活函数将特征值映射到[0,1]，其公式如下：
[0094][0095]
非线性特征矩阵h公式为h
ij
＝s(w
i
·
x
j
+b
i
)。
[0096]
经过激活函数后，特征数据矩阵的元素大小处于[
‑
1,1]。
[0097]
步骤2.3，在步骤2.2基础上，对于一个有l个隐藏节点的单隐藏层神经网络可以表示为
[0098][0099]
其中，s为sigmoid激活函数，n为输入特征数据矩阵的列数，b
i
为隐藏层第i个的偏置，β
i
为第i个隐藏节点的输出权重，o
j
为极限学习机的第j列特征值输出。
[0100]
步骤2.4，让单隐藏层神经网络的输出与目标值相差最小，因此可以表示为
[0101][0102]
其中t
j
为目标值，因此存在β
i
，w
i
和b
i
，使得
[0103][0104]
即hβ＝t，其中，h是隐藏节点的输出，β为输出权重，t为期望输出。
[0105]
输出权值β的形状为[50]，隐藏节点输出h的形状是[80000,50]，期望输出t的形状为[80000]。
[0106]
步骤2.5，训练单隐藏层极限学习机模型，等价于得到使得
[0107][0108]
即最小化损失函数为
[0109][0110]
步骤2.6，一旦随机确定输入权值w和隐藏层偏置b，那么输出矩阵h也将会被唯一确定。因此求解单隐藏层神经网络即为hβ＝t。输出权重可以被确认为：
[0111]
其中是moore
‑
penrose(mp)广义逆，
[0112]
步骤3，搭建多隐藏层极限学习机分类模型(mlelm)。(多隐藏层极限学习机网络结构参考附图图4)。
[0113]
本发明步骤3包含以下步骤：
[0114]
步骤3.1，多隐藏层极限学习机含有多个隐藏层，具体层数由用户给定的隐藏节点数量列表l的长度n决定。
[0115]
步骤3.2，假设列表l＝[a1,a2,a3,
…
a
n
]，a
i
对应了第i个隐藏层的隐藏节点数量。
[0116]
本实验列表l＝[50,100,150]。
[0117]
步骤3.3，对于α1，其求解输出权重β1的步骤与步骤2中的单隐藏层模型求解步骤一致，将其输出β1存入输出权重列表m。
[0118]
因此β1的形状为[50]。
[0119]
步骤3.4，对于a2,
…
,a
n
‑1，每层隐藏层的输入特征矩阵是该层前面隐藏层的输出权重依次与原始输入特征矩阵做内积的结果，公式表示如下：
[0120]
x2＝β1·
x
[0121]
x3＝β2·
(β1·
x)＝β2·
x2[0122]
……
[0123]
x
n
‑1＝β
n
‑2·
x
n
‑2[0124]
β2的形状为[100]。
[0125]
步骤3.5，对于a
n
，即最后一层隐藏层，其输入矩阵为
[0126]
x
n
＝β
n
‑1·
x
n
‑1[0127]
之后代入单隐藏层极限学习机，得到最终的β
n
。
[0128]
β3的形状为[150]。
[0129]
步骤4，训练模型，求出输出权重。
[0130]
本发明步骤4包含以下步骤：
[0131]
步骤4.1，分别将训练集输入单隐藏层极限学习机和多隐藏层极限学习机，得到输
出权重β。
[0132]
本实验中单隐藏层极限学习机模型输出权重β为[50]，多隐藏层极限学习机模型输出权重β为[150]。
[0133]
步骤5，利用训练好的模型对测试集进行分类。
[0134]
本发明步骤5包含以下步骤：
[0135]
步骤5.1，在已知输出权重β，输入权重w和隐藏层偏置b的情况下，通过以下公式对输入的测试集进行分类，输出判别数组predicts。
[0136][0137]
其中，β
i
为输出权重，x为测试数据，sign()是符号函数，其表达式如下：
[0138][0139]
输入数据形状为[20000,74]，输出判别数组predicts形状为[20000]，其值为[
‑
1,1,
‑
1,
‑
1,
…
,1,1,
‑
1]。
[0140]
步骤6，模型评价。
[0141]
本发明步骤6包含以下步骤：
[0142]
步骤6.1，计算步骤5中得到的判别数组predicts与测试数据真实标签test_y的差异。
[0143]
步骤6.2，根据predicts和test_y画出两个模型的roc曲线。
[0144]
步骤6.3，根据predicts和test_y计算各项评价指标，例如准确率(acc)、误报率(far)和检测率(dr)，精准率(precision)、召回率(recall)和f1
‑
measure，并记录训练和分类消耗时间。
[0145]
该方法为二元分类检测，即正常或异常，因此预测有四种结局，即真阳性(tp)：检测为异常，实际上也为异常；伪阳性(fp)：检测为异常，实际上正常；真阴性(tn)：检测为正常，实际上也正常；假阴性(fn)：检测为正常，实际上异常。按照该标准来计算模型预测结果的准确率(acc)、误报率(far)和检测率(dr)，精准率(precision)、召回率(recall)和f1
‑
measure。
[0146][0147][0148][0149][0150]
[0151][0152]
roc曲线适用于评估分类器的整体性能且不会随着样本中的正负样本比例变化而产生明显变化。本方法提到的两个模型评估指标如下表1所示。
[0153][0154][0155]
表1
[0156]
mlelm模型在acc、far和dr，precision、recall和f1
‑
measure六项指标上均优于elm，但是由于隐藏层的增多，其测试时间消耗大于elm。
[0157]
两模型的roc图见附图图5。分析实验结果可以得出mlelm模型的性能比elm更加出色。