3 个月前

一种用于恶意软件分类的分层卷积神经网络

一种用于恶意软件分类的分层卷积神经网络

摘要

恶意软件检测与分类是一个具有挑战性且持续活跃的研究领域。其中一项关键难题在于如何对恶意可执行文件进行最优处理与预处理,以便输入机器学习算法。现有文献中的新方法通常将可执行文件视为字节序列或汇编语言指令序列,但这些方法未充分考虑程序的层次化结构。实际上,可执行文件表现出多层次的空间相关性:相邻的指令在空间上具有相关性,但这种相关性并非始终成立。函数调用和跳转指令会将程序控制流转移到指令流中的另一位置,而这种控制流的不连续性在将二进制文件视为字节序列时依然存在。此外,若地址被正确重定位,函数的排列可能呈现随机性。为应对上述问题,本文提出一种用于恶意软件分类的层次化卷积网络(Hierarchical Convolutional Network, HCN)。该网络在助记符层级(mnemonic-level)与函数层级(function-level)分别应用两层卷积模块,从而在构建恶意软件表征时,能够从两个层次提取类似n-gram的特征。我们在微软恶意软件分类挑战赛(Microsoft Malware Classification Challenge)发布的数据集上对所提HCN方法进行了验证,结果表明其性能优于文献中几乎所有的深度学习方法。

基准测试

基准方法指标
malware-classification-on-microsoft-malwareCNN+BiLSTM
Accuracy (10-fold): 0.9820
LogLoss: 0.0744
Macro F1 (10-fold): 0.9605
malware-classification-on-microsoft-malwareHierarchical Attention Network
Accuracy (10-fold): 0.9742
LogLoss: 0.0933
Macro F1 (10-fold): 0.9468
malware-classification-on-microsoft-malwareMalConv
Accuracy (10-fold): 0,9641
LogLoss: 0.3071
Macro F1 (10-fold): 0.8902
malware-classification-on-microsoft-malwareDeepConv
Accuracy (10-fold): 0.9756
LogLoss: 0.1602
Macro F1 (10-fold): 0.9071
malware-classification-on-microsoft-malwareHierarchical Convolutional Network
Accuracy (10-fold): 0.9913
LogLoss: 0.0419
Macro F1 (10-fold): 0.9830

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
一种用于恶意软件分类的分层卷积神经网络 | 论文 | HyperAI超神经