(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210980514.2 (22)申请日 2022.08.16 (71)申请人 中国电信股份有限公司 地址 100033 北京市西城区金融大街31号 (72)发明人 段赟　刘东鑫　汪来富　董浩　 (74)专利代理 机构 北京律智知识产权代理有限 公司 11438 专利代理师 孙宝海 (51)Int.Cl. H04L 9/40(2022.01) H04L 41/16(2022.01) G06K 9/62(2022.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 加密流量检测模型的训练方法、 装置、 设备 及存储介质 (57)摘要 本发明提供了加密流量检测模型的训练方 法、 装置、 设备及存储介质， 该方法包括： 获取流 量样本数据， 所述流量样本数据包括流量数据以 及流量数据的分类标签； 对所述流量数据进行预 处理， 获得第一样本集合以及第二样本集合； 将 所述第一样本集合和所述第二样本集合中的多 个样本数据包转换为第一图像； 将所述第二样本 集合输入生成对抗网络模型中， 所述生成对抗网 络模型的生成器基于所述第二样本集合生成第 二图像； 将所述第二图像和所述第一图像， 输入 至分类器中， 获得分类器输出的分类结果； 根据 所述分类器输出的分类结果与所述分类标签的 比较， 调整所述分类器的参数。 本发明避免加密 流量检测任务中样本不平衡造成的应用效果不 佳。 权利要求书2页 说明书9页 附图5页 CN 115529159 A 2022.12.27 CN 115529159 A 1.一种加密流 量检测模型的训练方法， 其特 征在于， 包括： 获取流量样本数据， 所述 流量样本数据包括 流量数据以及流 量数据的分类标签； 对所述流量数据进行预处理， 获得第 一样本集合以及第 二样本集合， 其中， 所述第一样 本集合包括多个第一样本数据包， 所述第二样本集合包括多个第二样本数据包， 所述第一 样本集合中的第一样本数据包的数量大于所述第二样本集合的所述第二样本数据包的数 量； 将所述第一样本集 合和所述第二样本集 合中的多个样本数据包转换为第一图像； 将所述第二样本集合输入生成对抗网络模型中， 所述生成对抗网络模型的生成器基于 所述第二样本集合生成第二图像， 所述生成对抗网络模型的判别器用于识别所述第二图像 与对应的所述第一图像的差异； 将所述第二图像和所述第一图像， 输入至分类 器中， 获得分类 器输出的分类结果； 根据所述分类 器输出的分类结果与所述分类标签的比较， 调整所述分类 器的参数。 2.根据权利要求1所述的加密流量检测模型的训练方法， 其特征在于， 所述生成对抗网 络模型的生成器以所述生成对抗网络模型的判别器将所述第二图像识别为对应的第一图 像的概率大于设定阈值 为目标进行训练。 3.根据权利要求2所述的加密流量检测模型的训练方法， 其特征在于， 所述生成器用于 向所述第二样本集 合中的多个第二样本数据包增 加随机噪声， 以生成第二图像。 4.根据权利要求1所述的加密流量检测模型的训练方法， 其特征在于， 所述对所述流量 数据进行 预处理包括： 将第一流量数据和第二流量数据分别按会话层切分为多个第一样本数据包和多个第 二样本数据包； 将多个第一样本数据包和多个第二样本数据包分别按由大至小的顺序进行排序； 根据多个第一样本数据包中排序顺序中前N个第一样本数据包， 生成第一样本集合， N 为大于2的整数； 根据多个第二样本数据包中排序顺序中前N/2个第二样本数据包， 生成第二样本集 合。 5.根据权利要求4所述的加密流量检测模型的训练方法， 其特征在于， 所述将第 一流量 数据和第二流量数据分别按会话层切分为多个第一样本数据包和多个第二样本数据包之 后包括： 使用零地址替换 所述第一样本数据包和第二样本数据包中的地址信息 。 6.根据权利要求4所述的加密流量检测模型的训练方法， 其特征在于， 所述将第 一流量 数据和第二流量数据分别按会话层切分为多个第一样本数据包和多个第二样本数据包之 后包括： 将各所述第一样本数据包和所述第二样本数据包统一 为同一数据长度。 7.一种加密流 量检测方法， 其特 征在于， 包括： 获取待检测流 量数据； 对所述待检测流 量数据进行 预处理； 将预处理后的所述待检测流 量数据转换为第三图像； 将所述第三图像输入至由权利要求1至6任一项所述的加密流量检测模型的训练方法 训练获得的分类 器中， 获得分类 器输出的分类结果。权　利　要　求　书 1/2 页 2 CN 115529159 A 28.一种加密流 量检测模型的训练装置， 其特 征在于， 包括： 第一获取模块， 配置成获取流量样本数据， 所述流量样本数据包括流量数据以及流量 数据的分类标签； 第二获取模块， 配置成对所述流量数据进行预处理， 获得第一样本集合以及第二样本 集合， 其中， 所述第一样本集合包括多个第一样本数据包， 所述第二样本集合包括多个第二 样本数据包， 所述第一样本集合中的第一样本数据包的数量大于所述第二样本集合的所述 第二样本数据包的数量； 第一转换模块， 配置成将所述第 一样本集合和所述第 二样本集合中的多个样本数据包 转换为第一图像； 第一输入模块， 配置成将所述第二样本集合输入生成对抗网络模型中， 所述生成对抗 网络模型的生成器基于所述第二样本集合生成第二图像， 所述生成对抗网络模型的判别器 用于识别所述第二图像与对应的所述第一图像的差异； 第二输入模块， 配置成将所述第 二图像和所述第 一图像， 输入至分类器 中， 获得分类器 输出的分类结果； 调整模块， 配置成根据所述分类器输出的分类结果与所述分类标签的比较， 调整所述 分类器的参数。 9.一种加密流 量检测装置， 其特 征在于， 包括： 第三获取模块， 配置成获取待检测流 量数据； 预处理模块， 配置成对所述待检测流 量数据进行 预处理； 第二转换模块， 配置成将预处 理后的所述待检测流 量数据转换为第三图像； 第三输入模块， 配置成将所述第 三图像输入至由权利要求1至6任一项所述的加密流量 检测模型的训练方法训练获得的分类 器中， 获得分类 器输出的分类结果。 10.一种加密流 量检测模型的训练 处理设备， 其特 征在于， 包括： 处理器； 存储器， 其中存 储有所述处 理器的可 执行指令； 其中， 所述处 理器配置为经由执 行所述可执行指令来执 行： 权利要求1至 6任意一项所述加密流 量检测模型的训练方法； 和/或 权利要求7 所述的加密流 量检测方法。 11.一种计算机可读存 储介质， 用于存 储程序， 其特 征在于， 所述 程序被执 行时实现： 权利要求1至 6任意一项所述加密流 量检测模型的训练方法； 和/或 权利要求7 所述的加密流 量检测方法。权　利　要　求　书 2/2 页 3 CN 115529159 A 3