arXiv:2509.25706cs.LG2025-09

训练时动态合并节点,提升大图GNN效率与适应性。

Adaptive Graph Coarsening for Efficient GNN Training

  • 用K-means根据节点嵌入聚类,边训练边粗化图
  • 在同质与异质数据上均实现高效分类,精度稳定
  • 无需预处理,能自适应学习任务,适合复杂图结构

随着真实世界图规模增大,直接处理变得愈发困难甚至不可行。为应对这一挑战,我们提出一种自适应图粗化方法,在训练过程中联合学习图神经网络(GNN)参数并利用K-means聚类基于节点嵌入合并节点。不同于以往将粗化作为预处理的方法,我们的方法允许在训练期间动态合并节点,避免了预处理步骤,并使节点聚类能够根据学习任务自适应调整,而非仅依赖图的连通性或特征。该方法特别适用于异质性数据等复杂场景。我们在同质与异质节点分类数据集上验证了该方法的有效性,并通过可视化节点嵌入与聚类关系,展示了粗化图在训练中对学习任务的适应能力。

原文摘要 · Abstract (English)

We propose an adaptive graph coarsening method to jointly learn graph neural network (GNN) parameters and merge nodes via K-means clustering during training. As real-world graphs grow larger, processing them directly becomes increasingly challenging and sometimes infeasible. Tailoring algorithms to large-scale data may sacrifice performance, so we instead consider graph reduction to decrease the amount of data used during training. In particular, we propose a method to simultaneously train a GNN and coarsen its graph by partitioning nodes via K-means clustering based on their embeddings. Unlike past graph coarsening works, our approach allows us to merge nodes during training. Not only does this preclude coarsening as a preprocessing step, but our node clusters can adapt to the learning task instead of relying solely on graph connectivity and features. Thus, our method is amenable to scenarios that are challenging for other methods, such as heterophilic data. We validate our approach on both homophilic and heterophilic node classification datasets. We further visualize relationships between node embeddings and their corresponding clusters to illustrate that our coarsened graph adapts to the learning task during training.

图神经网络图粗化自适应

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。