CN113726545B - 基于知识增强生成对抗网络的网络流量生成方法及装置 - Google Patents

基于知识增强生成对抗网络的网络流量生成方法及装置 Download PDF

Info

Publication number
CN113726545B
CN113726545B CN202110697665.2A CN202110697665A CN113726545B CN 113726545 B CN113726545 B CN 113726545B CN 202110697665 A CN202110697665 A CN 202110697665A CN 113726545 B CN113726545 B CN 113726545B
Authority
CN
China
Prior art keywords
network
generated
data
knowledge graph
knowledge
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Active
Application number
CN202110697665.2A
Other languages
English (en)
Other versions
CN113726545A (zh
Inventor
李勇
惠铄迪
金德鹏
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Tsinghua University
Original Assignee
Tsinghua University
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Tsinghua University filed Critical Tsinghua University
Priority to CN202110697665.2A priority Critical patent/CN113726545B/zh
Publication of CN113726545A publication Critical patent/CN113726545A/zh
Application granted granted Critical
Publication of CN113726545B publication Critical patent/CN113726545B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • H—ELECTRICITY
    • H04—ELECTRIC COMMUNICATION TECHNIQUE
    • H04L—TRANSMISSION OF DIGITAL INFORMATION, e.g. TELEGRAPHIC COMMUNICATION
    • H04L41/00—Arrangements for maintenance, administration or management of data switching networks, e.g. of packet switching networks
    • H04L41/14—Network analysis or design
    • H04L41/145—Network analysis or design involving simulating, designing, planning or modelling of a network
    • G—PHYSICS
    • G06—COMPUTING; CALCULATING OR COUNTING
    • G06F—ELECTRIC DIGITAL DATA PROCESSING
    • G06F16/00—Information retrieval; Database structures therefor; File system structures therefor
    • G06F16/30—Information retrieval; Database structures therefor; File system structures therefor of unstructured textual data
    • G06F16/36—Creation of semantic tools, e.g. ontology or thesauri
    • G06F16/367—Ontology
    • H—ELECTRICITY
    • H04—ELECTRIC COMMUNICATION TECHNIQUE
    • H04L—TRANSMISSION OF DIGITAL INFORMATION, e.g. TELEGRAPHIC COMMUNICATION
    • H04L67/00—Network arrangements or protocols for supporting network services or applications
    • H04L67/01—Protocols
    • H04L67/12—Protocols specially adapted for proprietary or special-purpose networking environments, e.g. medical networks, sensor networks, networks in vehicles or remote metering networks

Landscapes

  • Engineering & Computer Science (AREA)
  • Computer Networks & Wireless Communication (AREA)
  • Signal Processing (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Computational Linguistics (AREA)
  • Data Mining & Analysis (AREA)
  • Databases & Information Systems (AREA)
  • Animal Behavior & Ethology (AREA)
  • General Engineering & Computer Science (AREA)
  • General Physics & Mathematics (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Health & Medical Sciences (AREA)
  • Computing Systems (AREA)
  • General Health & Medical Sciences (AREA)
  • Medical Informatics (AREA)
  • Data Exchanges In Wide-Area Networks (AREA)

Abstract

本发明提供一种基于知识增强生成对抗网络的网络流量生成方法及装置，该方法包括：根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备的知识图谱嵌入向量；将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，其为包括设备类型、流量序列长度属性的网络流量；其中，生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和生成网络数据进行对抗训练后得到。该方法需要数据量较少，能够对设备类型进行可靠区分，网络数据的真实性高。通过较少的数据，得到可靠的大规模物联网流量。

Description

基于知识增强生成对抗网络的网络流量生成方法及装置
技术领域
本发明涉及互联网技术领域，尤其涉及一种基于知识增强生成对抗网络的网络流量生成方法及装置。
背景技术
网络流量生成(Network Traffic Generation)是指对某一个互联网中存在的实体，生成其在指定时间片段内的流量数据序列，典型的应用场景是在网络硬件承压能力测试时，通过生成指定数量、类别的实体在指定时段内的流量，并在待测试平台中传输，根据传输的准确性和时效来判断网络硬件是否能够满足设定的要求。而在现有的实际生产应用下，大量网络流量数据的生成主要设计三种方法：1)根据预设参数的网络流量分布模型生成，例如泊松分布；2)根据给定的网络流量特点生成，例如数据包大小、到达时间间隔等；3)根据给定的互联网协议生产，如TCP。这些方法能偶生成大量的网络流量，但真实性、准确性无法得到保证。
物联网网络流量生成是指生成物联网设备的网络流量。物联网是一种计算设备、机械、数字机器相互关系的系统，具备通用唯一识别码，并具有通过网络传输数据的能力，无需人与人、或是人与设备的交互。物联网相关的功能主要通过物联网设备实现，例如智慧医疗设备、智能家居设备等。物联网设备通常通过网络连接到云平台，接受指令并行使功能，其网络流量受到物联网用户、服务商、使用环境等等多方因素的影响。
当前使用机器学习方法进行网络流量生成任务时，主要将其视作时间序列生成问题的子问题进行分析，生成一段时间内的网络流量特征序列。常用的传统方法主要包括生成对抗网络模型(Generative Adversarial Network model，简称GAN模型)等。其中，原始的生成对抗网络模型在面对复杂多变的时序序列时难以保证生成样本的多样性和真实性。物联网网络流量受多种环境、人为因素影响，时序上复杂多变，分布非常不均衡，而现有的物联网网络流量生成工作主要针对单一场景下的少量设备，实用性不足。因此，如何生成大量真实可靠的物联网设备网络流量成为了亟待解决的技术问题。
发明内容
针对现有技术存在的问题，本发明提供一种基于知识增强生成对抗网络的网络流量生成方法及装置。
本发明提供一种基于知识增强生成对抗网络的网络流量生成方法，包括：根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到。
根据本发明一个实施例的基于知识增强生成对抗网络的网络流量生成方法，所述将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络之前，还包括：根据样本知识图谱嵌入向量，输入构建的生成对抗网络模型的生成器网络，得到样本生成网络数据；根据样本生成网络数据，和样本的真实网络数据的各属性进行判别，以对所述构建的生成对抗网络模型进行训练，得到所述训练完成的生成对抗网络模型；其中，所述各属性包括，网络流量序列中各元素的到达时间间隔、数据包个数、数据包平均包长以及网络流量的序列长度。
根据本发明一个实施例的基于知识增强生成对抗网络的网络流量生成方法，所述基于设备及相关的用户、平台、位置和环境各属性确定知识图谱，包括：将设备标识作为头实体，将包括用户、平台、位置和环境作为尾实体，构建知识图谱。
根据本发明一个实施例的基于知识增强生成对抗网络的网络流量生成方法，所述将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，包括：将知识图谱嵌入向量和随机噪声作为设备类型生成器的输入，输出生成的设备类型；将知识图谱嵌入向量，生成的设备类型和随机噪声作为网络流量序列长度生成器的输入，输出生成的网络流量序列长度；将知识图谱嵌入向量，生成的设备类型和生成的网络流量序列长度，输入采用循环神经网络和自注意力机制相结合的网络流量序列生成器，输出生成网络数据。
根据本发明一个实施例的基于知识增强生成对抗网络的网络流量生成方法，所述对所述构建的生成对抗网络模型进行训练，包括：通过多层感知机作为判别器，判别器以真实数据和生成的数据作为输入，输出判别结果；根据所述判别结果和预设的损失函数，对所述构建的生成对抗网络模型进行训练。
根据本发明一个实施例的基于知识增强生成对抗网络的网络流量生成方法，所述根据所述判别结果和预设的损失函数，对所述构建的生成对抗网络模型进行训练，包括：采用添加梯度惩罚的 Wasserstein距离作为损失函数，对所述构建的生成对抗网络模型进行训练。
根据本发明一个实施例的基于知识增强生成对抗网络的网络流量生成方法，所述损失函数包括：
Figure BDA0003129174960000041
其中，D(O)和
Figure BDA0003129174960000042
分别表示真实数据和生成数据的判别结果，
Figure BDA0003129174960000043
和
Figure BDA0003129174960000044
分别表示真实数据和生成数据的分布，
Figure BDA0003129174960000045
表示真实-生成样本对之间的均匀采样，
Figure BDA0003129174960000046
和
Figure BDA0003129174960000047
分别表示样本的判别结果和分布。
本发明还提供一种基于知识增强生成对抗网络的网络流量生成装置，包括：向量表达模块，用于根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；流量生成模块，用于将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到。
本发明还提供一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现如上述任一种所述基于知识增强生成对抗网络的网络流量生成方法的步骤。
本发明还提供一种非暂态计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现如上述任一种所述基于知识增强生成对抗网络的网络流量生成方法的步骤。
本发明提供的基于知识增强生成对抗网络的网络流量生成方法及装置，只需对设备属性进行知识图谱的表达，需要数据量较少，而通过设备属性生成的嵌入向量，使设备相关的背景信息关联到网络流量中，并且对设备类型进行可靠区分，增强生成网络数据的真实性。最后，再通过生成对抗网络进行学习，从而通过较少的数据，得到可靠的大规模物联网流量。
附图说明
为了更清楚地说明本发明或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作一简单地介绍，显而易见地，下面描述中的附图是本发明的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
图1是本发明提供的基于知识增强生成对抗网络的网络流量生成方法的流程示意图；
图2是本发明提供的物联网设备的网络流量生成方法的流程示意图；
图3是本发明提供的物联网设备网络流量序列数据示意图；
图4是本发明提供的物联网知识图谱示意图；
图5是本发明提供的物联网设备网络流量生成模型示意图；
图6是本发明提供的基于知识增强生成对抗网络的网络流量生成装置的结构示意图；
图7是本发明提供的电子设备的结构示意图。
具体实施方式
为使本发明的目的、技术方案和优点更加清楚，下面将结合本发明中的附图，对本发明中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。
现有的相关工作存在以下几点局限：(1)仅考虑了网络流量信息，未能关注产生网络流量的设备相关的背景信息；(2)仅能生成网络流量序列，无法生成物联网设备的类别等相关属性；(3)生成模型难以挖掘实际使用中复杂多变的物联网流量时序特征，生成的流量不够真实、可靠。
下面结合图1-图7描述本发明的基于知识增强生成对抗网络的网络流量生成方法及装置。图1是本发明提供的基于知识增强生成对抗网络的网络流量生成方法的流程示意图，如图1所示，本发明提供基于知识增强生成对抗网络的网络流量生成方法，包括：
101、根据物联网设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个物联网设备对应的知识图谱嵌入向量。
以物联网为例，可结合图2参照，首先基于设备属性和网络流量，可以提取物联网相关的知识，并利用设备和平台、用户之间的通信关系形成知识图谱。利用知识图谱嵌入模型，如TransE，可以得到物联网设备所对应的知识图谱嵌入向量。
102、将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量。
其中，生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到。
在网络训练过程中，先构建生成对抗网络，构建生成器分别生成物联网设备的设备类型、流量序列长度、具体的网络流量，再利用判别器判别真实数据和生成的数据，对抗训练生成器和判别器，逐步得到可靠的生成对抗网络模型。
将101的知识图谱嵌入向量输入得到的生成对抗网络模型，生成对应的物联网流量。
本发明的基于知识增强生成对抗网络的网络流量生成方法，只需对设备属性进行知识图谱的表达，需要数据量较少，而通过设备属性生成的嵌入向量，使设备相关的背景信息关联到网络流量中，并且对设备类型进行可靠区分，增强生成网络数据的真实性。最后，再通过生成对抗网络进行学习，从而通过较少的数据，得到可靠的大规模物联网流量。
在一个实施例中，所述将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络之前，还包括：根据样本知识图谱嵌入向量，输入构建的生成对抗网络模型的生成器网络，得到样本生成网络数据；根据样本生成网络数据，和样本的真实网络数据的各属性进行判别，以对所述构建的生成对抗网络模型进行训练，得到所述训练完成的生成对抗网络模型；其中，所述各属性包括，网络流量序列中各元素的到达时间间隔、数据包个数、数据包平均包长以及网络流量的序列长度。
生成物联网设备网络流量的问题可以表示为如下形式：给定真实的物联网设备网络流量数据集
Figure BDA0003129174960000071
其中Oi代表数据集中的第 i个物联网设备的数据。对于每个物联网设备，Oi＝(Ci，Ti)，其中Ci表示物联网设备的设备类别，Ti表示网络流量序列。如图3所示，本发明中
Figure BDA0003129174960000072
其中
Figure BDA0003129174960000073
表示第i个物联网设备在序列中第j个元素的到达时间间隔，
Figure BDA0003129174960000074
表示第i个物联网设备在序列中第j个元素的数据包个数，
Figure BDA0003129174960000075
表示第i个物联网设备在序列中第j个元素的数据包平均包长，Mi表示第i个物联网设备的网络流量序列Ti的长度。那么给定真实的物联网流量数据集S，本发明学习生成器G，生成可靠的数据集
Figure BDA0003129174960000081
生成对抗网络是一种基于对抗学习的生成模型，典型情况下分为生成器G和判别器D两部分，两者进行交替训练。训练过程中，生成器G负责生成与真实数据相似的仿真数据，而判别器D负责分辨真实样本和仿真样本，从而形成如下竞争：
Figure BDA0003129174960000082
其中，θ和φ分别是生成器G和判别器D的参数，x是相关样本，pd代表真实数据的概率分布,Gθ代表生成器生成的数据的概率分布。
本发明中，通过网络流量序列中各元素的到达时间间隔、数据包个数、数据包平均包长以及网络流量的序列长度，对网络流量序列进行精准的刻画，从而提高生成对抗网络模型输出数据的准确性。
在一个实施例中，所述基于设备及相关的用户、平台、位置和环境各属性确定知识图谱，包括：将设备标识作为头实体，将包括用户、平台、位置和环境作为尾实体，构建知识图谱。
物联网设备的背景信息同时包括语义知识和网络结构。首先，根据全球移动通信系统联盟(Global System for Mobile Communications Alliance，简称GSMA)的设备类型数据集，生产厂商、供应商和用户的描述可以提取到物联网设备的硬件型号等背景知识；接下来，根据物联网流量中所包含的可读内容，可以提取出相关的用户、平台、位置、环境等知识；最后，根据物联网流量中包含的源、目的IP可以获取物联网设备、用户、平台之间的通信关系。
本发明中，将物联网设备作为头实体，将用户、平台、位置、环境等信息作为尾实体，则可以得到如图4所示的物联网知识图谱，三元组表示示例如下：
·<device di,manufactured_by,manufacturer:XX>,
·<device di,located_in,city:BB>,
·<device di,communicate_with,platform:XXyun>.
利用知识图谱嵌入模型TransE，可以得到第i个物联网设备di所对应的知识图谱嵌入向量Ki。
在一个实施例中，所述将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，包括：将知识图谱嵌入向量和随机噪声作为设备类型生成器的输入，输出生成的设备类型；将知识图谱嵌入向量，生成的设备类型和随机噪声作为网络流量序列长度生成器的输入，输出生成的网络流量序列长度；将知识图谱嵌入向量，生成的设备类型和生成的网络流量序列长度，输入采用循环神经网络和自注意力机制相结合的网络流量序列生成器，输出生成网络数据。
如图5所示，物联网网络流量生成器分为三个子生成器：设备类型生成器，序列长度生成器，网络流量生成器。通过条件机制把知识图谱嵌入向量Ki引入每个子生成器中，同时子生成器所生成的内容也根据不同的条件机制被引入其它子生成器内。
条件机制的具体情况如下所示：
Figure BDA0003129174960000091
首先，采用多层感知机(multilayer perceptrons,简称MLP)作为设备类型生成器
Figure BDA0003129174960000092
设备类型生成器
Figure BDA0003129174960000093
包括多层网络结构单元，并以sigmoid函数作为激活函数。每个网络结构单元由一个线性层、一个批归一化层、一个非线性层(ReLU)组成。设备类型生成器以知识图谱嵌入向量K和随机噪声N作为输入，并输出生成的设备类型
Figure BDA0003129174960000094
接下来，采用和设备类型生成器类似的多层感知机模型作为网络流量序列长度生成器
Figure BDA0003129174960000101
同时，sigmoid函数输出数值在(0,1)区间内，生成器
Figure BDA0003129174960000102
根据需要生成的最大序列长度对激活函数的输出进行缩放。网络流量序列长度生成器
Figure BDA0003129174960000103
以知识图谱嵌入向量K，生成的设备类型
Figure BDA0003129174960000104
和随机噪声N作为输入，并输出生成的网络流量序列长度
Figure BDA0003129174960000105
最后，采用时间序列网络，如长短时记忆网络(Long Short Term Memory，简称LSTM)和自注意力机制(Self-attention)相结合设计 -生成器
Figure BDA0003129174960000106
长短时记忆网络是循环神经网络模型(Recurrent Neural Network model，简称RNN模型)的一种，常用于处理时间序列中的较长时间片段内时序特征，该模型在每一步的循环中输出固定长度的样本。自注意力机制则通过将向量投影到key，query，value三个不同的维度表示并发掘向量内部的相关性。
在本发明中，以知识图谱嵌入向量K，生成的设备类型
Figure BDA0003129174960000107
和随机噪声N作为每一步的输入，同时生成长度为B的物联网网络流量序列所对应的向量
Figure BDA0003129174960000108
此处k＝1,2,…,Q，
Figure BDA0003129174960000109
是所需的总步数，通过网络流量序列长度生成器输出的序列长度决定。接下来，第k步产生的
Figure BDA00031291749600001010
被继续输入到自注意力层中，并输出长度为B的物联网网络流量序列
Figure BDA00031291749600001011
具体过程如下所示：
Figure BDA00031291749600001012
Figure BDA00031291749600001013
在Q步之后，生成器将生成的所有样本
Figure BDA00031291749600001014
恢复成生成的物联网网络流量序列
Figure BDA0003129174960000111
最后，由三个子生成器可以逐步输出生成的数据集
Figure BDA0003129174960000112
在一个实施例中，所述对所述构建的生成对抗网络模型进行训练，包括：通过多层感知机作为判别器，判别器以真实数据和生成的数据作为输入，输出判别结果；根据所述判别结果和预设的损失函数，对所述构建的生成对抗网络模型进行训练。
系统采用多层感知机(MLP)作为判别器Dφ，包括多层网络结构单元，并以线性层作为激活函数。每个网络结构单元由一个线性层、一个层归一化层、一个非线性层(ReLU)组成。判别器以真实数据和生成的数据作为输入，并以0和1的形式输出判别结果。如下所示：
Figure BDA0003129174960000113
在一个实施例中，所述根据所述判别结果和预设的损失函数，对所述构建的生成对抗网络模型进行训练，包括：采用添加梯度惩罚的 Wasserstein距离作为损失函数，对所述构建的生成对抗网络模型进行训练。
本发明采用添加梯度惩罚的Wasserstein距离作为损失函数，作为可选实施例，如下所示：
Figure BDA0003129174960000114
其中，D(O)和
Figure BDA0003129174960000115
分别表示真实数据和生成数据的判别结果，
Figure BDA0003129174960000116
和
Figure BDA0003129174960000117
分别表示真实数据和生成数据的分布，
Figure BDA0003129174960000118
表示真实-生成样本对之间的均匀采样，
Figure BDA0003129174960000119
和
Figure BDA00031291749600001110
分别表示这些样本的判别结果和分布，λ为常数，可取值为10，||表示范数，
Figure BDA00031291749600001111
表示对应的梯度。
训练时，生成器通过使损失函数最小化来生成尽可能接近真实的样本，判别器通过使损失函数最大化来尽可能地判别样本的真实性。
下面结合上述各实施例进行举例说明如下：
使用者想利用10,187个移动物联网设备在3天内的网络流量生成仿真数据集，真实数据集中的设备分为10种不同的类型，根据网络流量可以得到针对每个设备到达时间间隔、数据包个数、数据包平均包长的特征序列。
同时，根据设备描述信息和网络流量内容，可以抽取得到知识图谱，其中包含10,187个设备作为头实体，涉及20种关系。对该知识图谱使用TransE模型，获取每个设备(头实体)的知识图谱嵌入向量。
接下来开始训练模型，选取M＝64作为每批用于训练的设备样本数目。在每一次迭代中，先训练并更新判别器N＝5次，再更新生成器一次。在训练判别器时，首先随机选取M个设备作为真实样本，再使用这M个设备的知识图谱嵌入向量和随机生成的(0,1)区间内噪声通过生成器生成M个仿真样本，随后将仿真样本和真实样本同时输入判别器，根据判别器的判别结果计算损失函数，并利用自适应矩估计(adaptive moment estimation，简称Adam)优化器最大化损失函数，更新判别器参数。训练判别器N次后，随机选取M个设备的知识图谱嵌入向量和随机生成的(0,1)区间内噪声通过生成器生成 M个仿真样本，利用已更新参数的判别器对这些仿真样本进行判别，根据判别结果计算损失函数，再利用自适应矩估计优化器最小化损失函数，更新生成器参数。在约300次迭代后，模型达到收敛阈值，训练完成。
最后，设定需要生成的样本数目Q，从真实设备的知识图谱嵌入向量集合中随机采样Q次，得到Q个知识图谱嵌入向量，和随机生成的(0,1)区间内噪声向量同时输入生成器，得到Q个仿真样本。
下面对本发明提供的基于知识增强生成对抗网络的网络流量生成装置进行描述，下文描述的基于知识增强生成对抗网络的网络流量生成装置与上文描述的基于知识增强生成对抗网络的网络流量生成方法可相互对应参照。
图6是本发明提供的基于知识增强生成对抗网络的网络流量生成装置的结构示意图，如图6所示，该基于知识增强生成对抗网络的网络流量生成装置包括：向量表达模块601和流量生成模块602。其中，向量表达模块601用于根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；流量生成模块602用于将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到。
本发明实施例提供的装置实施例是为了实现上述各方法实施例的，具体流程和详细内容请参照上述方法实施例，此处不再赘述。
本发明实施例提供的基于知识增强生成对抗网络的网络流量生成装置，只需对设备属性进行知识图谱的表达，需要数据量较少，而通过设备属性生成的嵌入向量，使设备相关的背景信息关联到网络流量中，并且对设备类型进行可靠区分，增强生成网络数据的真实性。最后，再通过生成对抗网络进行学习，从而通过较少的数据，得到可靠的大规模物联网流量。
图7是本发明提供的电子设备的结构示意图，如图7所示，该电子设备可以包括：处理器(processor)701、通信接口(Communications Interface)702、存储器(memory)703和通信总线704，其中，处理器701，通信接口702，存储器703通过通信总线704完成相互间的通信。处理器701可以调用存储器703中的逻辑指令，以执行基于知识增强生成对抗网络的网络流量生成方法，该方法包括：根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到。
此外，上述的存储器703中的逻辑指令可以通过软件功能单元的形式实现并作为独立的产品销售或使用时，可以存储在一个计算机可读取存储介质中。基于这样的理解，本发明的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来，该计算机软件产品存储在一个存储介质中，包括若干指令用以使得一台计算机设备(可以是个人计算机，服务器，或者网络设备等)执行本发明各个实施例所述方法的全部或部分步骤。而前述的存储介质包括：U盘、移动硬盘、只读存储器(ROM，Read-Only Memory)、随机存取存储器(RAM，Random Access Memory)、磁碟或者光盘等各种可以存储程序代码的介质。
另一方面，本发明还提供一种计算机程序产品，所述计算机程序产品包括存储在非暂态计算机可读存储介质上的计算机程序，所述计算机程序包括程序指令，当所述程序指令被计算机执行时，计算机能够执行上述各方法所提供的基于知识增强生成对抗网络的网络流量生成方法，该方法包括：根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到。
又一方面，本发明还提供一种非暂态计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现以执行上述各实施例提供的基于知识增强生成对抗网络的网络流量生成方法，该方法包括：根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到。
以上所描述的装置实施例仅仅是示意性的，其中所述作为分离部件说明的单元可以是或者也可以不是物理上分开的，作为单元显示的部件可以是或者也可以不是物理单元，即可以位于一个地方，或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。本领域普通技术人员在不付出创造性的劳动的情况下，即可以理解并实施。
通过以上的实施方式的描述，本领域的技术人员可以清楚地了解到各实施方式可借助软件加必需的通用硬件平台的方式来实现，当然也可以通过硬件。基于这样的理解，上述技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来，该计算机软件产品可以存储在计算机可读存储介质中，如ROM/RAM、磁碟、光盘等，包括若干指令用以使得一台计算机设备(可以是个人计算机，服务器，或者网络设备等)执行各个实施例或者实施例的某些部分所述的方法。
最后应说明的是：以上实施例仅用以说明本发明的技术方案，而非对其限制；尽管参照前述实施例对本发明进行了详细的说明，本领域的普通技术人员应当理解：其依然可以对前述各实施例所记载的技术方案进行修改，或者对其中部分技术特征进行等同替换；而这些修改或者替换，并不使相应技术方案的本质脱离本发明各实施例技术方案的精神和范围。

