Proceedings

Heterogeneous Graph Matching Networks for Unknown Malware Detection

Download BibTeX Cite

Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence. Main track. Pages 3762-3770.

https://doi.org/10.24963/ijcai.2019/522

Abstract

Information systems have widely been the target of malware attacks. Traditional signature-based malicious program detection algorithms can only detect known malware and are prone to evasion techniques such as binary obfuscation, while behavior-based approaches highly rely on the malware training samples and incur prohibitively high training cost. To address the limitations of existing techniques, we propose MatchGNet, a heterogeneous Graph Matching Network model to learn the graph representation and similarity metric simultaneously based on the invariant graph modeling of the program's execution behaviors. We conduct a systematic evaluation of our model and show that it is accurate in detecting malicious program behavior and can help detect malware attacks with less false positives. MatchGNet outperforms the state-of-the-art algorithms in malware detection by generating 50% less false positives while keeping zero false negatives.

  1. [Babić et al., 2011] Domagoj Babić, Daniel Reynaud, and Dawn Song. Malware analysis with tree automata infer- ence. In Computer Aided Verification, pages 116-131, 2011.
  2. [Bernardi et al., 2018] Mario Luca Bernardi, Marta Cimitile, Damiano Distante, Fabio Martinelli, and Francesco Mer- caldo. Dynamic malware detection and phylogeny analysis using process mining. International Journal of Informa- tion Security, Jun 2018.
  3. [Cao et al., 2018] Cheng Cao, Zhengzhang Chen, James Caverlee, Lu-An Tang, Chen Luo, and Zhichun Li. Behavior-based community detection: Application to host assessment in enterprise information networks. In CIKM, pages 1977-1985, 2018.
  4. [Chen et al., 2016] Ting Chen, Lu-An Tang, Yizhou Sun, Zhengzhang Chen, and Kai Zhang. Entity embedding- based anomaly detection for heterogeneous categorical events. In IJCAI, 2016.
  5. [Chen et al., 2018] Jie Chen, Tengfei Ma, and Cao Xiao. Fastgcn: fast learning with graph convolutional net- works via importance sampling. arXiv preprint arXiv:1801.10247, 2018.
  6. [Cheng et al., 2016] Wei Cheng, Kai Zhang, Haifeng Chen, Guofei Jiang, Zhengzhang Chen, and Wei Wang. Rank- ing causal anomalies via temporal and dynamical analysis on vanishing correlations. In SIGKDD, pages 805-814. ACM, 2016.
  7. [Damodaran et al., 2017] Anusha Damodaran, Fabio Di Troia, Corrado Aaron Visaggio, Thomas H. Austin, and Mark Stamp. A comparison of static, dynamic, and hy- brid analysis for malware detection. Journal of Computer Virology and Hacking Techniques, 13(1):1-12, Feb 2017.
  8. [Das and Schneider, 2007] Kaustav Das and Jeff Schneider. Detecting anomalous records in categorical datasets. In SIGKDD, pages 220-229, 2007.
  9. [David et al., 2017] Baptiste David, Eric Filiol, and Kévin Gallienne. Structural analysis of binary executable head- ers for malware detection optimization. Journal of Com- puter Virology and Hacking Techniques, 13(2):87-93, May 2017.
  10. [Defferrard et al., 2016] Michaël Defferrard, Xavier Bres- son, and Pierre Vandergheynst. Convolutional neural net- works on graphs with fast localized spectral filtering. In NIPS, pages 3844-3852, 2016.
  11. [Dong et al., 2017] Boxiang Dong, Zhengzhang Chen, Hui (Wendy) Wang, Lu-An Tang, Kai Zhang, Ying Lin, Zhichun Li, and Haifeng Chen. Efficient discovery of abnormal event sequences in enterprise security systems. In CIKM, pages 707-715, 2017.
  12. [Fout et al., 2017] Alex Fout, Jonathon Byrd, Basir Shariat, and Asa Ben-Hur. Protein interface prediction using graph convolutional networks. In NIPS, pages 6530-6539, 2017.
  13. [Gilmer et al., 2017] Justin Gilmer, Samuel S Schoenholz, Patrick F Riley, Oriol Vinyals, and George E Dahl. Neural message passing for quantum chemistry. arXiv preprint arXiv:1704.01212, 2017.
  14. [Hamilton et al., 2017] Will Hamilton, Zhitao Ying, and Jure Leskovec. Inductive representation learning on large graphs. In NIPS, pages 1024-1034, 2017.
  15. [Harang and Kott, 2017] R. Harang and A. Kott. Burstiness of intrusion detection process: Empirical evidence and a modeling approach. IEEE Transactions on Information Forensics and Security, 2017.
  16. [Huang et al., 2017] Gao Huang, Zhuang Liu, Laurens Van Der Maaten, and Kilian Q Weinberger. Densely con- nected convolutional networks. In CVPR, pages 4700- 4708, 2017.
  17. [Jha et al., 2013] S. Jha, M. Fredrikson, M. Christodoresu, R. Sailer, and X. Yan. Synthesizing near-optimal malware specifications from suspicious behaviors. In 2013 8th In- ternational Conference on Malicious and Unwanted Soft- ware: "The Americas" (MALWARE), pages 41-50, Oct 2013.
  18. [Jyothsna and Prasad, 2011] V Jyothsna and V V Rama Prasad. A review of anomaly based intrusion detection systems. International Journal of Computer Applications, 28(7):26-35, 2011.
  19. [Kingma and Ba, 2014] Diederik P Kingma and Jimmy Ba. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980, 2014.
  20. [Kipf and Welling, 2016] Thomas N Kipf and Max Welling. Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907, 2016.
  21. [Lin et al., 2012] Shih-Wei Lin, Kuo-Ching Ying, Chou- Yuan Lee, and Zne-Jung Lee. An intelligent algo- rithm with feature selection and decision rules applied to anomaly intrusion detection. Applied Soft Computing, 12(10):3285-3290, 2012.
  22. [Lin et al., 2018] Ying Lin, Zhengzhang Chen, Cheng Cao, Lu-An Tang, Kai Zhang, Wei Cheng, and Zhichun Li. Col- laborative alert ranking for anomaly detection. In CIKM, pages 1987-1995, 2018.
  23. [Luo et al., 2018] Chen Luo, Zhengzhang Chen, and et al. TINET: learning invariant networks via knowledge trans- fer. In SIGKDD, pages 1890-1899, 2018.
  24. [Mao et al., 2019] Chengsheng Mao, Liang Yao, and Yuan Luo. Medgcn: Graph convolutional networks for multiple medical tasks. arXiv preprint arXiv:1904.00326, 2019.
  25. [Pal and Mitra, 1992] Sankar K Pal and Sushmita Mitra. Multilayer perceptron, fuzzy sets, and classification. IEEE Transactions on neural networks, 3(5):683-697, 1992.
  26. [Raymond et al., 2002] John W Raymond, Eleanor J Gar- diner, and Peter Willett. Rascal: Calculation of graph similarity using maximum common edge subgraphs. The Computer Journal, 45(6):631-644, 2002.
  27. [Rieck et al., 2008] Konrad Rieck, Thorsten Holz, Carsten Willems, Patrick Düssel, and Pavel Laskov. Learning and classification of malware behavior. In Detection of Intru- sions and Malware, and Vulnerability Assessment, pages 108-125, 2008.
  28. [Saracino et al., 2018] A. Saracino, D. Sgandurra, G. Dini, and F. Martinelli. Madam: Effective and efficient behavior-based android malware detection and prevention. IEEE Transactions on Dependable and Secure Computing, 15(1):83-97, Jan 2018.
  29. [Shasha et al., 2002] Dennis Shasha, Jason TL Wang, and Rosalba Giugno. Algorithmics and applications of tree and graph searching. In Proceedings of the twenty-first ACM SIGMOD-SIGACT-SIGART symposium on Princi- ples of database systems, pages 39-52. ACM, 2002.
  30. [Sun et al., 2011] Yizhou Sun, Jiawei Han, and et al. Path- sim: Meta path-based top-k similarity search in heteroge- neous information networks. VLDB Endowment, 2011.
  31. [Sun et al., 2014] Yi Sun, Yuheng Chen, Xiaogang Wang, and Xiaoou Tang. Deep learning face representation by joint identification-verification. In NIPS, pages 1988- 1996, 2014.
  32. [Tong et al., 2006] Hanghang Tong, Christos Faloutsos, and Jia-Yu Pan. Fast random walk with restart and its applica- tions. In ICDM, pages 613-622, 2006.
  33. [Velickovic et al., 2017] Petar Velickovic, Guillem Cucurull, Arantxa Casanova, Adriana Romero, Pietro Liò, and Yoshua Bengio. Graph attention networks. CoRR, abs/1710.10903(2), 2017.
  34. [Wang et al., 2017] Shen Wang, Lifang He, Bokai Cao, Chun-Ta Lu, Philip S. Yu, and Ann B. Ragin. Structural deep brain network mining. In SIGKDD, pages 475-484, 2017.
  35. [Wang et al., 2019a] Shen Wang, Zhengzhang Chen, Ding Li, Zhichun Li, Lu-An Tang, Jingchao Ni, Junghwan Rhee, Haifeng Chen, and Philip S Yu. Attentional heterogeneous graph neural network: Application to program reidentifi- cation. In SDM, pages 693-701, 2019.
  36. [Wang et al., 2019b] Shen Wang, Zhengzhang Chen, Jingchao Ni, Xiao Yu, Zhichun Li, Haifeng Chen, and Philip S Yu. Adversarial defense framework for graph neural network. arXiv preprint arXiv:1905.03679, 2019.
  37. [Weinberger and Saul, 2009] Kilian Q Weinberger and Lawrence K Saul. Distance metric learning for large mar- gin nearest neighbor classification. Journal of Machine Learning Research, 10(Feb):207-244, 2009.
  38. [Willett et al., 1998] Peter Willett, John M Barnard, and Geoffrey M Downs. Chemical similarity searching. Journal of chemical information and computer sciences, 38(6):983-996, 1998.
  39. [Wu et al., 2019] Zonghan Wu, Shirui Pan, Fengwen Chen, Guodong Long, Chengqi Zhang, and Philip S Yu. A comprehensive survey on graph neural networks. arXiv preprint arXiv:1901.00596, 2019.
  40. [Yan et al., 2004] Xifeng Yan, Philip S Yu, and Jiawei Han. Graph indexing: a frequent structure-based approach. In Proceedings of the 2004 ACM SIGMOD international con- ference on Management of data, pages 335-346. ACM, 2004.
  41. [Ying et al., 2018] Rex Ying, Jiaxuan You, Christopher Mor- ris, Xiang Ren, William L Hamilton, and Jure Leskovec. Hierarchical graph representation learning withdifferen- tiable pooling. arXiv preprint arXiv:1806.08804, 2018.
  42. [Zagoruyko and Komodakis, 2015] Sergey Zagoruyko and Nikos Komodakis. Learning to compare image patches via convolutional neural networks. In CVPR, pages 4353- 4361, 2015.
  43. [Zhang et al., 2015] Kai Zhang, Qiaojun Wang, Zhengzhang Chen, Ivan Marsic, Vipin Kumar, Guofei Jiang, and Jie Zhang. From categorical to numerical: Multiple transitive distance learning and embedding. In SDM, pages 46-54, 2015.
  44. [Zitnik et al., 2018] Marinka Zitnik, Monica Agrawal, and Jure Leskovec. Modeling polypharmacy side effects with graph convolutional networks. Bioinformatics, 34(13):i457-i466, 2018.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards