scholarly journals A malicious code family classification method based on self-attention mechanism

2021 ◽  
Vol 2010 (1) ◽  
pp. 012066
Author(s):  
Ru Zhang ◽  
Xinjian Zhao ◽  
Jiaqi Li ◽  
Song Zhang ◽  
Zhijie Shang
2020 ◽  
Vol 2020 ◽  
pp. 1-11
Author(s):  
Jun Chen ◽  
Shize Guo ◽  
Xin Ma ◽  
Haiying Li ◽  
Jinhong Guo ◽  
...  

Since the number of malware is increasing rapidly, it continuously poses a risk to the field of network security. Attention mechanism has made great progress in the field of natural language processing. At the same time, there are many research studies based on malicious code API, which is also like semantic information. It is a worthy study to apply attention mechanism to API semantics. In this paper, we firstly study the characters of the API execution sequence and classify them into 17 categories. Secondly, we propose a novel feature extraction method based on API execution sequence according to its semantics and structure information. Thirdly, based on the API data characteristics and attention mechanism features, we construct a detection framework SLAM based on local attention mechanism and sliding window method. Experiments show that our model achieves a better performance, which is a higher accuracy of 0.9723.


Author(s):  
Shuaicong Hu ◽  
Wenjie Cai ◽  
Tijie Gao ◽  
Jiajun Zhou ◽  
Mingjie Wang

Abstract Objective: Electrocardiography is a common method for screening cardiovascular diseases. Accurate heartbeat classification assists in diagnosis and has attracted great attention. In this paper, we proposed an automatic heartbeat classification method based on a transformer neural network using a self-attention mechanism. Approach: An adaptive heartbeat segmentation method was designed to selectively focus on the time-dependent representation of heartbeats. A one-dimensional convolution layer was used to embed wave characteristics into symbolic representations, and then, a transformer block using multi-head attention was applied to deal with the dependence of wave-embedding. The model was trained and evaluated using the MIT-BIH arrhythmia database (MIT-DB). To improve the model performance, the model pre-trained on MIT-BIH supraventricular arrhythmia database (MIT-SVDB) was used and fine-tuned on MIT-DB. Main results: The proposed method was verified using the MIT-DB for two groups. In the first group, our method attained F1 scores of 0.86 and 0.96 for the supraventricular ectopic beat (SVEB) class and ventricular ectopic beat (VEB) class, respectively. In the second group, our method achieved an average F1 value of 99.83% and better results than other state-of-the-art methods. Significance: We proposed a novel heartbeat classification method based on a transformer model. This method provides a new solution for real-time electrocardiogram heartbeat classification, which can be applied to wearable devices.


2019 ◽  
Vol 9 (20) ◽  
pp. 4205 ◽  
Author(s):  
Yanchen Qiao ◽  
Bin Zhang ◽  
Weizhe Zhang ◽  
Arun Kumar Sangaiah ◽  
Hualong Wu

Currently, many cyberattacks use the Domain Generation Algorithm (DGA) to generate random domain names, so as to maintain communication with the Communication and Control (C&C) server. Discovering DGA domain names in advance could help to detect attacks and response in time. However, in recent years, the General Data Protection Regulation (GDPR) has been promulgated and implemented, and the method of DGA classification based on the context information, such as the WHOIS (the information about the registered users or assignees of the domain name) , is no longer applicable. At the same time, acquiring the DGA algorithm by reversing malware samples encounters the problem of no malware samples for various reasons, such as fileless malware. We propose a DGA domain name classification method based on Long Short-Term Memory (LSTM) with attention mechanism. This method is oriented to the character sequence of the domain name, and it uses the LSTM combined with attention mechanism to construct the DGA domain name classifier to achieve the rapid classification of domain names. The experimental results show that the method has a good classification result.


2021 ◽  
Vol 1748 ◽  
pp. 042050
Author(s):  
Sitao Zeng ◽  
Yongchun Cao ◽  
Qiang Lin ◽  
Zhengxing Man ◽  
Tao Deng ◽  
...  

Sign in / Sign up

Export Citation Format

Share Document