skip to main content
10.1145/3469213.3470713acmotherconferencesArticle/Chapter ViewAbstractPublication PagesicaiisConference Proceedingsconference-collections
research-article

A Malware Detection Method Based on Machine Learning and Ensemble of Regression Trees

Published: 18 August 2021 Publication History

Abstract

In the context of the current large number of malicious codes, the detection and protection of malicious codes is particularly important. In recent years, a method of using deep learning to detect malicious code has emerged. Thus, in this paper, we propose a new detection method that converts binary files of malicious code into decimal arrays and use 1-D CNN to perform classification and recognition. Aiming at the imbalance in the number of code families, we choose xgboost, which performs well in the classification prediction competition. We conduct experiments on 9,458 malware samples from 25 different malware families in the Vision Research Lab. The experimental results show that our classification prediction reaches 97% accuracy.

References

[1]
Z. Cui, F. Xue, X. Cai, “Detection of Malicious Code Variants Based on Deep Learning”. IEEE Transactions on Industrial Informatics, vol. 14, no. 7, pp. 3187-3196, 2018.
[2]
M. Ozsoy, K. Khasawneh, C. Donovick, “Hardware-based Malware Detection using Low-Level Architectural Features”. IEEE Transactions on Computers, vol. 65, no. 11, pp. 3332–3344, 2016.
[3]
H. Minh, D. Nguyen, M. Xuan, “Auto-detection of sophisticated malware using lazy-binding control flow graph and deep learning”. Computers & Security, vol. 76, 2018.
[4]
A. Moser, C. Kruegel, E. Kirda, “Limits of Static Analysis for Malware Detection”. 23rd Annual Computer Security Applications Conference, pp. 421-300, 2007.
[5]
W. Tobias, C. Aleksander, O. Martin, “Leveraging Compression-Based Graph Mining for Behavior-Based Malware Detection”. IEEE Transactions on Dependable and Secure Computing, vol. 16, no. 1, pp. 99-112, 2019.
[6]
A. Saracino, D. Sgandurra, G. Dini, “Madam: Effective and Efficient Behavior-based Android Malware Detection and Prevention”. IEEE Transactions on Dependable and Secure Computing, vol. 15, no. 1, pp. 83-97, 2018.
[7]
Y. Ye, L. Chen, S. Hou, “DeepAM: a heterogeneous deep learning framework for intelligent malware detection”. Knowledge and Information Systems, vol. 54, no. 2, pp. 265-285, 2018.
[8]
N. Lakshmanan, Y. Vinod, P. Phillip, “A Comparative Assessment of Malware Classification Using Binary Texture Analysis and Dynamic Analysis”. Proceedings of the 4th ACM Workshop on Security and Artificial Intelligence, pp. 21-30, 2011.
[9]
N. Lakshmanan, K. Shanmugavadivel, Gregoire, “Malware Images: Visualization and Automatic Classification”. Proceedings of the 8th International Symposium on Visualization for Cyber Security, no. 4, 2011.

Cited By

View all
  • (2024)A systematic literature review on Windows malware detectionJournal of Systems and Software10.1016/j.jss.2023.111921209:COnline publication date: 14-Mar-2024
  • (2023)Classification of Malware Using Deep Learning: A Study2023 IEEE International Carnahan Conference on Security Technology (ICCST)10.1109/ICCST59048.2023.10474230(1-6)Online publication date: 11-Oct-2023

Recommendations

Comments

Information & Contributors

Information

Published In

cover image ACM Other conferences
ICAIIS 2021: 2021 2nd International Conference on Artificial Intelligence and Information Systems
May 2021
2053 pages
ISBN:9781450390200
DOI:10.1145/3469213
Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than ACM must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]

Publisher

Association for Computing Machinery

New York, NY, United States

Publication History

Published: 18 August 2021

Permissions

Request permissions for this article.

Check for updates

Author Tags

  1. Cnn
  2. deep learning
  3. malware
  4. xgboost

Qualifiers

  • Research-article
  • Research
  • Refereed limited

Conference

ICAIIS 2021

Contributors

Other Metrics

Bibliometrics & Citations

Bibliometrics

Article Metrics

  • Downloads (Last 12 months)8
  • Downloads (Last 6 weeks)0
Reflects downloads up to 05 Mar 2025

Other Metrics

Citations

Cited By

View all
  • (2024)A systematic literature review on Windows malware detectionJournal of Systems and Software10.1016/j.jss.2023.111921209:COnline publication date: 14-Mar-2024
  • (2023)Classification of Malware Using Deep Learning: A Study2023 IEEE International Carnahan Conference on Security Technology (ICCST)10.1109/ICCST59048.2023.10474230(1-6)Online publication date: 11-Oct-2023

View Options

Login options

View options

PDF

View or Download as a PDF file.

PDF

eReader

View online with eReader.

eReader

HTML Format

View this article in HTML Format.

HTML Format

Figures

Tables

Media

Share

Share

Share this Publication link

Share on social media