LLM2D
MASKDROID:基于掩码图表示的鲁棒Android恶意软件检测
MASKDROID: Robust Android Malware Detection with Masked Graph Representations
作者: Jingnan Zheng, Jiaohao Liu, An Zhang, Jun Zeng, Ziqi Yang, Zhenkai Liang, Tat-Seng Chua
发布日期: 10/1/2024
arXiv ID: oai:arXiv.org:2409.19594v1

摘要

安卓恶意软件攻击对移动用户构成严重威胁,迫切需要自动检测系统。在恶意软件检测中使用的各种工具中,图表示(例如,函数调用图)在表征安卓应用程序的行为方面发挥了关键作用。然而,尽管在恶意软件检测中取得了令人印象深刻的性能,但当前最先进的基于图的恶意软件检测器容易受到对抗性样本的影响。这些对抗性样本是通过对正常的恶意输入引入特定扰动精心制作的。为了防御对抗性攻击,现有的防御机制通常是检测器的补充,并存在重大局限性,通常依赖于对抗性样本的先验知识,并且无法有效防御未知类型的攻击。在本文中,我们提出了MASKDROID,这是一种功能强大的检测器,具有强大的判别能力来识别恶意软件,并且对对抗性攻击具有非凡的鲁棒性。具体而言,我们在基于图神经网络 (GNN) 的框架中引入了掩蔽机制,迫使MASKDROID使用一小部分(例如,20%)随机选择的节点恢复整个输入图。这种策略使模型能够理解恶意语义并学习更稳定的表示,从而增强其对对抗性攻击的鲁棒性。在以图结构内部依赖项的形式捕获稳定恶意语义的同时,我们进一步采用对比模块来鼓励MASKDROID为良性和恶意类别学习更紧凑的表示,以提高其从良性应用程序和对抗性样本中检测恶意软件的判别能力。