Claims (8)

1.一种基于知识增强生成对抗网络的网络流量生成方法，其特征在于，包括：
根据设备及相关的用户、平台、位置和环境各属性，确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；
将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据；所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；
其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到；
所述根据设备及相关的用户、平台、位置和环境各属性，确定知识图谱，包括：将设备标识作为头实体，将包括用户、平台、位置和环境作为尾实体，构建知识图谱；
所述将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，包括：
将知识图谱嵌入向量和随机噪声作为设备类型生成器的输入，输出生成的设备类型；
将知识图谱嵌入向量，生成的设备类型和随机噪声作为网络流量序列长度生成器的输入，输出生成的网络流量序列长度；
将知识图谱嵌入向量，生成的设备类型和生成的网络流量序列长度，输入采用循环神经网络和自注意力机制相结合的网络流量序列生成器，输出生成网络数据。
2.根据权利要求1所述的基于知识增强生成对抗网络的网络流量生成方法，其特征在于，所述将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络之前，还包括：
根据样本知识图谱嵌入向量，输入构建的生成对抗网络模型的生成器网络，得到样本生成网络数据；
根据样本生成网络数据，和样本的真实网络数据的各属性进行判别，以对所述构建的生成对抗网络模型进行训练，得到所述训练完成的生成对抗网络模型；
其中，所述各属性包括，网络流量序列中各元素的到达时间间隔、数据包个数、数据包平均包长以及网络流量的序列长度。
3.根据权利要求2所述的基于知识增强生成对抗网络的网络流量生成方法，其特征在于，所述对所述构建的生成对抗网络模型进行训练，包括：
通过多层感知机作为判别器，判别器以真实数据和生成的数据作为输入，输出判别结果；
根据所述判别结果和预设的损失函数，对所述构建的生成对抗网络模型进行训练。
4.根据权利要求3所述的基于知识增强生成对抗网络的网络流量生成方法，其特征在于，所述根据所述判别结果和预设的损失函数，对所述构建的生成对抗网络模型进行训练，包括：
采用添加梯度惩罚的Wasserstein距离作为损失函数，对所述构建的生成对抗网络模型进行训练。
5.根据权利要求4所述的基于知识增强生成对抗网络的网络流量生成方法，其特征在于，所述损失函数包括：
Figure FDA0003849658630000021
其中，D(O)和
Figure FDA0003849658630000022
分别表示真实数据和生成数据的判别结果，
Figure FDA0003849658630000023
和
Figure FDA0003849658630000024
分别表示真实数据和生成数据的分布，
Figure FDA0003849658630000025
表示真实-生成样本对之间的均匀采样，
Figure FDA0003849658630000026
和
Figure FDA0003849658630000027
分别表示样本的判别结果和分布，λ为常数。
6.一种基于知识增强生成对抗网络的网络流量生成装置，其特征在于，包括：
向量表达模块，用于根据设备及相关的用户、平台、位置和环境各属性确定知识图谱，基于知识图谱嵌入模型，得到每个设备对应的知识图谱嵌入向量；
流量生成模块，用于将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，所述生成网络数据，为包括设备类型、流量序列长度属性的网络流量；
其中，所述生成对抗网络模型，是基于多个已知流量设备的知识图谱嵌入向量，输入生成器网络，输出生成网络数据，再利用判别器对真实流量数据和所述生成网络数据进行对抗训练后得到；
所述根据设备及相关的用户、平台、位置和环境各属性，确定知识图谱，包括：将设备标识作为头实体，将包括用户、平台、位置和环境作为尾实体，构建知识图谱；
所述将每个设备的知识图谱嵌入向量，输入训练完成的生成对抗网络模型的生成器网络，输出生成网络数据，包括：
将知识图谱嵌入向量和随机噪声作为设备类型生成器的输入，输出生成的设备类型；
将知识图谱嵌入向量，生成的设备类型和随机噪声作为网络流量序列长度生成器的输入，输出生成的网络流量序列长度；
将知识图谱嵌入向量，生成的设备类型和生成的网络流量序列长度，输入采用循环神经网络和自注意力机制相结合的网络流量序列生成器，输出生成网络数据。
7.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至5任一项所述基于知识增强生成对抗网络的网络流量生成方法的步骤。
8.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至5任一项所述基于知识增强生成对抗网络的网络流量生成方法的步骤。
CN202110697665.2A 2021-06-23 2021-06-23 基于知识增强生成对抗网络的网络流量生成方法及装置 Active CN113726545B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202110697665.2A CN113726545B (zh) 2021-06-23 2021-06-23 基于知识增强生成对抗网络的网络流量生成方法及装置

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202110697665.2A CN113726545B (zh) 2021-06-23 2021-06-23 基于知识增强生成对抗网络的网络流量生成方法及装置

