Evaluating Out-of-Distribution Robustness in Graph-Based Android Malware Classification: A New Principled Benchmark
Not on the map. Why?
Abstract (arXiv)
While graph-based Android malware classifiers report strong benchmark accuracy of over 94%, their performance sharply decreases up to 45% when exposed to previously unseen variants of known malware families. In this work, we systematically investigate this critical yet overlooked challenge for real-world deployment by introducing a benchmarking suite designed to simulate two prevalent scenarios: MalNet-Tiny-Common for covariate shift, and MalNet-Tiny-Distinct for domain shift. We further identify an inherent limitation of existing benchmarks where input representation is limited to structure-only function call graphs, discarding the semantic signals needed for robust cross-distribution reasoning. To verify this, we propose a semantic enrichment framework that extends raw graph topology with function-level attributes, combining lightweight metadata with LLM-based code embeddings. Empirical evaluations confirm the effectiveness of our data-centric methodology, with which classification performs better under distribution shift compared to model-based approaches, and consistently further enhances robustness when used in conjunction. We release our precomputed datasets alongside an extensible pipeline implementation, laying the groundwork for more resilient malware detection systems in evolving threat environments.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 32 of its 53 references that are in the map. First posted on arXiv 8 Aug 2025.
Builds on
32 of 53 references.
References · 32
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2025Qwen3 Technical Reportcited by 8,820
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2018How Powerful are Graph Neural Networks?cited by 4,393
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2022Recipe for a General, Powerful, Scalable Graph Transformercited by 713
- 2024Parameter-Efficient Fine-Tuning for Large Models: A Comprehensive Surveycited by 656
- 2022UniXcoder: Unified Cross-Modal Pre-training for Code Representationcited by 361
- 2023Exphormer: Sparse Transformers for Graphscited by 170
- 2023GraphPrompt: Unifying Pre-Training and Downstream Tasks for Graph Neural Networkscited by 167
20 more
- 2018EMBER: An Open Dataset for Training Static PE Malware Machine Learning Modelscited by 165
- 2022Universal Prompt Tuning for Graph Neural Networkscited by 100
- 2022Empowering Graph Representation Learning with Test-Time Graph Transformationcited by 67
- 2020A Large-Scale Database for Graph Representation Learningcited by 55
- 2020SOREL-20M: A Large Scale Benchmark Dataset for Malicious PE Detectioncited by 49
- 2024CodeXEmbed: A Generalist Embedding Model Family for Multiligual and Multi-task Code Retrievalcited by 34
- 2023GraphPatcher: Mitigating Degree Bias for Graph Neural Networks via Test-time Augmentationcited by 31
- 2023AdapterGNN: Parameter-Efficient Fine-Tuning Improves Generalization in GNNscited by 26
- 2023G-Adapter: Towards Structure-Aware Parameter-Efficient Transfer Learning for Graph Transformer Networkscited by 25
- 2024Graph Learning under Distribution Shifts: A Comprehensive Survey on Domain Adaptation, Out-of-distribution, and Continual Learningcited by 18
- 2025Edge Prompt Tuning for Graph Neural Networkscited by 18
- 2021MOTIF: A Large Malware Reference Dataset with Ground Truth Family Labelscited by 17
- 2022Test-Time Training for Graph Neural Networkscited by 17
- 2024A Survey of Deep Graph Learning under Distribution Shifts: from Graph Out-of-Distribution Generalization to Adaptationcited by 17
- 2022GraphTTA: Test Time Adaptation on Graph Neural Networkscited by 14
- 2021MalNet: A Large-Scale Image Database of Malicious Softwarecited by 11
- 2025LAMDA: A Longitudinal Android Malware Benchmark for Concept Drift Analysiscited by 6
- 2023A Comparison of Graph Neural Networks for Malware Classificationcited by 3
- 2025Structural Alignment Improves Graph Test-Time Adaptationcited by 2
- 2025MalVis: A Large-Scale Image-Based Framework and Dataset for Advancing Android Malware Classificationcited by 2
Built on by
Not cited yet.