提出CLGNN模型,高效预测时序图中节点的中心性。
CLGNN: A Contrastive Learning-based GNN Model for Betweenness Centrality Prediction on Temporal Graphs
- 用对比学习分离高低中三类中心性节点,缓解数据分布不均。
- 相比精确计算快663.7倍,比现有方法误差低至1/31.4。
- 适合需要快速识别关键节点的时序网络分析场景。
时序介数中心性(TBC)衡量节点在最优时序路径中出现频率,反映其在时序网络中的重要性。但精确计算成本极高,且真实TBC分布极度不平衡,导致学习模型过度拟合零中心性节点,无法准确识别真正关键节点。现有图神经网络方法或无法处理此类不平衡,或忽略时序依赖。为此,我们提出一种可扩展、归纳式的对比学习图神经网络(CLGNN),用于高效准确地预测TBC。CLGNN构建实例图以保持路径有效性和时序顺序,通过均值聚合与边到节点多头注意力机制编码结构和时序特征,并引入时序路径计数和时间编码增强表示。设计稳定性引导的聚类对比模块(KContrastNet),在表示空间中分离高、中、低中心性节点,缓解类别不平衡;同时使用回归模块(ValueNet)估计具体TBC值。该模型支持多种最优路径定义,适应不同语义。大量实验表明,CLGNN在多个基准上均表现优异:相比最先进的精确计算方法,速度提升最高达663.7倍;优于主流静态GNN基线,平均绝对误差降低最多31.4倍,斯皮尔曼相关系数提升16.7倍;超越先进时序GNN,MAE降低最多5.7倍,相关系数提高3.9倍。
原文摘要 · Abstract (English)
Temporal Betweenness Centrality (TBC) measures how often a node appears on optimal temporal paths, reflecting its importance in temporal networks. However, exact computation is highly expensive, and real-world TBC distributions are extremely imbalanced. The severe imbalance leads learning-based models to overfit to zero-centrality nodes, resulting in inaccurate TBC predictions and failure to identify truly central nodes. Existing graph neural network (GNN) methods either fail to handle such imbalance or ignore temporal dependencies altogether. To address these issues, we propose a scalable and inductive contrastive learning-based GNN (CLGNN) for accurate and efficient TBC prediction. CLGNN builds an instance graph to preserve path validity and temporal order, then encodes structural and temporal features using dual aggregation, i.e., mean and edge-to-node multi-head attention mechanisms, enhanced by temporal path count and time encodings. A stability-based clustering-guided contrastive module (KContrastNet) is introduced to separate high-, median-, and low-centrality nodes in representation space, mitigating class imbalance, while a regression module (ValueNet) estimates TBC values. CLGNN also supports multiple optimal path definitions to accommodate diverse temporal semantics. Extensive experiments demonstrate the effectiveness and efficiency of CLGNN across diverse benchmarks. CLGNN achieves up to a 663.7~$\times$ speedup compared to state-of-the-art exact TBC computation methods. It outperforms leading static GNN baselines with up to 31.4~$\times$ lower MAE and 16.7~$\times$ higher Spearman correlation, and surpasses state-of-the-art temporal GNNs with up to 5.7~$\times$ lower MAE and 3.9~$\times$ higher Spearman correlation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。