[发明专利]一种基于知识蒸馏的数据包分类方法及相关装置

说明书

本申请公开了一种基于知识蒸馏的数据包分类方法及相关装置，所述方法包括基于经过训练的教师模型，确定预设训练集中的各训练数据包各自对应的类别概率向量；基于各训练数据包及其对应的属性数据和类别概率向量构建二进制决策树模型；根据二进制决策树模型生成三元匹配流表并将三元匹配流表部署于可编程交换机，并通过可编程交换机对数据包进行分类。本申请基于预先训练的教师模型所学的知识构建二进制决策树模型，以将复杂的教师模型转换为简单的二进制决策树模型，使得学习到教师模型所学的知识的二进制决策树模型可以布置于可编程交换机，这样一方面可以利用可编程交换机的数据包处理能力，另一方面可以提高可编程交换机的数据包分类效率。

技术领域

本申请涉及网络技术领域，特别涉及一种基于知识蒸馏的数据包分类方法及相关装置。

背景技术

数据包分类可以理解为根据数据包包头数据确定数据包所匹配的类别，并按照类别对应的操作动作对数据包进行处理，比如，丢弃数据包或者接受数据包等。目前普遍采用可编程交换机来执行数据包分类操作，可编程交换机具有低延迟和高吞吐等性能，并且可编程交换机在进行数据包分类过程中，允许管理员通过简单的编程指令对数据包进行操作。然而，可编程交换机所支持的编程指令所能执行的操作比较少(例如，仅支持整数的加减和位移操作等)，这使得具有高精度的分类网络模型(例如，卷积神经网络以及循环神经网络等)，从而限制了可编程交换机的数据包分类效率。

因而现有技术还有待改进和提高。

发明内容

本申请要解决的技术问题在于，针对现有技术的不足，提供一种基于知识蒸馏的数据包分类方法及相关装置。

为了解决上述技术问题，本申请实施例第一方面提供了一种基于知识蒸馏的数据包分类方法，所述的方法包括：

基于经过训练的教师模型，确定预设训练集中的各训练数据包各自对应的类别概率向量；

基于各训练数据包及其对应的属性数据和类别概率向量构建二进制决策树模型，其中，所述二进制决策树模型中的各叶节点均对应有目标分类类别；

根据所述二进制决策树模型生成用于数据包分类的三元匹配流表；

将所述三元匹配流表部署于可编程交换机，并通过所述可编程交换机对数据包进行分类。

所述的基于知识蒸馏的数据包分类方法，其中，所述属性数据为基于训练数据包的二进制数据包头确定的。

所述的基于知识蒸馏的数据包分类方法，其中，所述基于各训练数据包及其对应的属性数据和类别概率向量构建二进制决策树模型具体包括：

将各训练数据包及其对应的类别概率向量构成的训练数据集作为目标训练集，并将各训练数据包对应的属性数据构成的属性数据集作为目标属性数据集；

基于所述目标属性数据集确定目标属性位，并基于所述目标属性位将所述目标训练集中的各训练数据包划分至两个节点上；

对于两个节点中所包括的训练数据包的数量小于预设数量阈值的第一节点，将所述节点作为叶节点，并基于所述叶节点所包括的各训练数据包各自对应的类别概率向量确定所述叶节点对应的目标分类类别；

对于两个节点中所包括的训练数据包的数量大于或者等于预设数量阈值的第二节点，将所述目标属性数据集中的各属性数据中位于目标属性位的属性元素去除以得到候选属性数据；

将所述第二节点所包括的各训练数据包以其对应的类别概率向量所形成的训练数据作为目标数据集，将所述第二节点所包括的各训练数据包各自对应的候选属性数据所形成的属性数据集作为目标属性数据集；

继续执行基于所述目标属性数据集确定目标属性位的步骤，直至将所述预设训练集中的各训练数据包均划分至叶节点内。