[发明专利]网络流量特征序列生成模型的训练方法、生成方法及装置在审
申请号: | 202211043347.5 | 申请日: | 2022-08-29 |
公开(公告)号: | CN115496180A | 公开(公告)日: | 2022-12-20 |
发明(设计)人: | 刘军;李鉴学;肖扬;吴家威;刘芳;周文莉 | 申请(专利权)人: | 北京邮电大学 |
主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08;H04L9/40 |
代理公司: | 北京金咨知识产权代理有限公司 11612 | 代理人: | 薛海波 |
地址: | 100876 *** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 网络流量 特征 序列 生成 模型 训练 方法 装置 | ||
本发明提供一种网络流量特征序列生成模型的训练方法、生成方法及装置,针对网络流量的包含基础特征、数据包特征、时间特征和恶意流量特征的特征序列进行仿真,能够关注到应用层数据的特征。基于生成对抗网络进行无监督学习,简化了配置工作,提高了对网络流量特征序列进行仿真的效率。其中,基于强化学习构建生成器,能够提高对网络特征仿真的泛化能力,以适应更复杂的网络环境。采用蒙特卡洛学习构建生成器能够提高生成器的泛化能力。采用Wasserstein Loss作为生成器的损失函数,能够免生成对抗网络中常见的过拟合和模式崩溃问题。
技术领域
本发明涉及数据处理技术领域,尤其涉及一种网络流量特征序列生成模型的训练方法、生成方法及装置。
背景技术
网络流量分析是提取、记录和分析网络流量通信模式的过程。高效的网络流量分析模型可以优化网络的整体表现,增强网络的安全性和提高网络资源的管理效率。然而网络流量数据的不足将严重的影响这些分析模型的表现。但是,根据用户隐私协议,网络流量数据通常是被严格监管的,意味着需要花费很长的时间来获取访问网络流量数据的权限。同时,数据量的大小是很有限的,通常不足以支持复杂的分析模型。所以,网络工程师一般很难获取足量的、高质量的和干净的网络流量数据。
为了解决数据不足的问题,网络工程师们常常通过生成新的符合真实网络环境的网络流量数据来增强已有数据。这些生成的流量不是来源于真实的网络环境,而是由人工合成的。现有的网络流量生成仿真器绝大多数都是基于概率模型。例如,仿真器Harpoon从现有的网络流量记录中提取网络流量分布的概率参数,再根据这些参数生成新的网络流量数据。
然而,现有的网络流量生成器一般只能生成UDP(User Datagram Protocol)和TCP(Transmission Control Protocol)流,而不能生成应用层的流量特征;并且现有的网络流量生成器需要大量的手动配置,大大增加了使用者的时间成本和出错率;现有的网络流量生成器一般使用若干个网络流量分布的概率参数来描述网络流量数据,在复杂的网络环境中,网络拓扑和网络行为的复杂度都会大大提高。有限个数的概率参数通常不足以描述网络的复杂性,生成的网络流量质量不高。
发明内容
鉴于此,本发明实施例提供了一种网络流量特征序列生成模型的训练方法、生成方法及装置,以消除或改善现有技术中存在的一个或更多个缺陷,解决现有技术无法针对应用层的流量特征进行仿真的问题。
本发明的一个方面提供了一种网络流量特征序列生成模型的训练方法,该方法包括以下步骤:
获取指定多个网络设备的真实网络流量,并以数据包序列的形式记录为第一设定格式文件;
采用预设工具提取各网络设备之间所述网络流量的特征序列并记录为第二设定格式文件,所述特征序列包括基础特征、数据包特征、时间特征和恶意流量特征;
获取生成对抗网络初始模型,所述生成对抗网络初始模型包括生成器和判别器,所述生成器为单层门控那个循环单元,所述判别器为双层门控循环单元;以各网络设备之间仿真网络流量的特征序列作为状态空间,以要生成的所述仿真网络流量的特征序列作为动作空间,通过所述生成器建立所述状态空间到所述动作空间的映射更新动作空间,将所述动作空间添加至所述状态空间以更新状态空间,从所述判别器获取所述仿真网络流量的特征序列的期望评估作为奖励,通过强化学习的方式生成仿真网络流量的特征序列;所述判别器用于计算所述仿真网络流量或真实网络流量的特征序列对应的期望评估,并进行真伪判断;
基于生成对抗交替训练所述生成器和所述判别器,基于生成对抗损失对所述生成器和所述判别器进行迭代更新,将迭代更新得到的所述生成器作为网络流量特征序列生成模型。
在一些实施例中,获取指定的多个网络设备的网络流量,并以数据包序列的形式记录为第一设定格式文件,包括:
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京邮电大学,未经北京邮电大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202211043347.5/2.html,转载请声明来源钻瓜专利网。