GNN在疫情源头检测中表现远超传统方法,验证了其有效性。
Graph Neural Networks for Source Detection: A Review and Benchmark Study
- 用四种GNN架构对比传统与MLP方法,控制条件测试性能。
- GNN在多种网络结构下均显著优于其他方法,尤其在复杂拓扑中。
- 适合研究图神经网络评估、流行病溯源或网络安全领域的学者。
当疫情在社交网络中传播时,源节点检测的目标是识别其起源点。自2010年Shah和Zaman提出该问题以来,已有研究引入谣言中心性等概念。随着图神经网络(GNN)的发展,多项工作尝试将其应用于源检测。然而,方法清晰度与可复现性仍有不足,目前尚不清楚GNN是否在可比条件下真正优于传统方法。本文系统回顾现有GNN方法,明确其适用场景与模型结构;随后在受控条件下复现并对比四种代表性GNN架构与多样化的传统及MLP基线。我们还研究了检测能力随时间变化、训练集规模影响及观测时间与传播参数不确定性下的敏感性。实验表明,GNN在多种网络拓扑中显著优于所有测试方法。尽管初衷是质疑GNN的有效性,结果却证明其在该任务中表现卓越。为确保可复现性,代码与数据已开源。最后,我们认为疫情源检测是评估GNN架构的自然且有吸引力的基准任务。
原文摘要 · Abstract (English)
The source detection problem arises when an epidemic process unfolds over a contact network, and the objective is to identify its point of origin, i.e., the source node. Research on this problem began with the seminal work of Shah and Zaman in 2010, who formally defined it and introduced the notion of rumor centrality. With the emergence of Graph Neural Networks (GNNs), several studies have proposed GNN-based approaches to source detection. However, there is room to strengthen methodological clarity and reproducibility across these works. As a result, it remains unclear whether GNNs truly outperform more traditional source detection methods across comparable settings. In this paper, we first systematically review existing GNN-based methods for source detection, clearly outlining the specific settings each addresses and the architectures they employ. We then reproduce and benchmark four representative GNN architectures against a diverse set of traditional and MLP-based baselines under controlled, comparable conditions. We also investigate key questions surrounding this problem, including how detectability evolves over time, how performance scales with training set size, and how sensitive methods are to uncertainty in observation timing and epidemic parameters. Our experiments show that GNNs substantially outperform all other methods we test across a variety of network topologies. Although we initially set out to challenge the notion of GNNs as a solution to source detection, our results instead demonstrate their remarkable effectiveness for this task. To ensure full reproducibility, we release all code and data on GitHub. Finally, we argue that epidemic source detection constitutes a natural and attractive benchmark task for evaluating GNN architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。