基于图卷积网络和主题模型的恶意代码分类

摘要

本发明提出一种基于图卷积网络和主题模型的恶意代码同源性方法。本方法可以自动地从恶意代码中提取函数调用图和函数指令分布，使用图卷积网络对函数调用图进行嵌入，使用主题模型对指令分布进行变换和降维。主要包括以下几个步骤：(1)提取恶意代码的函数调用图；(2)提取恶意代码的函数指令分布；(3)选取最佳的主题数量；(4)将函数指令分布变换为函数主题分布；(5)将函数主题分布和函数调用图输入到分类模型中，训练模型；(6)将训练好的模型用于判定对恶意代码的家族分类。