首个半监督节点重要性估计框架,通过建模不确定性提升未知节点预测精度。
Semi-supervised Node Importance Estimation with Informative Distribution Modeling for Uncertainty Regularization
- 提出DJE架构,用分布表示同时捕捉节点重要性和不确定性
- 在三个真实数据集上显著优于对比方法,尤其对未标注节点效果更优
- 适合处理标签不全的异构图场景,如社交网络、推荐系统
节点重要性估计是网络分析中的经典问题,支撑多种网络应用。现有方法多依赖拓扑特征或额外信息进行特征增强,但通常采用有监督学习,忽视了实际中真实重要性标签常为部分标注的事实。本文提出首个半监督节点重要性估计框架EASING,用于提升异构图中未标注数据的学习质量。不同于以往方法,EASING显式建模不确定性以反映模型预测置信度。通过深度编码-解码架构DJE,实现节点分布表示,同时输出重要性与不确定性估计,并生成有效伪标签扩充训练集。基于标注与伪标签数据,构建具有可变不确定性正则化的半监督异方差学习机制。在三个真实数据集上的大量实验表明,EASING性能显著优于现有方法。代码已开源:https://github.com/yankai-chen/EASING。
原文摘要 · Abstract (English)
Node importance estimation, a classical problem in network analysis, underpins various web applications. Previous methods either exploit intrinsic topological characteristics, e.g., graph centrality, or leverage additional information, e.g., data heterogeneity, for node feature enhancement. However, these methods follow the supervised learning setting, overlooking the fact that ground-truth node-importance data are usually partially labeled in practice. In this work, we propose the first semi-supervised node importance estimation framework, i.e., EASING, to improve learning quality for unlabeled data in heterogeneous graphs. Different from previous approaches, EASING explicitly captures uncertainty to reflect the confidence of model predictions. To jointly estimate the importance values and uncertainties, EASING incorporates DJE, a deep encoder-decoder neural architecture. DJE introduces distribution modeling for graph nodes, where the distribution representations derive both importance and uncertainty estimates. Additionally, DJE facilitates effective pseudo-label generation for the unlabeled data to enrich the training samples. Based on labeled and pseudo-labeled data, EASING develops effective semi-supervised heteroscedastic learning with varying node uncertainty regularization. Extensive experiments on three real-world datasets highlight the superior performance of EASING compared to competing methods. Codes are available via https://github.com/yankai-chen/EASING.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。