Skip to main content

Vision-Based Malware Detection and Classification Using Lightweight Deep Learning Paradigm

  • Conference paper
  • First Online:
Computer Vision and Image Processing (CVIP 2019)

Part of the book series: Communications in Computer and Information Science ((CCIS,volume 1148))

Included in the following conference series:

Abstract

Cyber attackers develop new malicious software to attack their targets every year. Recent sophisticated malware targets financial data and steals the credentials of users. Security analysts design novel methods to defend against malware attacks, but, unfortunately, with the proliferation of newly discovered malware, the methods are inefficient. The need for automated detection of unknown and new malware is still challenging in cybersecurity research. Machine learning approaches are applied for malware detection, however, they require larger feature extraction and feature engineering. The proposed work analyzes and classifies malware based on visualization technique and employs Lightweight Convolutional Neural Networks deep learning model. The model performed better achieving an accuracy of 97% and 95% for the two malware datasets including benign samples. They did not require more hardware resources and model is trained with a low computational cost. The model was evaluated on Malimg dataset and Kaggle’s Microsoft Malware Classification Challenge (BIG 2015) dataset.

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD 84.99
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD 109.99
Price excludes VAT (USA)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free shipping worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

References

  1. Bergeron, J., Debbabi, M., Desharnais, J., Erhioui, M.M., Lavoie, Y., Tawbi, N.: Static detection of malicious code in executable programs. Int. J. Req. Eng. 2001(184–189), 79 (2001)

    Google Scholar 

  2. Nataraj, L., Karthikeyan, S., Jacob, G., Manjunath, B.S.: Malware images: Visualization and automatic classification. In: Proceedings of the 8th International Symposium on Visualization for Cyber Security, p. 4. ACM (2011)

    Google Scholar 

  3. Kolter, J.Z., Maloof, M.A.: Learning to detect and classify malicious executables in the wild. J. Mach. Learn. Res. 7, 2721–2744 (2006)

    MathSciNet  MATH  Google Scholar 

  4. Rieck, K., Trinius, P., Willems, C., Holz, T.: Automatic analysis of malware behavior using machine learning. J. Comput. Secur. 19(4), 639–668 (2011). https://doi.org/10.3233/jcs-2010-0410

    Article  Google Scholar 

  5. Islam, R., Tian, R., Batten, L., Versteeg, S.: Classification of malware based on integrated static and dynamic features. J. Netw. Comput. Appl. 36(2), 646–656 (2013). https://doi.org/10.1016/j.jnca.2012.10.004

    Article  Google Scholar 

  6. Kancherla, K., Mukkamala, S.: Image visualization based malware detection. In: 2013 IEEE Symposium on Computational Intelligence in Cyber Security (CICS), pp. 40–44. IEEE (2013)

    Google Scholar 

  7. Han, K., Kang, B., Im, E.G.: Malware analysis using visualized image matrices. Sci. World J. 2014, 1–15 (2014). https://doi.org/10.1155/2014/132713

    Google Scholar 

  8. LeCun, Y., Bengio, Y., Hinton, G.: Deep learning. Nature 521(7553), 436–444 (2015)

    Article  Google Scholar 

  9. Ni, S., Qian, Q., Zhang, R.: Malware identification using visualization images and deep learning. Comput. Secur. 77, 871–885 (2018). https://doi.org/10.1016/j.cose.2018.04.005

    Article  Google Scholar 

  10. Cui, Z., Xue, F., Cai, X., Cao, Y., Wang, G., Chen, J.: Detection of malicious code variants based on deep learning. IEEE Trans. Industr. Inf. 14(7), 3187–3196 (2018). https://doi.org/10.1109/tii.2018.2822680

    Article  Google Scholar 

  11. Vinayakumar, R., Alazab, M., Soman, K.P., Poornachandran, P., Venkatraman, S.: Robust intelligent malware detection using deep learning. IEEE Access 7, 46717–46738 (2019). https://doi.org/10.1109/access.2019.2906934

    Article  Google Scholar 

  12. Fu, J., Xue, J., Wang, Y., Liu, Z., Shan, C.: Malware visualization for fine-grained classification. IEEE Access 6, 14510–14523 (2018). https://doi.org/10.1109/access.2018.2805301

    Article  Google Scholar 

  13. Hardy, W., Chen, L., Hou, S., Ye, Y., Li, X.: DL4MD: a deep learning framework for intelligent malware detection. In: Proceedings of the International Conference on Data Mining (DMIN). The Steering Committee of the World Congress in Computer Science, Computer Engineering and Applied Computing (WorldComp), p. 61 (2016)

    Google Scholar 

  14. Agarap, A.F., Pepito, F.J.H.: Towards building an intelligent anti-malware system: a deep learning approach using support vector machine (SVM) for malware classification (2017). arXiv preprint arXiv:1801.00318

  15. Ronen, R., Radu, M., Feuerstein, C., Yom-Tov, E., Ahmadi, M.: Microsoft malware classification challenge (2018). arXiv preprint arXiv:1802.10135

  16. Kingma, D.P., Ba, J.: Adam: A method for stochastic optimization (2014). arXiv preprint arXiv:1412.6980

  17. Gibert, D.: Convolutional Neural Networks for Malware Classification. University Rovira i Virgili, Tarragona (2016)

    Google Scholar 

  18. Janocha, K., Czarnecki, W.M.: On loss functions for deep neural networks in classification (2017). arXiv preprint arXiv:1702.05659

  19. Agarap, A.F.: Deep learning using rectified linear units (ReLU) (2018). arXiv preprint arXiv:1803.08375

Download references

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to S. Geetha .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2020 Springer Nature Singapore Pte Ltd.

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Abijah Roseline, S., Hari, G., Geetha, S., Krishnamurthy, R. (2020). Vision-Based Malware Detection and Classification Using Lightweight Deep Learning Paradigm. In: Nain, N., Vipparthi, S., Raman, B. (eds) Computer Vision and Image Processing. CVIP 2019. Communications in Computer and Information Science, vol 1148. Springer, Singapore. https://doi.org/10.1007/978-981-15-4018-9_6

Download citation

  • DOI: https://doi.org/10.1007/978-981-15-4018-9_6

  • Published:

  • Publisher Name: Springer, Singapore

  • Print ISBN: 978-981-15-4017-2

  • Online ISBN: 978-981-15-4018-9

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics