FPGN: follower prediction framework for infectious disease prevention

Sep 16, 2023·
Jianke Yu
Jianke Yu
,
Xianhang Zhang
,
Hanchen Wang
,
Xiaoyang Wang
,
Wenjie Zhang
,
Ying Zhang
· 1 min read
CCF-B CAS Zone 4 JCR Q2
Abstract

Abstract In recent years, how to prevent the widespread transmission of infectious diseases in communities has been a research hot spot. Tracing close contact with infected individuals is one of the most severe problems. In this work, we present a model called Follower Prediction Graph Network (FPGN) to identify high-risk visitors, which is known as follower prediction. The model is designed to identify visitors who may be infected with a disease by tracking their activities at the exact location of infected visitors. FPGN is inspired by the state-of-the-art temporal graph edge prediction algorithm TGN and draws on the shortcomings of existing algorithms. It utilizes graph structure information based on (

$$\alpha $$

α ,

$$\beta $$

β )-core, time interval statistics by using the statistics of timestamp information, and a GAT-based prediction module to achieve high accuracy in follower prediction. Extensive experiments are conducted on two real datasets, demonstrating the progress of FPGN. The experimental results show that FPGN can achieve the highest results compared with other SOTA baselines. Its AP scores are higher than 0.46, and its AUC scores are higher than 0.62.

Type
Publication
World Wide Web
Status
Peer-reviewed Open access
Awards
CCF-B
World Wide Web · 2023
CAS Zone 4
World Wide Web · 2023
JCR Q2
World Wide Web · 2023
publications

World Wide Web, 2023

Abstract In recent years, how to prevent the widespread transmission of infectious diseases in communities has been a research hot spot. Tracing close contact with infected individuals is one of the most severe problems. In this work, we present a model called Follower Prediction Graph Network (FPGN) to identify high-risk visitors, which is known as follower prediction. The model is designed to identify visitors who may be infected with a disease by tracking their activities at the exact location of infected visitors. FPGN is inspired by the state-of-the-art temporal graph edge prediction algorithm TGN and draws on the shortcomings of existing algorithms. It utilizes graph structure information based on (

$$\alpha $$

α ,

$$\beta $$

β )-core, time interval statistics by using the statistics of timestamp information, and a GAT-based prediction module to achieve high accuracy in follower prediction. Extensive experiments are conducted on two real datasets, demonstrating the progress of FPGN. The experimental results show that FPGN can achieve the highest results compared with other SOTA baselines. Its AP scores are higher than 0.46, and its AUC scores are higher than 0.62.

Jianke Yu
Authors
Jianke Yu (he/him)
PhD (2026) in Graph Machine Learning & Databases
I completed my PhD at the University of Technology Sydney (UTS) in 2026, supervised by Prof. Ying Zhang, A/Prof. Lu Qin and Dr Hanchen Wang. My research develops machine learning algorithms for graph-structured data and database systems — graph neural network architectures, and learning-based methods that improve database and data-mining algorithms (fraud detection, graph similarity, supergraph search, malware detection). My work has appeared in KDD and IEEE TKDE. I am joining Zhejiang Gongshang University as a faculty member in 2027.