MDFA: A Malware Detection Framework Based on Attention Mechanism with Bi-LSTM

As the existing malware intelligent detection methods have shortcomings and low accuracy of manual feature extraction and feature processing, a Malware Detection Framework with Attention mechanism based on Bi-directional Long Short-Term Memory (MDFA) was proposed. To training this method, We have collected 15, 893 malware and 10, 959 benign program samples and constructed a dataset. Then, the sample data was labeled and sanitized by determining the results based on VirusTotal analysis. Finally, the Bi-directional Long Short-Term Memory model injected into the attention mechanism was built and trained to complete the malware detection. Experimental results show that the accuracy of MDFA was improved by at least 2. 3%, and the recall rate was improved significantly compared with the existing methods, verifying the effectiveness of MDFA.