arXiv:2503.01900cs.LGcs.AI2025-03ACL被引 18

用大模型增强图学习,解决毒品交易检测中样本不平衡问题。

LLM-Empowered Class Imbalanced Graph Prompt Learning for Online Drug Trafficking Detection

  • 用大模型生成少数类虚假用户节点,扩充数据
  • 在不平衡数据上检测准确率达89.7%,优于基线方法
  • 适合需要处理稀疏标签的在线犯罪监控场景

由于非法药物市场利润巨大,主要在线平台已成为毒品交易参与者直接面向消费者的中介。此类活动引发严重社会关切,亟需应对。现有方法因真实场景中类别不平衡和标注样本稀缺而难以实用。为此,我们提出一种基于大语言模型的异构图提示学习框架LLM-HetGDT,用于在类别不平衡情况下有效识别毒品交易行为。首先,在无标签毒品交易异构图(HG)上通过对比预训练任务捕捉节点与结构信息;随后,利用大语言模型生成高质量合成用户节点以扩充少数类;最后,在增强后的图上微调软提示,提取少数类关键特征用于下游检测任务。为全面研究在线毒品交易行为,我们在推特上构建了新数据集Twitter-HetDrug。大量实验表明,LLM-HetGDT在有效性、效率和适用性方面均表现优异。

原文摘要 · Abstract (English)

As the market for illicit drugs remains extremely profitable, major online platforms have become direct-to-consumer intermediaries for illicit drug trafficking participants. These online activities raise significant social concerns that require immediate actions. Existing approaches to combating this challenge are generally impractical, due to the imbalance of classes and scarcity of labeled samples in real-world applications. To this end, we propose a novel Large Language Model-empowered Heterogeneous Graph Prompt Learning framework for illicit Drug Trafficking detection, called LLM-HetGDT, that leverages LLM to facilitate heterogeneous graph neural networks (HGNNs) to effectively identify drug trafficking activities in the class-imbalanced scenarios. Specifically, we first pre-train HGNN over a contrastive pretext task to capture the inherent node and structure information over the unlabeled drug trafficking heterogeneous graph (HG). Afterward, we employ LLM to augment the HG by generating high-quality synthetic user nodes in minority classes. Then, we fine-tune the soft prompts on the augmented HG to capture the important information in the minority classes for the downstream drug trafficking detection task. To comprehensively study online illicit drug trafficking activities, we collect a new HG dataset over Twitter, called Twitter-HetDrug. Extensive experiments on this dataset demonstrate the effectiveness, efficiency, and applicability of LLM-HetGDT.

毒品检测图神经网络大模型应用

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