用网络控制理论构造节点特征,提升社交网络分类效果
Feature Construction Using Network Control Theory and Rank Encoding for Graph Machine Learning
- 基于平均可控制性与度中心性等拓扑指标构建节点特征
- 新编码方法使图神经网络在GitHub数据集上AUC提升至73.9%
- 适合无原始属性的社交网络场景,尤其对隐私受限数据有效
本文利用图的平均可控制性概念及一种新型秩编码方法,提升图神经网络(GNN)在社交网络分类任务中的性能。尽管GNN在各类网络学习任务中表现优异,但其性能高度依赖节点特征的表达能力。在社交网络中,由于隐私限制或缺乏固有属性,节点特征常不可得,制约了GNN的表现。为此,我们提出两种节点特征构造策略:首先引入平均可控制性及其他中心性度量(记为NCT-EFA),作为刻画网络拓扑关键特性的节点级指标;在此基础上,开发秩编码方法,将平均可控制性或其他图论指标映射至固定维度特征空间,增强特征表示。我们在四个社交网络数据集上,使用六种基准GNN模型进行大量数值实验,对比不同特征构造方法。结果表明,将平均可控制性纳入特征空间显著提升GNN性能;所提秩编码方法优于传统独热度编码,在GraphSAGE模型下使GitHub Stargazers数据集的ROC AUC从68.7%提升至73.9%,验证了其在生成高表达性、高效节点表示方面的有效性。
原文摘要 · Abstract (English)
In this article, we utilize the concept of average controllability in graphs, along with a novel rank encoding method, to enhance the performance of Graph Neural Networks (GNNs) in social network classification tasks. GNNs have proven highly effective in various network-based learning applications and require some form of node features to function. However, their performance is heavily influenced by the expressiveness of these features. In social networks, node features are often unavailable due to privacy constraints or the absence of inherent attributes, making it challenging for GNNs to achieve optimal performance. To address this limitation, we propose two strategies for constructing expressive node features. First, we introduce average controllability along with other centrality metrics (denoted as NCT-EFA) as node-level metrics that capture critical aspects of network topology. Building on this, we develop a rank encoding method that transforms average controllability or any other graph-theoretic metric into a fixed-dimensional feature space, thereby improving feature representation. We conduct extensive numerical evaluations using six benchmark GNN models across four social network datasets to compare different node feature construction methods. Our results demonstrate that incorporating average controllability into the feature space significantly improves GNN performance. Moreover, the proposed rank encoding method outperforms traditional one-hot degree encoding, improving the ROC AUC from 68.7% to 73.9% using GraphSAGE on the GitHub Stargazers dataset, underscoring its effectiveness in generating expressive and efficient node representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。