A Lightweight Model for Malicious Code Classification Based on Structural Reparameterisation and Large Convolutional Kernels

被引:2
|
作者
Li, Sicong [1 ]
Wang, Jian [1 ]
Song, Yafei [1 ]
Wang, Shuo [2 ]
Wang, Yanan [1 ]
机构
[1] AF Engn Univ, Air & Missile Def Coll, Xian 710051, Peoples R China
[2] Chinese Peoples Liberat Army PLA, Unit 95285, Guilin 541000, Peoples R China
基金
中国国家自然科学基金;
关键词
Malware variant detection; Convolutional neural network; Structurral reparameterisation; Large kernel convolution; Image size normalisation;
D O I
10.1007/s44196-023-00400-9
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
With the advancement of adversarial techniques for malicious code, malevolent attackers have propagated numerous malicious code variants through shell coding and code obfuscation. Addressing the current issues of insufficient accuracy and efficiency in malicious code classification methods based on deep learning, this paper introduces a detection strategy for malicious code, uniting Convolutional Neural Networks (CNNs) and Transformers. This approach utilizes deep neural architecture, incorporating a novel fusion module to reparametrize the structure, which mitigates memory access costs by eliminating residual connections within the network. Simultaneously, overparametrization during linear training time and significant kernel convolution techniques are employed to enhance network precision. In the data preprocessing stage, a pixel-based image size normalization algorithm and data augmentation techniques are utilized to remedy the loss of texture information in the malicious code image scaling process and class imbalance in the dataset, thereby enhancing essential feature expression and alleviating model overfitting. Empirical evidence substantiates this method has improved accuracy and the most recent malicious code detection technologies.
引用
收藏
页数:18
相关论文
共 50 条
  • [1] A Lightweight Model for Malicious Code Classification Based on Structural Reparameterisation and Large Convolutional Kernels
    Sicong Li
    Jian Wang
    Yafei Song
    Shuo Wang
    Yanan Wang
    [J]. International Journal of Computational Intelligence Systems, 17
  • [2] TriCh-LKRepNet: A large kernel convolutional malicious code classification network for structure reparameterisation and triple-channel mapping
    Li, Sicong
    Wang, Jian
    Song, Yafei
    Wang, Shuo
    [J]. COMPUTERS & SECURITY, 2024, 144
  • [3] Lightweight Malicious Code Classification Method Based on Improved SqueezeNet
    Li, Li
    Kong, Youran
    Zhang, Qing
    [J]. CMC-COMPUTERS MATERIALS & CONTINUA, 2024, 78 (01): : 551 - 567
  • [4] Decompiled APK based malicious code classification
    Mateless, Roni
    Rejabek, Daniel
    Margalit, Oded
    Moskovitch, Robert
    [J]. FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2020, 110 : 135 - 147
  • [5] Malicious Code Classification Method Based on Deep Forest
    Lu, Xi-Dong
    Duan, Zhe-Min
    Qian, Ye-Kui
    Zhou, Wei
    [J]. Ruan Jian Xue Bao/Journal of Software, 2020, 31 (05): : 1454 - 1464
  • [6] RepRCNN: A Structural Reparameterisation Convolutional Neural Network Object Detection Algorithm Based on Branch Matching
    Li, Xudong
    Lv, Xinyao
    Sun, Linghui
    Zhang, Jingzhi
    Lan, Ruoming
    [J]. ELECTRONICS, 2023, 12 (19)
  • [7] An Efficient Malicious Code Detection System Based on Convolutional Neural Networks
    Cao, Dongzhi
    Zhang, Xinglan
    Ning, Zhenhu
    Zhao, Jianfeng
    Xue, Fei
    Yang, Yongli
    [J]. PROCEEDINGS OF 2018 THE 2ND INTERNATIONAL CONFERENCE ON COMPUTER SCIENCE AND ARTIFICIAL INTELLIGENCE (CSAI 2018) / 2018 THE 10TH INTERNATIONAL CONFERENCE ON INFORMATION AND MULTIMEDIA TECHNOLOGY (ICIMT 2018), 2018, : 86 - 89
  • [8] Visualization Feature and CNN Based Homology Classification of Malicious Code
    CHU Qianfeng
    LIU Gongshen
    ZHU Xinyu
    [J]. Chinese Journal of Electronics, 2020, 29 (01) : 154 - 160
  • [9] Visualization Feature and CNN Based Homology Classification of Malicious Code
    Chu, Qianfeng
    Liu, Gongshen
    Zhu, Xinyu
    [J]. CHINESE JOURNAL OF ELECTRONICS, 2020, 29 (01) : 154 - 160
  • [10] Malicious Code Family Classification Method Based on Vision Transformer
    Chen, Shi
    Liu, Ying
    Hu, Wei
    Liu, Jianyi
    Gao, Yating
    Lin, Bingjie
    [J]. 2022 IEEE 10TH INTERNATIONAL CONFERENCE ON INFORMATION, COMMUNICATION AND NETWORKS (ICICN 2022), 2022, : 704 - 709