用图神经网络检测安卓恶意软件,还能防御对抗攻击。
IoT-based Android Malware Detection Using Graph Neural Network With Adversarial Defense
- 构建应用API图,融合权限与意图特征进行分类
- 在CICMaldroid和Drebin数据集上准确率达98.33%~98.68%
- 提出对抗生成攻击并验证防御有效性,适合安全研究者
随着物联网广泛使用安卓应用,检测恶意安卓应用至关重要。近年来,基于图的深度学习方法通过将应用关系建模为图来生成图嵌入。本文首先证明了基于图神经网络(GNN)的分类器在生成API图嵌入方面的有效性,将图嵌入与权限、意图特征结合,训练多种机器学习与深度学习模型实现安卓恶意软件检测,在CICMaldroid数据集上准确率达98.33%,在Drebin数据集上达98.68%。然而,这类模型易受攻击,攻击者可添加虚假关系以逃避检测。为此,本文提出一种基于生成对抗网络(GAN)的攻击算法VGAE-MalGAN,其生成器生成对抗性恶意软件API图,替代检测器则模拟目标检测器行为。实验表明,该攻击显著降低GNN分类器的检测率;但通过用生成的对抗样本重新训练模型,可提升鲁棒性并缓解攻击影响。
原文摘要 · Abstract (English)
Since the Internet of Things (IoT) is widely adopted using Android applications, detecting malicious Android apps is essential. In recent years, Android graph-based deep learning research has proposed many approaches to extract relationships from applications as graphs to generate graph embeddings. First, we demonstrate the effectiveness of graph-based classification using a Graph Neural Network (GNN)-based classifier to generate API graph embeddings. The graph embeddings are combined with Permission and Intent features to train multiple machine learning and deep learning models for Android malware detection. The proposed classification approach achieves an accuracy of 98.33 percent on the CICMaldroid dataset and 98.68 percent on the Drebin dataset. However, graph-based deep learning models are vulnerable, as attackers can add fake relationships to evade detection by the classifier. Second, we propose a Generative Adversarial Network (GAN)-based attack algorithm named VGAE-MalGAN targeting graph-based GNN Android malware classifiers. The VGAE-MalGAN generator produces adversarial malware API graphs, while the VGAE-MalGAN substitute detector attempts to mimic the target detector. Experimental results show that VGAE-MalGAN can significantly reduce the detection rate of GNN-based malware classifiers. Although the model initially fails to detect adversarial malware, retraining with generated adversarial samples improves robustness and helps mitigate adversarial attacks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。