Representation-Centric Approach for Android Malware Classification: Interpretability-Driven Feature Engineering on Function Call Graphs
Source abstract
Abstract
The existing research on Android malware detection using graph neural networks (GNNs) has largely focused on architectural improvements, while input node feature representations have received less systematic attention. This study adopts a representation-centric approach to enhance function call graph (FCG)-based malware classification through interpretability-driven feature engineering. We propose a dual-level structural feature framework integrating local topological patterns with global graph-level properties. The initial feature set comprises 13 dimensions: five local degree profile (LDP) features and eight global structural features capturing community structure, execution flow, and connectivity patterns. To mitigate the curse of dimensionality, we apply an interpretability-driven selection using integrated gradients (IG), gradient-weighted class activation mapping (GradCAM), and Shapley additive explanations (SHAP), yielding an optimized seven-dimensional subset. Experiments on the MalNet-Tiny benchmark demonstrate that the proposed approach achieves 94.47 ± 0.25% accuracy with jumping knowledge GraphSAGE (JK-GraphSAGE), improving the LDP-only baseline by 0.32 percentage points while reducing feature dimensionality by 46%. The selected features exhibit consistent importance across four GNN architectures and multiple message-passing layers, demonstrating model-agnostic effectiveness. The results reveal that aggregation mechanisms critically influence feature utility, highlighting the necessity of interpretability-guided design for robust malware detection. This work provides a systematic methodology for feature engineering in graph-based security applications.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Advanced Malware Detection Techniques · Software Engineering Research · Spam and Phishing Detection
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Information Security, Pai Chai University, Daejeon 35345, Republic of Korea
Funding
Institute of Information & Communications Technology Planning & Evaluation (IITP)-Innovative Human Resource Development for Local Intellectualization program grant funded by the Korean government
References (27)
Show source-provided references
1. Zhou, Y., and Jiang, X. (2012, January 20–23). Dissecting Android Malware: Characterization and Evolution. Proceedings of the 2012 IEEE Symposium on Security and Privacy, San Francisco, CA, USA.10.1109/SP.2012.16 ↗
2. Arp, D., Spreitzenbarth, M., Hubner, M., Gascon, H., and Rieck, K. (2014, January 23–26). DREBIN: Effective and Explainable Detection of Android Malware in Your Pocket. Proceedings of the 21st Annual Network and Distributed System Security Symposium (NDSS), San Diego, CA, USA.10.14722/ndss.2014.23247 ↗
3. Malware Analysis and Classification: A Survey
4. Kipf, T.N., and Welling, M. (2017, January 24–26). Semi-Supervised Classification with Graph Convolutional Networks. Proceedings of the 5th International Conference on Learning Representations (ICLR), Toulon, France.
5. Hamilton, W.L., Ying, R., and Leskovec, J. (2017, January 4–9). Inductive Representation Learning on Large Graphs. Proceedings of the 31st International Conference on Neural Information Processing Systems (NeurIPS), Long Beach, CA, USA.
6. Xu, K., Hu, W., Leskovec, J., and Jegelka, S. (2019, January 6–9). How Powerful are Graph Neural Networks?. Proceedings of the 7th International Conference on Learning Representations (ICLR), New Orleans, LA, USA.
7. Xu, K., Li, C., Tian, Y., Sonobe, T., Kawarabayashi, K., and Jegelka, S. (2018, January 10–15). Representation Learning on Graphs with Jumping Knowledge Networks. Proceedings of the 35th International Conference on Machine Learning (ICML), Stockholm, Sweden.
8. Freitas, S., Dong, Y., Neil, J., and Chau, D.H. (2021, January 6–14). A Large-Scale Database for Graph Representation Learning. Proceedings of the 35th Conference on Neural Information Processing Systems (NeurIPS) Datasets and Benchmarks Track, Virtual. Available online: https://openreview.net/forum?id=1xDTDk3XPW.
9. Lo, W.W., Layeghy, S., Sarhan, M., Gallagher, M., and Portmann, M. (2022, January 22–24). Graph Neural Network-based Android Malware Classification with Jumping Knowledge. Proceedings of the 2022 IEEE Conference on Dependable and Secure Computing (DSC), Edinburgh, UK.10.1109/DSC54232.2022.9888878 ↗
10. Niu, W., Cao, R., Zhang, X., Ding, K., Zhang, K., and Li, T. (2020). OpCode-level Function Call Graph Based Android Malware Classification Using Deep Learning. Sensors, 20.10.3390/s20133645 ↗
11. GDroid: Android Malware Detection and Classification with Graph Convolutional Network10.1016/j.cose.2021.102264 ↗
12. Representation Learning: A Review and New Perspectives10.1109/TPAMI.2013.50 ↗
13. A Comprehensive Survey of Graph Embedding: Problems, Techniques, and Applications10.1109/TKDE.2018.2807452 ↗
14. Sundararajan, M., Taly, A., and Yan, Q. (2017, January 6–11). Axiomatic Attribution for Deep Networks. Proceedings of the 34th International Conference on Machine Learning (ICML), Sydney, Australia.
15. Selvaraju, R.R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., and Batra, D. (2017, January 22–29). Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization. Proceedings of the IEEE International Conference on Computer Vision (ICCV), Venice, Italy.10.1109/ICCV.2017.74 ↗
16. Lundberg, S.M., and Lee, S.I. (2017, January 4–9). A Unified Approach to Interpreting Model Predictions. Proceedings of the 31st International Conference on Neural Information Processing Systems (NeurIPS), Long Beach, CA, USA.
17. Shokouhinejad, H., Razavi-Far, R., Higgins, G., and Ghorbani, A.A. (2025). Enhancing GNN Explanations for Malware Detection with Dual Subgraph Matching. Mach. Learn. Knowl. Extr., 8.10.3390/make8010002 ↗
18. Yang, H., Zhou, Y., Ji, X., Liu, Z., Tian, Z., Tang, Q., and Shi, Y. (2025). Advancing Graph Neural Networks for Complex Relational Learning: A Multi-Scale Heterogeneity-Aware Framework with Adversarial Robustness and Interpretable Analysis. Mathematics, 13.10.3390/math13182956 ↗
19. A Comparison of Static, Dynamic, and Hybrid Analysis for Malware Detection10.1007/s11416-015-0261-z ↗
20. Blasing, T., Batyuk, L., Schmidt, A.D., Camtepe, S.A., and Albayrak, S. (2010, January 19–20). An Android Application Sandbox System for Suspicious Software Detection. Proceedings of the 5th International Conference on Malicious and Unwanted Software (MALWARE), Nancy, France.10.1109/MALWARE.2010.5665792 ↗
21. Cai, L., and Wang, Y. (2018). A Simple Yet Effective Baseline for Non-Attributed Graph Classification. arXiv.
22. Finding and Evaluating Community Structure in Networks10.1103/PhysRevE.69.026113 ↗
23. Fast Unfolding of Communities in Large Networks10.1088/1742-5468/2008/10/P10008 ↗
24. Android Malware Familial Classification and Representative Sample Selection via Frequent Subgraph Analysis10.1109/TIFS.2018.2806891 ↗
25. Lightweight, Obfuscation-Resilient Detection and Family Identification of Android Malware
26. Depth-First Search and Linear Graph Algorithms10.1137/0201010 ↗
27. Scikit-learn: Machine Learning in Python