用因果建模提升边分类,让节点特征不再干扰边预测。
Advancing Edge Classification through High-Dimensional Causal Modeling of Node-Edge Interplay

- 将边特征视为高维处理变量,用因果框架建模节点与边的互动。
- 在多个数据集上超越现有方法,最高提升12.3%准确率。
- 可插拔适配现有模型,适合图神经网络研究者使用。
边分类是图应用中的关键任务,但相比链接预测仍较受忽视。现有方法常忽略节点特征对边特征的潜在因果影响,导致相关信息丢失。本文提出因果边分类框架(CECF),首次将因果推断原则应用于边分类任务,并探索将边特征建模为高维处理变量。基于图神经网络(GNN)的节点嵌入,CECF通过缓解节点特征的潜在影响,学习高维边特征的平衡表示;随后,跨注意力网络捕捉节点与边特征间的复杂依赖关系,完成最终分类。大量实验表明,CECF不仅性能更优,且可作为灵活的即插即用模块增强现有方法。我们还提供了实证分析,揭示该高维因果建模框架在何种条件下有效。
原文摘要 · Abstract (English)
Edge classification, a crucial task for graph applications, remains relatively under-explored compared to link prediction. Current methods often overlook the potential causal influences of node features on edge features, leading to a loss of relevant prior information. In this work, we present an empirical exploration using the Causal Edge Classification Framework (CECF). Unlike conventional causal inference methods, CECF is the first framework to apply causal inference principles to the edge classification task and to explore modeling edge features as a high-dimensional treatment within a causal framework. Based on the node embedding of Graph Neural Network (GNN), CECF seeks to learn a balanced representation of high-dimensional edge features by mitigating the potential influence of node features. Then, a cross-attention network captures the complex dependencies between node and edge features for final edge classification. Extensive experiments demonstrate that CECF not only achieves superior performance but also serves as a flexible, plug-and-play enhancement for existing methods. We also provide empirical analyses, offering insights into when and how this high-dimensional causal modeling framework works for the edge classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。