arXiv:2410.14886cs.LG2024-10IJCAI被引 33

一个模型跨数据集零样本检测图异常,无需微调

Zero-shot Generalist Graph Anomaly Detection with Unified Neighborhood Prompts

  • 用归一化节点属性和通用邻域提示统一不同图的特征空间
  • 在多个真实数据集上超越现有方法,零样本泛化性能优异
  • 适合需要跨图部署、无标注数据的工业级异常检测场景

图异常检测(GAD)旨在识别与正常模式显著偏离的图中节点,在诸多实际应用中至关重要。然而,现有方法均为‘一数据集一模型’,需为每张图单独训练,严重限制了实际应用。为此,本文提出新型零样本通用型GAD方法UNPrompt:仅需在一个图数据集上训练一个模型,即可无需重训或微调地推广至其他图数据集。核心思想是:1)节点属性的可预测性可作为通用异常度量;2)通过在规范化节点属性空间中进行属性预测,学习通用的正常与异常模式。UNPrompt包含两个关键模块:坐标归一化模块统一不同图的节点属性维度与语义,提示学习模块生成通用邻域提示,使属性可预测性成为跨数据集一致的异常分数。在多个真实世界GAD数据集上的实验表明,UNPrompt在通用型GAD设置下显著优于多种对比方法,且在传统‘一模型一数据集’设置下也表现卓越。代码已公开于https://github.com/mala-lab/UNPrompt。

原文摘要 · Abstract (English)

Graph anomaly detection (GAD), which aims to identify nodes in a graph that significantly deviate from normal patterns, plays a crucial role in broad application domains. However, existing GAD methods are one-model-for-one-dataset approaches, i.e., training a separate model for each graph dataset. This largely limits their applicability in real-world scenarios. To overcome this limitation, we propose a novel zero-shot generalist GAD approach UNPrompt that trains a one-for-all detection model, requiring the training of one GAD model on a single graph dataset and then effectively generalizing to detect anomalies in other graph datasets without any retraining or fine-tuning. The key insight in UNPrompt is that i) the predictability of latent node attributes can serve as a generalized anomaly measure and ii) generalized normal and abnormal graph patterns can be learned via latent node attribute prediction in a properly normalized node attribute space. UNPrompt achieves a generalist mode for GAD through two main modules: one module aligns the dimensionality and semantics of node attributes across different graphs via coordinate-wise normalization, while another module learns generalized neighborhood prompts that support the use of latent node attribute predictability as an anomaly score across different datasets. Extensive experiments on real-world GAD datasets show that UNPrompt significantly outperforms diverse competing methods under the generalist GAD setting, and it also has strong superiority under the one-model-for-one-dataset setting. Code is available at https://github.com/mala-lab/UNPrompt.

图异常检测零样本学习通用模型节点属性预测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。