通过标签影响力传播提升图上多标签分类效果
Multi-Label Node Classification with Label Influence Propagation

- 将GNN消息传递分解为传播与变换,分析标签间影响关系
- 构建标签影响图,动态调整标签权重以优化分类
- 在多个基准数据集上超越现有方法,适合复杂图数据
图是一种复杂且多用途的数据结构,在多个领域中广泛应用,其中可能包含多标签节点的情况尤为关键。例如蛋白质在PPI网络中具有多种功能,用户在社交或电商网络中表现出多样化兴趣。针对图上的多标签节点分类(MLNC)问题,已有多种方法被提出。一些方法利用图神经网络(GNNs)捕捉标签共现相关性,另一些则引入标签嵌入以表示标签之间的接近度。然而,这些方法未能充分考虑非欧几里得图数据中标签间的复杂相互影响。为此,我们把GNN中的消息传递过程分解为传播和变换两个操作,并对每个操作中标签间的影响相关性进行了全面分析与量化。基于此,我们提出一种新模型——标签影响力传播(LIP)。具体而言,我们根据整合的标签相关性构建标签影响图,并通过该图传播高阶影响,动态调整学习过程:增强有正向贡献的标签,抑制有负向影响的标签。最终,我们的框架在多个基准数据集上进行评估,各项设置下均持续优于当前最优(SOTA)方法,证明了其在多标签节点分类任务中的有效性。
原文摘要 · Abstract (English)
Graphs are a complex and versatile data structure used across various domains, with possibly multi-label nodes playing a particularly crucial role. Examples include proteins in PPI networks with multiple functions and users in social or e-commerce networks exhibiting diverse interests. Tackling multi-label node classification (MLNC) on graphs has led to the development of various approaches. Some methods leverage graph neural networks (GNNs) to exploit label co-occurrence correlations, while others incorporate label embeddings to capture label proximity. However, these approaches fail to account for the intricate influences between labels in non-Euclidean graph data. To address this issue, we decompose the message passing process in GNNs into two operations: propagation and transformation. We then conduct a comprehensive analysis and quantification of the influence correlations between labels in each operation. Building on these insights, we propose a novel model, Label Influence Propagation (LIP). Specifically, we construct a label influence graph based on the integrated label correlations. Then, we propagate high-order influences through this graph, dynamically adjusting the learning process by amplifying labels with positive contributions and mitigating those with negative influence. Finally, our framework is evaluated on comprehensive benchmark datasets, consistently outperforming SOTA methods across various settings, demonstrating its effectiveness on MLNC tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。