节点文献

基于迁移学习的恶意软件分类

Malware classification based on transfer learning

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 高泽安付东来李众薛震洪军

【Author】 GAO Ze-an;FU Dong-lai;LI Zhong;XUE Zhen;HONG Jun;School of Software,North University of China;School of Mathematics,North University of China;

【通讯作者】 付东来;

【机构】 中北大学软件学院中北大学数学学院

【摘要】 为了解决现有恶意软件分类方法在对抗检测能力和模型训练成本方面的不足,提出了一种基于迁移学习技术的轻量级恶意软件分类模型——MalDTL。将恶意软件的二进制文件转换为包含字节和信息熵特征的224×224的RGB图像。根据Windows操作系统的PE文件格式规范,在生成的RGB图像上附加标签框,形成RGBB图像。基于准确率、召回率、精确率和F1值,比较了VGG16、VGG19、InceptionV3和ResNet50这4种模型,选择ResNet50作为基准模型来构建MalDTL。实验结果表明,该模型在控制成本的同时,显著提高了恶意软件分类的准确性和对抗检测能力。

【Abstract】 To address the limitations of existing malware classification methods regarding adversarial detection capabilities and model training costs, a lightweight malware classification model based on transfer learning technology named MalDTL was proposed. The binary files of the malware were converted into 224×224 RGB images, incorporating byte and information entropy features. According to the PE file format specification for the Windows operating system, label boxes were added to these RGB images to create RGBB images. Based on the metrics of accuracy, recall, precision, and F1-score, four models—VGG16, VGG19, InceptionV3, and ResNet50—were compared, and ResNet50 was selected as the baseline model for constructing MalDTL. Experimental results demonstrate that the proposed model significantly enhances the accuracy of malware classification and adversarial detection capabilities while maintaining cost-effectiveness.

【基金】 山西省自然科学基金面上基金项目(202203021211088)
  • 【文献出处】 计算机工程与设计 ,Computer Engineering and Design , 编辑部邮箱 ,2026年02期
  • 【分类号】TP309;TP18;TP391.41
  • 【下载频次】41
节点文献中: 

本文链接的文献网络图示:

本文的引文网络