动态构建控制流图,提升恶意软件检测的可解释性。
On the Consistency of GNN Explanations for Malware Detection
- 结合规则编码与自编码器动态构建程序图特征
- 多方法融合解释生成,准确率超90%且解释一致
- 适合安全分析、模型可解释性研究者使用
控制流图(CFG)是分析程序执行和刻画恶意行为的关键。随着图神经网络(GNN)的广泛应用,基于CFG的表示在恶意软件检测中表现出色。本文提出一种新框架:动态构建CFG,通过规则编码与自编码器融合方式嵌入节点特征,再用GNN分类器识别恶意行为。为增强可解释性,采用GNNExplainer、PGExplainer及CaptumExplainer(含集成梯度、引导反向传播、显著性三种方法)进行解释。引入一种新聚合方法RankFusion,整合表现最佳解释器输出以提升解释质量。同时评估两种子图提取策略,包括提出的贪心边级组合(GEC)方法以增强结构连贯性。通过准确率、保真度与一致性指标的综合评估,证明该框架在精准识别恶意样本及生成可靠可解释解释方面效果显著。
原文摘要 · Abstract (English)
Control Flow Graphs (CFGs) are critical for analyzing program execution and characterizing malware behavior. With the growing adoption of Graph Neural Networks (GNNs), CFG-based representations have proven highly effective for malware detection. This study proposes a novel framework that dynamically constructs CFGs and embeds node features using a hybrid approach combining rule-based encoding and autoencoder-based embedding. A GNN-based classifier is then constructed to detect malicious behavior from the resulting graph representations. To improve model interpretability, we apply state-of-the-art explainability techniques, including GNNExplainer, PGExplainer, and CaptumExplainer, the latter is utilized three attribution methods: Integrated Gradients, Guided Backpropagation, and Saliency. In addition, we introduce a novel aggregation method, called RankFusion, that integrates the outputs of the top-performing explainers to enhance the explanation quality. We also evaluate explanations using two subgraph extraction strategies, including the proposed Greedy Edge-wise Composition (GEC) method for improved structural coherence. A comprehensive evaluation using accuracy, fidelity, and consistency metrics demonstrates the effectiveness of the proposed framework in terms of accurate identification of malware samples and generating reliable and interpretable explanations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。