Heterogeneous Graph Condensation via Role-Aware Clustering

2026-07-03Machine Learning

Machine Learning
AI summary

The authors created a new method called HGC-RC to make training special types of graph neural networks, called heterogeneous graph neural networks (HGNNs), faster and easier on big graphs. Their approach simplifies the graph by grouping nodes based on both labels and types while keeping important connections. This makes it quicker to train without losing important information or performance. They tested their method and found it worked better than previous techniques.

Heterogeneous Graph Neural NetworksGraph CondensationNode EmbeddingsClusteringClass-partitioned ClusteringUnsupervised ClusteringLarge-scale GraphsGraph ReconstructionMachine Learning EfficiencyRole-aware Clustering
Authors
Fuyan Ou, Yulin Hu, Ye Yuan
Abstract
Heterogeneous Graph Neural Networks (HGNNs) have exhibited remarkable efficacy in modeling complex systems with multiple types of nodes and relations, yet their training on large-scale heterogeneous graphs remains computationally prohibitive. Although graph condensation methods can effectively improve learning efficiency on large-scale graphs, existing condensation processes are mainly designed for homogeneous graphs and typically rely on computationally expensive gradient matching or bilevel optimization paradigms, rendering them impractical for heterogeneous settings. To address these limitations, we propose HGC-RC, a simple yet effective role-aware heterogeneous graph condensation framework. Specifically, HGC-RC first extracts semantically enhanced node embeddings via lightweight propagation. It then introduces a role-aware hybrid clustering strategy consisting of class-partitioned clustering for labeled target nodes to preserve class distributions and unsupervised type-wise clustering for non-target nodes to retain critical cross-type connectivity. Finally, a compact heterogeneous graph is efficiently reconstructed based on the resulting cluster assignments. Extensive experiments demonstrate that HGC-RC outperforms state-of-the-art baselines, offering a practical pathway to accelerate HGNN training on large-scale heterogeneous graphs without sacrificing task performance