让图神经网络自动选择传播深度,提升异质图分类效果
Beyond Fixed Depth: Adaptive Graph Neural Networks for Node Classification Under Varying Homophily
- 根据节点局部结构自适应调整信息聚合深度
- 在多个基准上显著提升标准GNN模型性能
- 统一处理同质与异质图,无需切换模型
图神经网络(GNN)在节点分类任务中取得显著进展,但在异质图(连接节点标签不同)上表现下降。现有方法多采用固定聚合深度,忽略节点间差异;且多数模型仅适配同质或异质场景,缺乏通用性。本文建立理论框架,揭示最优聚合深度随节点局部同质性与邻域结构变化,关键在于保留类别区分信息。基于此,提出新型自适应深度GNN架构,利用理论指导的度量动态选择每节点聚合深度。该方法在统一模型中无缝适应同质与异质模式。大量实验表明,其可一致提升多种标准GNN骨干网络性能。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have achieved significant success in addressing node classification tasks. However, the effectiveness of traditional GNNs degrades on heterophilic graphs, where connected nodes often belong to different labels or properties. While recent work has introduced mechanisms to improve GNN performance under heterophily, certain key limitations still exist. Most existing models apply a fixed aggregation depth across all nodes, overlooking the fact that nodes may require different propagation depths based on their local homophily levels and neighborhood structures. Moreover, many methods are tailored to either homophilic or heterophilic settings, lacking the flexibility to generalize across both regimes. To address these challenges, we develop a theoretical framework that links local structural and label characteristics to information propagation dynamics at the node level. Our analysis shows that optimal aggregation depth varies across nodes and is critical for preserving class-discriminative information. Guided by this insight, we propose a novel adaptive-depth GNN architecture that dynamically selects node-specific aggregation depths using theoretically grounded metrics. Our method seamlessly adapts to both homophilic and heterophilic patterns within a unified model. Extensive experiments demonstrate that our approach consistently enhances the performance of standard GNN backbones across diverse benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。