| Title | Pay Attention: Improving Classification of PE Malware Using Attention Mechanisms Based on System Call Analysis |
| Publication Type | Conference Paper |
| Year of Publication | 2021 |
| Authors | Or-Meir, Ori, Cohen, Aviad, Elovici, Yuval, Rokach, Lior, Nissim, Nir |
| Conference Name | 2021 International Joint Conference on Neural Networks (IJCNN) |
| Keywords | Human Behavior, machine learning algorithms, Malware, malware analysis, malware classification, Metrics, Neural networks, Prediction algorithms, Predictive Metrics, privacy, pubcrawl, resilience, Resiliency, static analysis, Training, Transformers |
| Abstract | Malware poses a threat to computing systems worldwide, and security experts work tirelessly to detect and classify malware as accurately and quickly as possible. Since malware can use evasion techniques to bypass static analysis and security mechanisms, dynamic analysis methods are more useful for accurately analyzing the behavioral patterns of malware. Previous studies showed that malware behavior can be represented by sequences of executed system calls and that machine learning algorithms can leverage such sequences for the task of malware classification (a.k.a. malware categorization). Accurate malware classification is helpful for malware signature generation and is thus beneficial to antivirus vendors; this capability is also valuable to organizational security experts, enabling them to mitigate malware attacks and respond to security incidents. In this paper, we propose an improved methodology for malware classification, based on analyzing sequences of system calls invoked by malware in a dynamic analysis environment. We show that adding an attention mechanism to a LSTM model improves accuracy for the task of malware classification, thus outperforming the state-of-the-art algorithm by up to 6%. We also show that the transformer architecture can be used to analyze very long sequences with significantly lower time complexity for training and prediction. Our proposed method can serve as the basis for a decision support system for security experts, for the task of malware categorization. |
| DOI | 10.1109/IJCNN52387.2021.9533481 |
| Citation Key | or-meir_pay_2021 |
Pay Attention: Improving Classification of PE Malware Using Attention Mechanisms Based on System Call Analysis