arrow
Return

SeMalBERT: Semantic-based malware detection with bidirectional encoder representations from transformers

delete2024-02-01
delete10
PRE
AI
J
Junming Liu
Y
Yuntao Zhao *
Y
Yongxin Feng
Y
Yutao Hu
X
Xiangyu Ma
DOI:10.1016/j.jisa.2023.103690delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Machine learning models are widely used for identifying malicious software. However, existing models suffer from issues such as imprecise polysemous representations and a lack of contextual semantic representations, leading to the failure to recognize certain types of malicious software. In this paper, we propose a semantic-based intelligent malware detection model called SeMalBERT for identifying malicious software in Windows systems. Specifically, the model utilizes the API function sequences of malicious software as the learned features. Firstly, BERT is applied to accomplish word representation tasks and extract semantic information from the sequences. Secondly, a hybrid discriminator based on Convolutional Neural Network (CNN) and Long Short-Term Memory (LSTM) is used to explore the chaining relationships between functions. Lastly, an external attention mechanism is added after the LSTM to enable the model to better focus on key information within the text. Experimental results demonstrate that SeMalBERT outperforms existing malware detection techniques in terms of accuracy, F1 score, and loss function value on a general dataset.
Keywords:
Cyber security
BERT
Malware
CNN
LSTM

Journal

Journal of Information Security and Applications cover
Journal of Information Security and Applications
IF:
3.7
Papers:
1.9K
Citations:
4.9K

Organization

S
Shenyang Ligong University
Scholars:
2.1K
Papers: 1.2K
Citations: 743