Publications (2)

Publication Number Publication Date
CN113726545A CN113726545A (zh) 2021-11-30
CN113726545B true CN113726545B (zh) 2022-12-23

Family

ID=78672906

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202110697665.2A Active CN113726545B (zh) 2021-06-23 2021-06-23 基于知识增强生成对抗网络的网络流量生成方法及装置

Country Status (1)

Country Link
CN (1) CN113726545B (zh)

Families Citing this family (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN115277086B (zh) * 2022-06-16 2023-10-20 西安电子科技大学 基于生成对抗网络的网络背景流量生成方法
CN115396346B (zh) * 2022-08-30 2023-11-28 北京知道创宇信息技术股份有限公司 仿真压力测试方法、装置、系统和电子设备
CN118070850B (zh) * 2024-04-18 2024-06-18 清华大学 数据中心网络流量生成方法、装置、介质及计算机程序

Citations (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN109639479A (zh) * 2018-12-07 2019-04-16 北京邮电大学 基于生成对抗网络的网络流量数据增强方法及装置
CN112287117A (zh) * 2020-10-30 2021-01-29 云南电网有限责任公司电力科学研究院 一种基于数据自动生成的资产管理知识库构建方法

Family Cites Families (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
EP3745309A1 (en) * 2019-05-27 2020-12-02 Robert Bosch GmbH Training a generative adversarial network
CN111866882B (zh) * 2019-12-17 2023-03-31 南京理工大学 一种基于生成对抗网络的移动应用流量生成方法
CN111476294B (zh) * 2020-04-07 2022-03-22 南昌航空大学 一种基于生成对抗网络的零样本图像识别方法及系统
CN111915090A (zh) * 2020-08-10 2020-11-10 哈尔滨安天科技集团股份有限公司 基于知识图谱的预测方法、装置、电子设备及存储介质

Patent Citations (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN109639479A (zh) * 2018-12-07 2019-04-16 北京邮电大学 基于生成对抗网络的网络流量数据增强方法及装置
CN112287117A (zh) * 2020-10-30 2021-01-29 云南电网有限责任公司电力科学研究院 一种基于数据自动生成的资产管理知识库构建方法

Also Published As

Publication number Publication date
CN113726545A (zh) 2021-11-30

Similar Documents

Publication Publication Date Title
CN113726545B (zh) 基于知识增强生成对抗网络的网络流量生成方法及装置
EP3992857A1 (en) Method and device for generating neural network model, and computer-readable storage medium
CN106953862A (zh) 网络安全态势的感知方法和装置及感知模型训练方法和装置
CN111400452B (zh) 文本信息分类处理方法、电子设备及计算机可读存储介质
KR20220049604A (ko) 객체 추천 방법 및 장치, 컴퓨터 기기 및 매체
EP4350572A1 (en) Method, apparatus and system for generating neural network model, devices, medium and program product
CN109246027B (zh) 一种网络维护的方法、装置和终端设备
Nazarenko et al. Features of application of machine learning methods for classification of network traffic (features, advantages, disadvantages)
CN110362663A (zh) 自适应多感知相似度检测和解析
WO2023206771A1 (zh) 基于决策流图的环境建模方法、装置和电子设备
CN112131578A (zh) 攻击信息预测模型的训练方法、装置、电子设备及存储介质
CN114330966A (zh) 一种风险预测方法、装置、设备以及可读存储介质
CN115048938A (zh) 基于语义和句法双通道的语句情感分析方法以及装置
CN114492601A (zh) 资源分类模型的训练方法、装置、电子设备及存储介质
Remmide et al. Detection of phishing URLs using temporal convolutional network
CN114116995B (zh) 基于增强图神经网络的会话推荐方法、系统及介质
CN115131058A (zh) 账号识别方法、装置、设备及存储介质
CN108122029A (zh) 一种相机特效的推荐方法及装置
CN111062019A (zh) 用户攻击检测方法、装置、电子设备
CN114330704A (zh) 语句生成模型更新方法、装置、计算机设备和存储介质
CN111090740B (zh) 一种用于对话系统的知识图谱生成方法
CN114780863B (zh) 基于人工智能的项目推荐方法、装置、计算机设备及介质
US20220343182A1 (en) Article processing method, electronic device, and program product
CN117375855A (zh) 一种异常检测方法、模型训练方法及相关设备
CN111459990B (zh) 对象处理方法、系统及计算机可读存储介质和计算机设备

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination
GR01 Patent grant
GR01 Patent grant