针对多模态漏洞攻击预测,提出高效自动设计神经网络架构的新方法。
A Modality-Aware Cooperative Co-Evolutionary Framework for Multimodal Graph Neural Architecture Search
- 分治式协同进化框架,按模态拆分基因独立演化
- 在3小时GPU内实现81.67%准确率,比现有方法高8.7%
- 适合需要快速构建多源数据模型的网络安全研究者
软件漏洞协同利用攻击对企业构成严重威胁,可通过分析异构多模态漏洞数据缓解。多模态图神经网络(MGNN)能融合跨模态互补信息,提升攻击预测精度。但设计有效MGNN架构困难,因需在每层协调特定模态组件,手动调优不可行。基于遗传算法的图神经架构搜索(GNAS)提供自然解法,但现有方法仅限单模态且忽略模态异质性。为此,提出面向多模态图神经架构搜索的模态感知协同进化算法(MACC-MGNAS)。首先,构建模态感知协同进化(MACC)框架:协调器将全局染色体种群按模态分割为基因组,本地工作者独立演化,协调器重组染色体进行联合评估,有效捕捉单模态GNAS忽略的模态异质性。其次,引入模态感知双轨代理(MADTS)以降低评估成本,加速局部基因演化。第三,设计基于相似性的种群多样性指标(SPDI),自适应平衡探索与利用,加速收敛并避免局部最优。在标准漏洞协同利用(VulCE)数据集上,MACC-MGNAS仅用3 GPU小时即达81.67% F1分数,优于最先进方法8.7个百分点,计算成本降低27%。
原文摘要 · Abstract (English)
Co-exploitation attacks on software vulnerabilities pose severe risks to enterprises, a threat that can be mitigated by analyzing heterogeneous and multimodal vulnerability data. Multimodal graph neural networks (MGNNs) are well-suited to integrate complementary signals across modalities, thereby improving attack-prediction accuracy. However, designing an effective MGNN architecture is challenging because it requires coordinating modality-specific components at each layer, which is infeasible through manual tuning. Genetic algorithm (GA)-based graph neural architecture search (GNAS) provides a natural solution, yet existing methods are confined to single modalities and overlook modality heterogeneity. To address this limitation, we propose a modality-aware cooperative co-evolutionary algorithm for multimodal graph neural architecture search, termed MACC-MGNAS. First, we develop a modality-aware cooperative co-evolution (MACC) framework under a divide-and-conquer paradigm: a coordinator partitions a global chromosome population into modality-specific gene groups, local workers evolve them independently, and the coordinator reassembles chromosomes for joint evaluation. This framework effectively captures modality heterogeneity ignored by single-modality GNAS. Second, we introduce a modality-aware dual-track surrogate (MADTS) method to reduce evaluation cost and accelerate local gene evolution. Third, we design a similarity-based population diversity indicator (SPDI) strategy to adaptively balance exploration and exploitation, thereby accelerating convergence and avoiding local optima. On a standard vulnerabilities co-exploitation (VulCE) dataset, MACC-MGNAS achieves an F1-score of 81.67% within only 3 GPU-hours, outperforming the state-of-the-art competitor by 8.7% F1 while reducing computation cost by 27%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。