Graph contrastive learning for source localization in social networks

Qing Bao, Ying Jiang, Wang Zhang, Pengfei Jiao, Jing Su

Abstract: Highlights•We propose a graph contrastive learning framework for source localization tasks.•We design a data augmentation strategy and a feature enrichment module.•We modify the loss function to balance the loss between source and non-source nodes.