电子学报2024,Vol.52Issue(7) :2331-2340.DOI:10.12263/DZXB.20240162

TriCh-LKRepNet:融合三通道映射与结构重参数化的大核卷积恶意代码分类网络

TriCh-LKRepNet:A Large Kernel Convolutional Malicious Code Classification Network for Structure Reparameterisation and Triple-Channel Mapping

李思聪 王坚 宋亚飞 王硕
电子学报2024,Vol.52Issue(7) :2331-2340.DOI:10.12263/DZXB.20240162

TriCh-LKRepNet:融合三通道映射与结构重参数化的大核卷积恶意代码分类网络

TriCh-LKRepNet:A Large Kernel Convolutional Malicious Code Classification Network for Structure Reparameterisation and Triple-Channel Mapping

李思聪 1王坚 1宋亚飞 1王硕2
扫码查看

作者信息

  • 1. 空军工程大学防空反导学院,陕西西安 710051
  • 2. 中国人民解放军95285部队,广西桂林 541000
  • 折叠

摘要

随着网络威胁的日益严峻,恶意代码的检测与分类变得尤为关键.传统分析方法依赖手动特征提取,不仅耗时且难以跟上恶意代码的快速变异.相比之下,深度学习技术在恶意代码分类方面展现出巨大潜力.然而,模型复杂度和资源消耗仍是实际部署的难题.本研究提出了TriCh-LKRepNet(Triple-Channel Large Kernel Reparameteri-sation Network),该网络专注于轻量化设计,旨在确保检测性能的同时降低计算和内存需求.通过提出的三通道映射技术,将恶意代码的多维信息有效转换为图像通道,增强了特征的区分性.结合卷积神经网络(Convolutional Neural Networks,CNN)和Transformer的优势,设计了一个高效的深度学习架构,并通过重参数化技术优化了连接路径,以降低内存消耗并提升运行效率.此外,引入的线性训练时间过参数化和大卷积核技术进一步降低了模型的参数量和计算负担.通过实验证明,TriCh-LKRepNet在提升恶意代码分类精度的同时实现了模型的轻量化,与现有技术相比,展现出更佳的性能和更广泛的应用潜力,特别是在资源受限和需要实时检测的环境中,提供了一种有效的解决方案.

Abstract

With the increasing severity of cyber threats,the detection and classification of malicious code has become particularly critical. Traditional analysis methods rely on manual feature extraction,which is time-consuming and difficult to keep up with the rapid mutation of malicious code. In contrast,deep learning techniques show great potential for mali-cious code classification. However,model complexity and resource consumption are still challenges for practical deploy-ment. In this study,we propose the TriCh-LKRepNet (Triple-Channel Large Kernel Reparameterisation Network),which fo-cuses on lightweight design and aims to ensure detection performance while reducing computation and memory require-ments. Through the proposed three-channel mapping technique,the multi-dimensional information of malicious code is ef-fectively converted into image channels,which enhances the differentiation of features. An efficient deep learning architec-ture is designed by combining the advantages of convolutional neural networks (CNN) and Transformer,and the connection paths are optimized by a reparameterization technique to reduce the memory consumption and enhance the operation effi-ciency. In addition,the introduced linear training time over-parameterization and large convolutional kernel techniques fur-ther reduce the number of parameters and computational burden of the model. It is experimentally demonstrated that TriCh-LKRepNet improves the malicious code classification accuracy while realizing the model's lightweight,which shows better performance and wider application potential than existing techniques,especially in resource-constrained environments where real-time detection is required,providing an effective solution.

关键词

恶意代码分类/恶意代码可视化/结构重参数化/大卷积核/汇编信息/语义关系

Key words

malicious code classification/malicious code visualization/structural reparameterisation/large convolutional kernel/assembly information/semantic relations

引用本文复制引用

基金项目

国家自然科学基金(61806219)

国家自然科学基金(61703426)

国家自然科学基金(61876189)

陕西省自然科学基金(2021JM226)

陕西省高校科协青年人才托举计划(20190108)

陕西省高校科协青年人才托举计划(20220106)

陕西省创新能力支撑计划(2020KJXX-065)

出版年

2024
电子学报
中国电子学会

电子学报

CSTPCD北大核心
影响因子:1.237
ISSN:0372-2112
段落导航相关论文